fix: transaction usage in the identity persister (#404)

ory · May 11, 2020 · 7f5072d · 7f5072d
1 parent b76d380
commit 7f5072d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/persistence/sql/persister_identity.go b/persistence/sql/persister_identity.go
@@ -148,7 +148,9 @@ func (p *Persister) CreateIdentity(ctx context.Context, i *identity.Identity) er
  return err
  }
 
- return sqlcon.HandleError(p.GetConnection(ctx).Transaction(func(tx *pop.Connection) error {
+ return sqlcon.HandleError(p.Transaction(ctx, func(tx *pop.Connection) error {
+ ctx := WithTransaction(ctx, tx)
+
  if err := tx.Create(i); err != nil {
  return err
  }
@@ -185,7 +187,8 @@ func (p *Persister) UpdateIdentity(ctx context.Context, i *identity.Identity) er
  return err
  }
 
- return sqlcon.HandleError(p.GetConnection(ctx).Transaction(func(tx *pop.Connection) error {
+ return sqlcon.HandleError(p.Transaction(ctx, func(tx *pop.Connection) error {
+ ctx := WithTransaction(ctx, tx)
  if count, err := tx.Where("id = ?", i.ID).Count(i); err != nil {
  return err
  } else if count == 0 {