@webda/core 3.6.0 → 3.8.0

package/lib/stores/store.js

@@ -57,6 +57,7 @@ export class StoreParameters extends ServiceParameters {
         this.forceModel ?? (this.forceModel = false);
         this.slowQueryThreshold ?? (this.slowQueryThreshold = 30000);
         this.modelAliases ?? (this.modelAliases = {});
+        this.additionalModels ?? (this.additionalModels = []);
     }
 }
 /**
@@ -139,6 +140,21 @@ class Store extends Service {
         if (!this.parameters.strict) {
             recursive(this._model.getHierarchy().children, 1);
         }
+        // Add additional models
+        if (this.parameters.additionalModels.length) {
+            // Strict mode is to only allow one model per store
+            if (this.parameters.strict) {
+                this.log("ERROR", "Cannot add additional models in strict mode");
+            }
+            else {
+                for (let modelType of this.parameters.additionalModels) {
+                    const model = app.getModel(modelType);
+                    this._modelsHierarchy[model.getIdentifier(false)] = 0;
+                    this._modelsHierarchy[model.getIdentifier()] = 0;
+                    recursive(model.getHierarchy().children, 1);
+                }
+            }
+        }
         if (this.getParameters().expose) {
             this.log("WARN", "Exposing a store is not recommended, use a DomainService instead to expose all your CoreModel");
         }
@@ -371,12 +387,7 @@ class Store extends Service {
      */
     initModel(object = {}) {
         var _a;
-        if (object instanceof CoreModel) {
-            object.__type ?? (object.__type = this.getWebda().getApplication().getModelFromInstance(object) || this._modelType);
-        }
-        else {
-            object.__type ?? (object.__type = this._modelType);
-        }
+        object.__type ?? (object.__type = this.getWebda().getApplication().getModelFromInstance(object) || this._modelType);
         // Make sure to send a model object
         if (!(object instanceof this._model)) {
             // Dynamic load type
@@ -391,11 +402,11 @@ class Store extends Service {
                     if (!this.parameters.defaultModel) {
                         throw new Error(`Unknown model ${object.__type} found for Store(${this.getName()})`);
                     }
-                    object = this._model.factory(this._model, object);
+                    object = this._model.factory(object);
                 }
             }
             else {
-                object = this._model.factory(this._model, object);
+                object = this._model.factory(object);
             }
         }
         if (!object.getUuid()) {
@@ -455,7 +466,7 @@ class Store extends Service {
         this.metrics.operations_total.inc({ operation: "increment" });
         await this._incrementAttributes(uid, params, updateDate);
         await this._cacheStore?._incrementAttributes(uid, params, updateDate);
-        return this.emitSync("Store.PartialUpdated", {
+        this.emitSync("Store.PartialUpdated", {
             object_id: uid,
             store: this,
             updateDate,
@@ -463,6 +474,7 @@ class Store extends Service {
                 increments: params
             }
         });
+        return updateDate;
     }
     /**
      * Helper function that call incrementAttributes
@@ -501,6 +513,7 @@ class Store extends Service {
                 }
             }
         });
+        return updateDate;
     }
     /**
      * Remove an item from an array in the model
@@ -527,29 +540,48 @@ class Store extends Service {
                 }
             }
         });
+        return updateDate;
     }
     /**
-     * Get all results without pagination
+     * Iterate through the results
      *
      * This can be resource consuming
      *
      * @param query
      * @param context
      */
-    async queryAll(query, context) {
-        const res = [];
+    async *iterate(query = "", context) {
         if (query.includes("OFFSET")) {
-            throw new Error("Cannot contain an OFFSET for queryAll method");
+            throw new Error("Cannot contain an OFFSET for iterate method");
         }
         let continuationToken;
         do {
             let q = query + (continuationToken !== undefined ? ` OFFSET "${continuationToken}"` : "");
             let page = await this.query(q, context);
+            for (let item of page.results) {
+                yield item;
+            }
             continuationToken = page.continuationToken;
-            res.push(...page.results);
         } while (continuationToken);
+    }
+    // REFACTOR . >= 4
+    /**
+     * Query all the results
+     *
+     *
+     * @param query
+     * @param context
+     * @returns
+     * @deprecated use iterate instead
+     */
+    async queryAll(query, context) {
+        let res = [];
+        for await (let item of this.iterate(query, context)) {
+            res.push(item);
+        }
         return res;
     }
+    // END_REFACTOR
     /**
      * Check that __type Comparison is only used with = and CONTAINS
      * If CONTAINS is used, move __type to __types
@@ -718,30 +750,42 @@ class Store extends Service {
     async create(object, ctx = undefined) {
         object = this.initModel(object);
         // Dates should be store by the Store
-        object._creationDate ?? (object._creationDate = new Date());
-        object._lastUpdate = new Date();
+        if (!object._creationDate) {
+            object._creationDate = object._lastUpdate = new Date();
+        }
+        else {
+            object._lastUpdate = new Date();
+        }
         const ancestors = this.getWebda().getApplication().getModelHierarchy(object.__type).ancestors;
         object.__types = [object.__type, ...ancestors].filter(i => i !== "Webda/CoreModel" && i !== "CoreModel");
         if (ctx) {
             object.setContext(ctx);
         }
-        await this.emitSync("Store.Save", {
+        // Handle object auto listener
+        const evt = {
             object: object,
             store: this,
             context: ctx
-        });
-        // Handle object auto listener
-        await object._onSave();
+        };
+        await Promise.all([
+            this.emitSync("Store.Save", evt),
+            object?.__class.emitSync("Store.Save", evt),
+            object._onSave()
+        ]);
         this.metrics.operations_total.inc({ operation: "save" });
         let res = await this._save(object);
         await this._cacheStore?._save(object);
         object = this.initModel(res);
-        await this.emitSync("Store.Saved", {
+        const evtSaved = {
             object: object,
             store: this,
             context: ctx
-        });
-        await object._onSaved();
+        };
+        await Promise.all([
+            this.emitSync("Store.Saved", evtSaved),
+            object?.__class.emitSync("Store.Saved", evtSaved),
+            object._onSaved()
+        ]);
         return object;
     }
     /**
@@ -863,30 +907,27 @@ class Store extends Service {
         }
         object._lastUpdate = new Date();
         this.metrics.operations_total.inc({ operation: "get" });
-        let load = await this._getFromCache(object[this._uuidField], true);
+        const uuid = object.getUuid ? object.getUuid() : object[this._uuidField];
+        let load = await this._getFromCache(uuid, true);
         if (load.__type !== this._modelType && this.parameters.strict) {
-            this.log("WARN", `Object '${object[this._uuidField]}' was not created by this store ${load.__type}:${this._modelType}`);
-            throw new StoreNotFoundError(object[this._uuidField], this.getName());
+            this.log("WARN", `Object '${uuid}' was not created by this store ${load.__type}:${this._modelType}`);
+            throw new StoreNotFoundError(uuid, this.getName());
         }
         loaded = this.initModel(load);
         if (object instanceof CoreModel) {
             loaded.setContext(object.getContext());
         }
-        if (partial) {
-            await this.emitSync(`Store.PatchUpdate`, {
-                object: loaded,
-                store: this,
-                update: object
-            });
-        }
-        else {
-            await this.emitSync(`Store.Update`, {
-                object: loaded,
-                store: this,
-                update: object
-            });
-        }
-        await loaded._onUpdate(object);
+        const update = object;
+        const evt = {
+            object: loaded,
+            store: this,
+            update
+        };
+        await Promise.all([
+            this.emitSync(partial ? `Store.PatchUpdate` : `Store.Update`, evt),
+            object?.__class?.emitSync(partial ? `Store.PatchUpdate` : `Store.Update`, evt),
+            loaded._onUpdate(object)
+        ]);
         let res;
         if (conditionField !== null) {
             conditionField ?? (conditionField = "_lastUpdate");
@@ -894,8 +935,10 @@ class Store extends Service {
         }
         if (partial) {
             this.metrics.operations_total.inc({ operation: "partialUpdate" });
-            await this._patch(object, object[this._uuidField], conditionValue, conditionField);
-            await this._cacheStore?._patch(object, object[this._uuidField], load._lastUpdate, "_lastUpdate");
+            await Promise.all([
+                this._patch(object, uuid, conditionValue, conditionField),
+                this._cacheStore?._patch(object, uuid, load._lastUpdate, "_lastUpdate")
+            ]);
             res = object;
         }
         else {
@@ -905,30 +948,27 @@ class Store extends Service {
             }
             object = this.initModel(object);
             this.metrics.operations_total.inc({ operation: "update" });
-            res = await this._update(object, object[this._uuidField], conditionValue, conditionField);
-            await this._cacheStore?._update(object, object[this._uuidField], load._lastUpdate, "_lastUpdate");
-        }
-        // Return updated
-        for (let i in res) {
-            loaded[i] = res[i];
-        }
-        for (let i in object) {
-            loaded[i] = object[i];
-        }
-        saved = this.initModel(loaded);
-        if (partial) {
-            await this.emitSync(`Store.PatchUpdated`, {
-                object: saved,
-                store: this
-            });
-        }
-        else {
-            await this.emitSync(`Store.Updated`, {
-                object: saved,
-                store: this
-            });
-        }
-        await saved._onUpdated();
+            res = (await Promise.all([
+                this._update(object, uuid, conditionValue, conditionField),
+                this._cacheStore?._update(object, uuid, load._lastUpdate, "_lastUpdate")
+            ]))[0];
+        }
+        // Reinit save
+        saved = this.initModel({
+            ...loaded,
+            ...res
+        });
+        const evtUpdated = {
+            object: saved,
+            store: this,
+            update,
+            previous: loaded
+        };
+        await Promise.all([
+            this.emitSync(partial ? `Store.PatchUpdated` : `Store.Updated`, evtUpdated),
+            saved?.__class.emitSync(partial ? `Store.PatchUpdated` : `Store.Updated`, evtUpdated),
+            saved._onUpdated()
+        ]);
         return saved;
     }
     /**
@@ -1133,12 +1173,16 @@ class Store extends Service {
                 throw new UpdateConditionFailError(to_delete.getUuid(), writeConditionField, writeCondition);
             }
         }
-        // Send preevent
-        await this.emitSync("Store.Delete", {
+        const evt = {
             object: to_delete,
             store: this
-        });
-        await to_delete._onDelete();
+        };
+        // Send preevent
+        await Promise.all([
+            this.emitSync("Store.Delete", evt),
+            to_delete?.__class.emitSync("Store.Delete", evt),
+            to_delete._onDelete()
+        ]);
         // If async we just tag the object as deleted
         if (this.parameters.asyncDelete && !sync) {
             this.metrics.operations_total.inc({ operation: "partialUpdate" });
@@ -1156,11 +1200,15 @@ class Store extends Service {
             await this._cacheStore?._delete(to_delete.getUuid(), writeCondition, writeConditionField);
         }
         // Send post event
-        await this.emitSync("Store.Deleted", {
+        const evtDeleted = {
             object: to_delete,
             store: this
-        });
-        await to_delete._onDeleted();
+        };
+        await Promise.all([
+            this.emitSync("Store.Deleted", evtDeleted),
+            to_delete.__class.emitSync("Store.Deleted", evtDeleted),
+            to_delete._onDeleted()
+        ]);
     }
     /**
      * By default we cannot know if the store will trigger or not
@@ -1224,12 +1272,12 @@ class Store extends Service {
         }
         object = this.initModel(object);
         object.setContext(ctx);
-        await this.emitSync("Store.Get", {
+        const evt = {
             object: object,
             store: this,
             context: ctx
-        });
-        await object._onGet();
+        };
+        await Promise.all([this.emitSync("Store.Get", evt), object.__class.emitSync("Store.Get", evt), object._onGet()]);
         return object;
     }
     /**
@@ -1328,7 +1376,7 @@ class Store extends Service {
     async operationCreate(ctx, model) {
         let body = await ctx.getInput();
         const modelPrototype = this.getWebda().getApplication().getModel(model);
-        let object = modelPrototype.factory(modelPrototype, body, ctx);
+        let object = modelPrototype.factory(body, ctx);
         object._creationDate = new Date();
         await object.checkAct(ctx, "create");
         try {
@@ -1343,12 +1391,13 @@ class Store extends Service {
         }
         await this.save(object, ctx);
         ctx.write(object);
-        await this.emitSync("Store.WebCreate", {
+        const evt = {
             context: ctx,
             values: body,
             object: object,
             store: this
-        });
+        };
+        await Promise.all([object.__class.emitSync("Store.WebCreate", evt), this.emitSync("Store.WebCreate", evt)]);
     }
     /**
      * Handle obect action
@@ -1361,23 +1410,28 @@ class Store extends Service {
             throw new WebdaError.NotFound("Object not found or is deleted");
         }
         await object.checkAct(ctx, action);
-        await this.emitSync("Store.Action", {
+        const evt = {
             action: action,
             object: object,
             store: this,
             context: ctx
-        });
+        };
+        await Promise.all([this.emitSync("Store.Action", evt), object.__class.emitSync("Store.Action", evt)]);
         const res = await object[action](ctx);
         if (res) {
             ctx.write(res);
         }
-        await this.emitSync("Store.Actioned", {
+        const evtActioned = {
             action: action,
             object: object,
             store: this,
             context: ctx,
             result: res
-        });
+        };
+        await Promise.all([
+            this.emitSync("Store.Actioned", evtActioned),
+            object?.__class.emitSync("Store.Actioned", evtActioned)
+        ]);
     }
     /**
      * Handle collection action
@@ -1385,23 +1439,25 @@ class Store extends Service {
      */
     async httpGlobalAction(ctx, model = this._model) {
         let action = ctx.getHttpContext().getUrl().split("/").pop();
-        await this.emitSync("Store.Action", {
+        const evt = {
             action: action,
             store: this,
             context: ctx,
             model
-        });
+        };
+        await Promise.all([this.emitSync("Store.Action", evt), model.emitSync("Store.Action", evt)]);
         const res = await model[action](ctx);
         if (res) {
             ctx.write(res);
         }
-        await this.emitSync("Store.Actioned", {
+        const evtActioned = {
             action: action,
             store: this,
             context: ctx,
             result: res,
             model
-        });
+        };
+        await Promise.all([this.emitSync("Store.Actioned", evtActioned), model?.emitSync("Store.Actioned", evtActioned)]);
     }
     /**
      * Handle HTTP Update for an object
@@ -1432,7 +1488,8 @@ class Store extends Service {
                 delete updateObject[i];
             });
             updateObject.setContext(ctx);
-            updateObject.load(body);
+            updateObject.setUuid(uuid);
+            updateObject.load(body, false, false);
             await this.patch(updateObject);
             object = undefined;
         }
@@ -1457,13 +1514,14 @@ class Store extends Service {
             object = await this.update(updateObject);
         }
         ctx.write(object);
-        await this.emitSync("Store.WebUpdate", {
+        const evt = {
             context: ctx,
             updates: body,
             object: object,
             store: this,
             method: ctx.getHttpContext().getMethod()
-        });
+        };
+        await Promise.all([object?.__class.emitSync("Store.WebUpdate", evt), this.emitSync("Store.WebUpdate", evt)]);
     }
     /**
      * Handle GET on object
@@ -1483,11 +1541,12 @@ class Store extends Service {
         }
         await object.checkAct(ctx, "get");
         ctx.write(object);
-        await this.emitSync("Store.WebGet", {
+        const evt = {
             context: ctx,
             object: object,
             store: this
-        });
+        };
+        await Promise.all([this.emitSync("Store.WebGet", evt), object.__class.emitSync("Store.WebGet", evt)]);
         ctx.write(object);
     }
     /**
@@ -1507,11 +1566,12 @@ class Store extends Service {
         // Might still run into: Have trouble to handle the Content-Length on API Gateway so returning an empty object for now
         ctx.writeHead(204, { "Content-Length": "0" });
         await this.delete(uuid);
-        await this.emitSync("Store.WebDelete", {
+        const evt = {
             context: ctx,
             object_id: uuid,
             store: this
-        });
+        };
+        await Promise.all([this.emitSync("Store.WebDelete", evt), object.__class.emitSync("Store.WebDelete", evt)]);
     }
     /**
      * Return the model uuid field