modify userSetEnvType set in create collection method

akto-api-security · Jan 7, 2025 · b5e7d3d · b5e7d3d
1 parent 32a779d
commit b5e7d3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/utils/src/main/java/com/akto/data_actor/DbLayer.java b/libs/utils/src/main/java/com/akto/data_actor/DbLayer.java
@@ -376,7 +376,7 @@ public static void createCollectionSimpleForVpc(int vxlanId, String vpcId) {
                         Updates.set(ApiCollection.VXLAN_ID, vxlanId),
                         Updates.setOnInsert("startTs", Context.now()),
                         Updates.setOnInsert("urls", new HashSet<>()),
-                        Updates.set("userSetEnvType", vpcId)
+                        Updates.setOnInsert("userSetEnvType", vpcId)
                 ),
                 updateOptions
         );
@@ -405,7 +405,7 @@ public static void createCollectionForHostAndVpc(String host, int id, String vpc
             Updates.setOnInsert("_id", id),
             Updates.setOnInsert("startTs", Context.now()),
             Updates.setOnInsert("urls", new HashSet<>()),
-            Updates.set("userSetEnvType", vpcId)
+            Updates.setOnInsert("userSetEnvType", vpcId)
         );
 
         ApiCollectionsDao.instance.getMCollection().findOneAndUpdate(Filters.eq(ApiCollection.HOST_NAME, host), updates, updateOptions);