@ejfdelgado/ejflab-back 1.20.0 → 1.20.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ejfdelgado/ejflab-back",
-  "version": "1.20.0",
+  "version": "1.20.1",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/ejfdelgado/ejflab-back.git"
@@ -18,7 +18,7 @@
   "license": "ISC",
   "private": false,
   "dependencies": {
-    "@ejfdelgado/ejflab-common": "1.12.0",
+    "@ejfdelgado/ejflab-common": "1.12.1",
     "@google-cloud/compute": "^4.7.0",
     "@google-cloud/firestore": "^7.9.0",
     "@google-cloud/storage": "^7.11.3",

package/srv/MilvusSrv.mjs

@@ -1,5 +1,7 @@
 import { MilvusClient, DataType } from "@zilliz/milvus2-sdk-node";
 import { encode, decode } from "@msgpack/msgpack";
+import { General } from "./common/General.mjs";
+import { CommandMilvus } from "@ejfdelgado/ejflab-common/src/flowchart/steps/CommandMilvus.js";
 
 export class MilvusSrv {
     // MilvusSrv.checkErrors(res);
@@ -27,6 +29,10 @@ export class MilvusSrv {
             client
         }
     }
+    static async releaseConnection(client) {
+        console.log("Milvus disconnect");
+        await client.closeConnection();
+    }
     static async existsDatabase(client, name) {
         const res = await client.listDatabases();
         return res.db_names.indexOf(name) >= 0;
@@ -41,9 +47,15 @@ export class MilvusSrv {
         const res = await client.listDatabases();
         MilvusSrv.checkErrors(res);
         const databases = res.db_names;
+        const response = { dbs: [] };
         for (let i = 0; i < databases.length; i++) {
             const database = databases[i];
             console.log(`- Database: ${database}`);
+            const myDb = {
+                name: database,
+                collections: []
+            };
+            response.dbs.push(myDb);
             const resUseDatabase = await client.useDatabase({ db_name: database });
             MilvusSrv.checkErrors(resUseDatabase);
             const resListCollections = await client.listCollections();
@@ -52,8 +64,10 @@ export class MilvusSrv {
             for (let j = 0; j < data.length; j++) {
                 const collection = data[j];
                 console.log(`    - Collection: ${collection.name}`);
+                myDb.collections.push(collection.name);
             }
         }
+        return response;
     }
     static async dropDatabaseTemp(client) {
         const res = await client.listDatabases();
@@ -140,6 +154,42 @@ export class MilvusSrv {
         const res = await client.createCollection(myCopy);
         MilvusSrv.checkErrors(res);
     }
+    /*
+    action:
+    database.create
+    database.recreate
+    collection.create
+    introspect
+    database.destroy
+    database.destroy_temp
+    */
+    static async admin(req, res, next) {
+        //this.context.getSuperContext().getMilvusClient();
+        const context = {
+            getSuperContext: () => {
+                return {
+                    getMilvusClient: () => {
+                        return MilvusSrv;
+                    }
+                };
+            }
+        };
+        const id = '';
+        const commandName = 'milvus';
+        const argsTxt = '';
+        const command = new CommandMilvus(context, id, commandName, argsTxt);
+        const action = General.readParam(req, "action", null);
+        const db = General.readParam(req, "db", null);
+        const collection = General.readParam(req, "collection", null);
+        command.args = [action, db, collection];
+        const answer = await command.computation();
+        const response = {
+            args: command.args,
+            answer: answer,
+            status: "ok",
+        };
+        res.status(200).send(response);
+    }
     static async ping(req, res, next) {
         const { client } = MilvusSrv.connect();
         const databases = await client.listDatabases();