update checks for collection creation

boomerang-io · Nov 9, 2021 · 6a63bd8 · 6a63bd8
1 parent 2eb58dd
commit 6a63bd8
Showing 1 changed file with 20 additions and 10 deletions.
diff --git a/src/main/java/net/boomerangplatform/migration/changesets/flow/FlowDatabaseChangeLog.java b/src/main/java/net/boomerangplatform/migration/changesets/flow/FlowDatabaseChangeLog.java
@@ -42,24 +42,34 @@ public void initialSetup(MongoDatabase db) throws IOException {
 
     logger.info("Running change log: #1 - Creating Collections for Boomerang Flow");
 
-    db.createCollection(collectionPrefix + "task_templates");
-    db.createCollection(collectionPrefix + "teams");
+    MongoCollection<Document> collection = db.getCollection(collectionPrefix + "task_templates");
+    if (collection == null) {
+      db.createCollection(collectionPrefix + "task_templates");
+    }
 
+    collection = db.getCollection(collectionPrefix + "teams");
+    if (collection == null) {
+      db.createCollection(collectionPrefix + "teams");
+    }
 
-    try {
+    collection = db.getCollection(collectionPrefix + "workflows");
+    if (collection == null) {
       db.createCollection(collectionPrefix + "workflows");
-    } catch (IllegalArgumentException e) {
-      e.printStackTrace();
     }
-    db.createCollection(collectionPrefix + "workflows_activity");
-    db.createCollection(collectionPrefix + "workflows_activity_task");
 
+    collection = db.getCollection(collectionPrefix + "workflows_activity");
+    if (collection == null) {
+      db.createCollection(collectionPrefix + "workflows_activity");
+    }
 
+    collection = db.getCollection(collectionPrefix + "workflows_activity_task");
+    if (collection == null) {
+      db.createCollection(collectionPrefix + "workflows_activity_task");
+    }
 
-    try {
+    collection = db.getCollection(collectionPrefix + "workflows_revisions");
+    if (collection == null) {
       db.createCollection(collectionPrefix + "workflows_revisions");
-    } catch (IllegalArgumentException e) {
-      e.printStackTrace();
     }
   }