feat: rollback transaction wrapper (#7706)

src/lib/db/transaction.ts

@@ -36,10 +36,15 @@ export type DeferredServiceFactory<S> = (db: Knex) => S;
 export type ServiceFactory<S> = (
     config: IUnleashConfig,
 ) => DeferredServiceFactory<S>;
+
 export type WithTransactional<S> = S & {
     transactional: <R>(fn: (service: S) => R) => Promise<R>;
 };
 
+export type WithRollback<S> = S & {
+    rollback: <R>(fn: (service: S) => R) => Promise<R>;
+};
+
 /**
  * @deprecated this is a temporary solution to deal with transactions at the store level.
  * Ideally, we should handle transactions at the service level (each service method should be transactional).
@@ -81,6 +86,25 @@ export function withTransactional<S>(
     return service;
 }
 
+export function withRollback<S>(
+    serviceFactory: (db: Knex) => S,
+    db: Knex,
+): WithRollback<S> {
+    const service = serviceFactory(db) as WithRollback<S>;
+
+    service.rollback = async <R>(fn: (service: S) => R) => {
+        const trx = await db.transaction({ isolationLevel: 'serializable' });
+        try {
+            const transactionService = serviceFactory(trx);
+            return fn(transactionService);
+        } finally {
+            await trx.rollback();
+        }
+    };
+
+    return service;
+}
+
 /** Just for testing purposes */
 export function withFakeTransactional<S>(service: S): WithTransactional<S> {
     const serviceWithFakeTransactional = service as WithTransactional<S>;