@coderich/autograph 0.13.100 → 0.13.101

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@coderich/autograph",
   "main": "index.js",
-  "version": "0.13.100",
+  "version": "0.13.101",
   "publishConfig": {
     "access": "public"
   },

package/src/data/Transaction.js

@@ -5,6 +5,7 @@ module.exports = class Transaction {
   #context;
   #resolver;
   #sourceMap;
+  #queries = [];
 
   constructor(config) {
     this.#schema = config.schema;
@@ -24,13 +25,15 @@ module.exports = class Transaction {
       }));
     }
 
-    return new QueryResolverTransaction({
+    this.#queries.push(new QueryResolverTransaction({
       resolver: this.#resolver,
       schema: this.#schema,
       context: this.#context,
       transaction: this.#sourceMap.get(client),
       query: { model: `${model}` },
-    });
+    }));
+
+    return this.#queries.at(-1);
   }
 
   commit() {
@@ -42,8 +45,10 @@ module.exports = class Transaction {
   }
 
   #close(op) {
-    return Promise.all(Array.from(this.#sourceMap.entries()).map(([client, promise]) => {
-      return promise.then(transaction => transaction[op]());
-    }));
+    return Promise.all(this.#queries.map(q => q.promise())).then(() => {
+      return Promise.all(Array.from(this.#sourceMap.entries()).map(([client, promise]) => {
+        return promise.then(transaction => transaction[op]());
+      }));
+    });
   }
 };

package/src/query/QueryResolver.js

@@ -1,7 +1,7 @@
 const get = require('lodash.get');
 const Util = require('@coderich/util');
 const QueryBuilder = require('./QueryBuilder');
-const { mergeDeep } = require('../service/AppService');
+const { mergeDeep, withResolvers } = require('../service/AppService');
 
 module.exports = class QueryResolver extends QueryBuilder {
   #model;
@@ -9,6 +9,7 @@ module.exports = class QueryResolver extends QueryBuilder {
   #config;
   #context;
   #resolver;
+  #resolution = withResolvers(); // Promise for when the query is resolved
 
   constructor(config) {
     const { schema, context, resolver, query } = config;
@@ -20,7 +21,21 @@ module.exports = class QueryResolver extends QueryBuilder {
     this.#model = schema.models[query.model];
   }
 
+  promise() {
+    return this.#resolution.promise;
+  }
+
   terminate() {
+    return this.#terminate().then((result) => {
+      this.#resolution.resolve(result);
+      return result;
+    }).catch((e) => {
+      this.#resolution.reject(e);
+      return Promise.reject(e);
+    });
+  }
+
+  #terminate() {
     const query = super.terminate();
     const { op, args: { input } } = query.toObject();
 
@@ -101,7 +116,7 @@ module.exports = class QueryResolver extends QueryBuilder {
         });
       }
       default: {
-        throw new Error(`Unknown operation "${op}"`);
+        return Promise.reject(new Error(`Unknown operation "${op}"`));
       }
     }
   }

package/src/service/AppService.js

@@ -44,3 +44,9 @@ exports.JSONParse = (mixed) => {
     return undefined;
   }
 };
+
+exports.withResolvers = () => {
+  let resolve, reject;
+  const promise = new Promise((res, rej) => { resolve = res; reject = rej; });
+  return { promise, resolve, reject };
+};