Recursion to ensure that collection exists (#26)

shotit · Apr 25, 2023 · d69ef93 · d69ef93
2 parents d9190cd + 6ce3ecd
commit d69ef93
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/loader.js b/loader.js
@@ -57,11 +57,21 @@ const initializeMilvusCollection = async () => {
     ],
   };
 
-  await milvusClient.collectionManager.releaseCollection({ collection_name: "shotit" });
-
-  await milvusClient.collectionManager.createCollection(params);
+  const fallBack = async () => {
+    try {
+      await milvusClient.collectionManager.releaseCollection({ collection_name: "shotit" });
+      await milvusClient.collectionManager.createCollection(params);
+      console.log('collection_name: "shotit" ensured');
+      milvusClient.closeConnection();
+    } catch (error) {
+      console.log(error);
+      console.log("initializeMilvusCollection reconnecting in 3 seconds");
+      await new Promise((resolve) => setTimeout(resolve, 1000 * 3));
+      await fallBack();
+    }
+  };
 
-  milvusClient.closeConnection();
+  await fallBack();
 };
 
 /**