feat(digitalTwin): implement generic document pipes triggers (#321)

kuzzleio · Oct 31, 2023 · 60a21fc · 60a21fc
1 parent 2ff4d61
commit 60a21fc
Show file tree

Hide file tree

Showing 10 changed files with 531 additions and 327 deletions.
diff --git a/lib/modules/asset/AssetService.ts b/lib/modules/asset/AssetService.ts
@@ -1,4 +1,4 @@
-import { BadRequestError, User } from "kuzzle";
+import { BadRequestError, KuzzleRequest, User } from "kuzzle";
 import {
   BaseRequest,
   DocumentSearchResult,
@@ -30,6 +30,7 @@ import {
   lock,
   onAsk,
   BaseService,
+  SearchParams,
 } from "../shared";
 
 import { AssetHistoryService } from "./AssetHistoryService";
@@ -69,41 +70,43 @@ export class AssetService extends BaseService {
 
   public async get(
     engineId: string,
-    assetId: string
+    assetId: string,
+    request: KuzzleRequest
   ): Promise<KDocument<AssetContent>> {
-    return this.sdk.document.get<AssetContent>(
+    return this.getDocument<AssetContent>(request, assetId, {
+      collection: InternalCollection.ASSETS,
       engineId,
-      InternalCollection.ASSETS,
-      assetId
-    );
+    });
   }
 
   /**
    * Updates an asset metadata
    */
   public async update(
-    user: User,
     engineId: string,
     assetId: string,
     metadata: Metadata,
-    { refresh }: { refresh: any }
+    request: KuzzleRequest
   ): Promise<KDocument<AssetContent>> {
     return lock(`asset:${engineId}:${assetId}`, async () => {
-      const asset = await this.get(engineId, assetId);
+      const asset = await this.get(engineId, assetId, request);
 
       const updatedPayload = await this.app.trigger<EventAssetUpdateBefore>(
         "device-manager:asset:update:before",
         { asset, metadata }
       );
 
-      const updatedAsset = await this.impersonatedSdk(
-        user
-      ).document.update<AssetContent>(
-        engineId,
-        InternalCollection.ASSETS,
-        assetId,
-        { metadata: updatedPayload.metadata },
-        { refresh, source: true }
+      const updatedAsset = await this.updateDocument<AssetContent>(
+        request,
+        {
+          _id: assetId,
+          _source: { metadata: updatedPayload.metadata },
+        },
+        {
+          collection: InternalCollection.ASSETS,
+          engineId,
+        },
+        { source: true }
       );
 
       await this.assetHistoryService.add<AssetHistoryEventMetadata>(engineId, [
@@ -133,12 +136,11 @@ export class AssetService extends BaseService {
   }
 
   public async create(
-    user: User,
     engineId: string,
     model: string,
     reference: string,
     metadata: JSONObject,
-    { refresh }: { refresh: any }
+    request: KuzzleRequest
   ): Promise<KDocument<AssetContent>> {
     const assetId = AssetSerializer.id(model, reference);
 
@@ -167,20 +169,24 @@ export class AssetService extends BaseService {
         measures[name] = null;
       }
 
-      const asset = await this.impersonatedSdk(
-        user
-      ).document.create<AssetContent>(
-        engineId,
-        InternalCollection.ASSETS,
+      const asset = await this.createDocument<AssetContent>(
+        request,
         {
-          linkedDevices: [],
-          measures,
-          metadata: { ...assetMetadata, ...metadata },
-          model,
-          reference,
+          _id: assetId,
+          _source: {
+            groups: [],
+            lastMeasuredAt: null,
+            linkedDevices: [],
+            measures,
+            metadata: { ...assetMetadata, ...metadata },
+            model,
+            reference,
+          },
         },
-        assetId,
-        { refresh }
+        {
+          collection: InternalCollection.ASSETS,
+          engineId,
+        }
       );
 
       await this.assetHistoryService.add<AssetHistoryEventMetadata>(engineId, [
@@ -202,13 +208,15 @@ export class AssetService extends BaseService {
   }
 
   public async delete(
-    user: User,
     engineId: string,
     assetId: string,
-    { refresh, strict }: { refresh: any; strict: boolean }
+    request: KuzzleRequest
   ) {
+    const user = request.getUser();
+    const strict = request.getBoolean("strict");
+
     return lock<void>(`asset:${engineId}:${assetId}`, async () => {
-      const asset = await this.get(engineId, assetId);
+      const asset = await this.get(engineId, assetId, request);
 
       if (strict && asset._source.linkedDevices.length !== 0) {
         throw new BadRequestError(
@@ -223,35 +231,22 @@ export class AssetService extends BaseService {
         );
       }
 
-      await this.sdk.document.delete(
+      await this.deleteDocument(request, assetId, {
+        collection: InternalCollection.ASSETS,
         engineId,
-        InternalCollection.ASSETS,
-        assetId,
-        {
-          refresh,
-        }
-      );
+      });
     });
   }
 
   public async search(
     engineId: string,
-    searchBody: JSONObject,
-    {
-      from,
-      size,
-      scroll,
-      lang,
-    }: { from?: number; size?: number; scroll?: string; lang?: string }
+    searchParams: SearchParams,
+    request: KuzzleRequest
   ): Promise<SearchResult<KHit<AssetContent>>> {
-    const result = await this.sdk.document.search<AssetContent>(
+    return await this.searchDocument<AssetContent>(request, searchParams, {
+      collection: InternalCollection.ASSETS,
       engineId,
-      InternalCollection.ASSETS,
-      searchBody,
-      { from, lang, scroll, size }
-    );
-
-    return result;
+    });
   }
 
   public async migrateTenant(

diff --git a/lib/modules/asset/AssetsController.ts b/lib/modules/asset/AssetsController.ts
@@ -124,7 +124,7 @@ export class AssetsController {
     const assetId = request.getId();
     const engineId = request.getString("engineId");
 
-    const asset = await this.assetService.get(engineId, assetId);
+    const asset = await this.assetService.get(engineId, assetId, request);
 
     return AssetSerializer.serialize(asset);
   }
@@ -133,16 +133,12 @@ export class AssetsController {
     const assetId = request.getId();
     const engineId = request.getString("engineId");
     const metadata = request.getBodyObject("metadata");
-    const refresh = request.getRefresh();
 
     const updatedAsset = await this.assetService.update(
-      request.getUser(),
       engineId,
       assetId,
       metadata,
-      {
-        refresh,
-      }
+      request
     );
 
     return AssetSerializer.serialize(updatedAsset);
@@ -153,17 +149,13 @@ export class AssetsController {
     const model = request.getBodyString("model");
     const reference = request.getBodyString("reference");
     const metadata = request.getBodyObject("metadata", {});
-    const refresh = request.getRefresh();
 
     const asset = await this.assetService.create(
-      request.getUser(),
       engineId,
       model,
       reference,
       metadata,
-      {
-        refresh,
-      }
+      request
     );
 
     return AssetSerializer.serialize(asset);
@@ -172,33 +164,16 @@ export class AssetsController {
   async delete(request: KuzzleRequest): Promise<ApiAssetDeleteResult> {
     const engineId = request.getString("engineId");
     const assetId = request.getId();
-    const refresh = request.getRefresh();
-    const strict = request.getBoolean("strict");
 
-    await this.assetService.delete(request.getUser(), engineId, assetId, {
-      refresh,
-      strict,
-    });
+    await this.assetService.delete(engineId, assetId, request);
   }
 
   async search(request: KuzzleRequest): Promise<ApiAssetSearchResult> {
-    const engineId = request.getString("engineId");
-    const {
-      searchBody,
-      from,
-      size,
-      scrollTTL: scroll,
-    } = request.getSearchParams();
-    const lang = request.getLangParam();
-
-    const result = await this.assetService.search(engineId, searchBody, {
-      from,
-      lang,
-      scroll,
-      size,
-    });
-
-    return result;
+    return await this.assetService.search(
+      request.getString("engineId"),
+      request.getSearchParams(),
+      request
+    );
   }
 
   async getMeasures(