@strapi/database 4.11.0-alpha.0 → 4.11.0-beta.1

package/lib/index.d.ts

@@ -168,6 +168,8 @@ export interface Database {
       trx: Knex.Transaction;
       rollback: () => Promise<void>;
       commit: () => Promise<void>;
+      onCommit: (cb) => void;
+      onRollback: (cb) => void;
     }) => Promise<unknown>
   ):
     | Promise<unknown>

package/lib/index.js

@@ -61,31 +61,29 @@ class Database {
 
     async function commit() {
       if (notNestedTransaction) {
-        transactionCtx.clear();
-        await trx.commit();
+        await transactionCtx.commit(trx);
       }
     }
 
     async function rollback() {
       if (notNestedTransaction) {
-        transactionCtx.clear();
-        await trx.rollback();
+        await transactionCtx.rollback(trx);
       }
     }
 
     if (!cb) {
-      return {
-        commit,
-        rollback,
-        get() {
-          return trx;
-        },
-      };
+      return { commit, rollback, get: () => trx };
     }
 
     return transactionCtx.run(trx, async () => {
       try {
-        const callbackParams = { trx, commit, rollback };
+        const callbackParams = {
+          trx,
+          commit,
+          rollback,
+          onCommit: transactionCtx.onCommit,
+          onRollback: transactionCtx.onRollback,
+        };
         const res = await cb(callbackParams);
         await commit();
         return res;

package/lib/transaction-context.js

@@ -6,7 +6,7 @@ const storage = new AsyncLocalStorage();
 
 const transactionCtx = {
   async run(store, cb) {
-    return storage.run({ trx: store }, cb);
+    return storage.run({ trx: store, commitCallbacks: [], rollbackCallbacks: [] }, cb);
   },
 
   get() {
@@ -14,11 +14,54 @@ const transactionCtx = {
     return store?.trx;
   },
 
-  clear() {
+  async commit(trx) {
     const store = storage.getStore();
+
+    // Clear transaction from store
+    if (store?.trx) {
+      store.trx = null;
+    }
+
+    // Commit transaction
+    await trx.commit();
+
+    if (!store?.commitCallbacks.length) return;
+
+    // Run callbacks
+    store.commitCallbacks.forEach((cb) => cb());
+    store.commitCallbacks = [];
+  },
+
+  async rollback(trx) {
+    const store = storage.getStore();
+
+    // Clear transaction from store
     if (store?.trx) {
       store.trx = null;
     }
+
+    // Rollback transaction
+    await trx.rollback();
+
+    if (!store?.rollbackCallbacks.length) return;
+
+    // Run callbacks
+    store.rollbackCallbacks.forEach((cb) => cb());
+    store.rollbackCallbacks = [];
+  },
+
+  onCommit(cb) {
+    const store = storage.getStore();
+    if (store?.commitCallbacks) {
+      store.commitCallbacks.push(cb);
+    }
+  },
+
+  onRollback(cb) {
+    const store = storage.getStore();
+    if (store?.rollbackCallbacks) {
+      store.rollbackCallbacks.push(cb);
+    }
   },
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@strapi/database",
-  "version": "4.11.0-alpha.0",
+  "version": "4.11.0-beta.1",
   "description": "Strapi's database layer",
   "homepage": "https://strapi.io",
   "bugs": {
@@ -33,7 +33,7 @@
     "lint": "run -T eslint ."
   },
   "dependencies": {
-    "date-fns": "2.29.3",
+    "date-fns": "2.30.0",
     "debug": "4.3.4",
     "fs-extra": "10.0.0",
     "knex": "2.4.0",
@@ -45,5 +45,5 @@
     "node": ">=14.19.1 <=18.x.x",
     "npm": ">=6.0.0"
   },
-  "gitHead": "ca5fb0fcfab1b6784208d2ddf6d51617b8893b35"
+  "gitHead": "a5fa3bd7e1c4680dd350580620d383612597d25d"
 }