perf(db): improve error checking

arsham · Mar 23, 2022 · 939260d · 939260d
1 parent eca2010
commit 939260d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/db.go b/db.go
@@ -101,6 +101,10 @@ func (t *Transaction) PGX(ctx context.Context, transactions ...func(pgx.Tx) erro
 				}()
 				err = fn(tx)
 			}()
+
+			if err == nil {
+				continue
+			}
 			if errors.Is(err, context.Canceled) {
 				err = &retry.StopError{Err: err}
 				ctx = context.Background()
@@ -142,6 +146,10 @@ func (t *Transaction) DB(ctx context.Context, transactions ...func(Tx) error) er
 				}()
 				err = fn(tx)
 			}()
+
+			if err == nil {
+				continue
+			}
 			if errors.Is(err, context.Canceled) {
 				err = &retry.StopError{Err: err}
 				ctx = context.Background()