Browse Source

add milvus clients

Signed-off-by: Gitea <zizhao.chen@zilliz.com>
Gitea 3 years ago
parent
commit
973927836b
1 changed files with 17 additions and 1 deletions
  1. 17 1
      express/src/milvus/milvus.service.ts

+ 17 - 1
express/src/milvus/milvus.service.ts

@@ -7,9 +7,11 @@ import {
 export class MilvusService {
   private milvusAddress: string;
   private milvusClient: MilvusClient;
+  private milvusClients: { [x: string]: MilvusClient };
 
   constructor() {
-    this.milvusAddress = "";
+    this.milvusAddress = '';
+    this.milvusClients = {};
   }
 
   get milvusAddressGetter() {
@@ -55,12 +57,26 @@ export class MilvusService {
         collection_name: "not_exist",
       });
       this.milvusAddress = address;
+      this.milvusClients = {
+        address: this.milvusClient,
+      };
       return { address: this.milvusAddress };
     } catch (error) {
       throw new Error("Connect milvus failed, check your milvus address.");
     }
   }
 
+  async logout(address: string) {
+    if (this.milvusClients[address]) {
+      delete this.milvusClients[address];
+    }
+    if (this.milvusAddress === address) {
+      this.milvusAddress = '';
+    }
+
+    return { logout: true };
+  }
+
   async checkConnect(address: string) {
     if (address !== this.milvusAddress) {
       return { connected: false };