proto.io 0.0.225 → 0.0.227

package/dist/index.mjs

@@ -262,6 +262,7 @@ const proxy = (x) => {
 //  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 //  THE SOFTWARE.
 //
+const afterCommitTasks = new WeakMap();
 class _ProtoQuery extends TQuery {
     _proto;
     _opts;
@@ -380,6 +381,15 @@ class _ProtoQuery extends TQuery {
             })();
         }
     }
+    _do_tasks(session, task) {
+        const registers = session && afterCommitTasks.get(session);
+        if (!_.isNil(registers)) {
+            registers.push(task);
+        }
+        else {
+            task();
+        }
+    }
     async insertMany(values, options) {
         const objs = this._objectMethods(await this._dispatcher(options).insert({
             className: this.className,
@@ -387,30 +397,38 @@ class _ProtoQuery extends TQuery {
             matches: this[PVK].options.matches,
             groupMatches: this[PVK].options.groupMatches,
         }, values));
-        this._on_upsert(objs);
-        if (!options?.silent || !options.master)
-            this._on_upsert_traggers(objs);
+        this._do_tasks(options?.session, () => {
+            this._on_upsert(objs);
+            if (!options?.silent || !options.master)
+                this._on_upsert_traggers(objs);
+        });
         return objs;
     }
     async updateMany(update, options) {
         const objs = this._objectMethods(await this._dispatcher(options).update(this._queryOptions, update));
-        this._on_upsert(objs);
-        if (!options?.silent || !options.master)
-            this._on_upsert_traggers(objs);
+        this._do_tasks(options?.session, () => {
+            this._on_upsert(objs);
+            if (!options?.silent || !options.master)
+                this._on_upsert_traggers(objs);
+        });
         return objs;
     }
     async upsertMany(update, setOnInsert, options) {
         const objs = this._objectMethods(await this._dispatcher(options).upsert(this._queryOptions, update, setOnInsert));
-        this._on_upsert(objs);
-        if (!options?.silent || !options.master)
-            this._on_upsert_traggers(objs);
+        this._do_tasks(options?.session, () => {
+            this._on_upsert(objs);
+            if (!options?.silent || !options.master)
+                this._on_upsert_traggers(objs);
+        });
         return objs;
     }
     async deleteMany(options) {
         const objs = this._objectMethods(await this._dispatcher(options).delete(this._queryOptions));
-        this._on_delete(objs);
-        if (!options?.silent || !options.master)
-            this._on_delete_traggers(objs);
+        this._do_tasks(options?.session, () => {
+            this._on_delete(objs);
+            if (!options?.silent || !options.master)
+                this._on_delete_traggers(objs);
+        });
         return objs;
     }
 }
@@ -1746,8 +1764,18 @@ class ProtoService extends ProtoType {
     lockTable(className, update) {
         return this.storage.lockTable(className, update);
     }
-    withTransaction(callback, options) {
-        return this.storage.withTransaction((storage) => callback(_.create(this, { _storage: storage })), options);
+    async withTransaction(callback, options) {
+        const tasks = afterCommitTasks.get(this) ?? [];
+        const result = await this.storage.withTransaction((storage) => {
+            const payload = _.create(this, { _storage: storage });
+            afterCommitTasks.set(payload, tasks);
+            return callback(payload);
+        }, options);
+        if (!afterCommitTasks.has(this)) {
+            for (const task of tasks)
+                task();
+        }
+        return result;
     }
     generateUploadToken(options = {}) {
         return this[PVK].jwtSign({