Execute dbMarkAsActivelyScheduled in a DB transaction

Author: Jeff Yanta

pkg/code/data/fulfillment/postgres/model.go

@@ -478,25 +478,27 @@ func dbPutAllInTx(ctx context.Context, tx *sqlx.Tx, models []*fulfillmentModel)
 }
 
 func dbMarkAsActivelyScheduled(ctx context.Context, db *sqlx.DB, id uint64) error {
-	if id == 0 {
-		return fulfillment.ErrFulfillmentNotFound
-	}
+	return pgutil.ExecuteInTx(ctx, db, sql.LevelDefault, func(tx *sqlx.Tx) error {
+		if id == 0 {
+			return fulfillment.ErrFulfillmentNotFound
+		}
 
-	query := `UPDATE ` + fulfillmentTableName + ` SET disable_active_scheduling = false WHERE id = $1`
-	res, err := db.ExecContext(ctx, query, id)
-	if err != nil {
-		return err
-	}
+		query := `UPDATE ` + fulfillmentTableName + ` SET disable_active_scheduling = false WHERE id = $1`
+		res, err := db.ExecContext(ctx, query, id)
+		if err != nil {
+			return err
+		}
 
-	rowsAffected, err := res.RowsAffected()
-	if err != nil {
-		return err
-	}
+		rowsAffected, err := res.RowsAffected()
+		if err != nil {
+			return err
+		}
 
-	if rowsAffected == 0 {
-		return fulfillment.ErrFulfillmentNotFound
-	}
-	return nil
+		if rowsAffected == 0 {
+			return fulfillment.ErrFulfillmentNotFound
+		}
+		return nil
+	})
 }
 
 func dbGetById(ctx context.Context, db *sqlx.DB, id uint64) (*fulfillmentModel, error) {