@ejfdelgado/ejflab-back 1.12.0 → 1.14.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ejfdelgado/ejflab-back",
-  "version": "1.12.0",
+  "version": "1.14.0",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/ejfdelgado/ejflab-back.git"

package/srv/MilvusSrv.mjs

@@ -1,4 +1,5 @@
 import { MilvusClient, DataType } from "@zilliz/milvus2-sdk-node";
+import { encode, decode } from "@msgpack/msgpack";
 
 export class MilvusSrv {
     // MilvusSrv.checkErrors(res);
@@ -149,4 +150,86 @@ export class MilvusSrv {
         };
         res.status(200).send(response);
     }
+    static async insert(req, res, next) {
+        console.log("Milvus insert...");
+        const { client } = MilvusSrv.connect();
+        let code = 200;
+        const response = {
+            status: "ok"
+        };
+        try {
+            // get the request
+            const buffer = req.body;
+            const decoded = decode(buffer);
+            const {
+                db_name,
+                collection_name,
+                data
+            } = decoded;
+            await MilvusSrv.useDatabase(client, db_name, false);
+            await client.insert({
+                collection_name: collection_name,
+                data: data
+            });
+            console.log("Milvus insert... OK");
+        } catch (err) {
+            console.log(err);
+            code = 500;
+            response.status = "error";
+            response.message = err.message;
+        } finally {
+            await client.closeConnection();
+            res.status(code).send(response);
+        }
+    }
+    static async search(req, res, next) {
+        const { client } = MilvusSrv.connect();
+        let code = 200;
+        const response = {
+            status: "ok"
+        };
+        try {
+            // get the request
+            const buffer = req.body;
+            const decoded = decode(buffer);
+            // Connect to database
+            const {
+                db_name,
+                collection_name,
+                embeed,
+                paging
+            } = decoded;
+            console.log(`Using database ${db_name}`);
+            await MilvusSrv.useDatabase(client, db_name, false);
+            const search_params = {
+                "metric_type": "IP",
+                "topk": paging['limit'],
+                "params": JSON.stringify({ nprobe: 1024 }),
+            };
+            const results = await client.search({
+                collection_name: collection_name,
+                data: [embeed],
+                limit: paging['limit'],
+                offset: paging['offset'],
+                consistency_level: "Strong",
+                search_params: search_params,
+                output_fields: ['id', 'document_id', 'face_path', 'millis', 'x1', 'y1', 'x2', 'y2', 'ref_id']
+            });
+            response.results = results.results.map((entity) => {
+                return {
+                    id: entity.id,
+                    distance: 0,
+                    entity
+                };
+            });
+        } catch (err) {
+            console.log(err);
+            code = 500;
+            response.status = "error";
+            response.message = err.message;
+        } finally {
+            await client.closeConnection();
+            res.status(code).send(response);
+        }
+    }
 }

package/srv/SocketIOCall.mjs

@@ -203,7 +203,7 @@ export class SocketIOCall {
     }
 
     static getCustomHeaders(socket) {
-        const headersKey = ["room", "uuid"];
+        const headersKey = ["room", "uuid", "model"];
         const headers = socket.handshake.headers;
         const response = {};
         headersKey.forEach((key) => {
@@ -216,6 +216,23 @@ export class SocketIOCall {
         await SocketIOCall.registerSocket(socket);
         // Le digo a los que están en el room quién llegó
         const headers = SocketIOCall.getCustomHeaders(socket);
+        // Update model if provided
+        let defaultModel;
+        try {
+            defaultModel = headers.model;
+            const room = headers.room;
+            if (defaultModel && room) {
+                const defaultModelJson = JSON.parse(defaultModel);
+                const dataRoom = SocketIOCall.getRoomLiveTupleModel(room);
+                if (dataRoom) {
+                    SimpleObj.recreate(dataRoom, `model.data.people.${socket.id}`, defaultModelJson);
+                    dataRoom.builder.trackDifferences(dataRoom.model, [], null, ["data", "data.people", `data.people.${socket.id}`]);
+                }
+            }
+        } catch (err) {
+            console.log(err);
+            console.log(`Can't create default model. Provided ${defaultModel}`);
+        }
         SocketIOCall.io.to(headers.room).emit("updateUserList", {
             socketIds: SocketIOCall.mapSockets[headers.room]
         });