chore: remove multienv from update

arextest · Jan 18, 2024 · f567760 · f567760
1 parent ccc5d8c
commit f567760
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/...n/java/com/arextest/config/repository/impl/ServiceCollectConfigurationRepositoryImpl.java b/...n/java/com/arextest/config/repository/impl/ServiceCollectConfigurationRepositoryImpl.java
@@ -17,7 +17,10 @@
 import com.mongodb.client.result.UpdateResult;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
+import java.util.Optional;
+import java.util.stream.Collectors;
 import javax.annotation.PostConstruct;
 import org.bson.conversions.Bson;
 
@@ -73,9 +76,7 @@ public boolean update(ServiceCollectConfiguration configuration) {
             RecordServiceConfigCollection.Fields.excludeServiceOperationSet,
             RecordServiceConfigCollection.Fields.timeMock,
             RecordServiceConfigCollection.Fields.extendField,
-            RecordServiceConfigCollection.Fields.serializeSkipInfoList,
-            Fields.multiEnvConfigs,
-            Fields.envTags
+            RecordServiceConfigCollection.Fields.serializeSkipInfoList
         ),
         Updates.set(RecordServiceConfigCollection.Fields.recordMachineCountLimit,
             configuration.getRecordMachineCountLimit() == null ? 1
@@ -114,7 +115,13 @@ public boolean removeByAppId(String appId) {
   @Override
   public boolean updateMultiEnvConfig(ServiceCollectConfiguration configuration) {
     Bson filter = Filters.eq(RecordServiceConfigCollection.Fields.appId, configuration.getAppId());
-    Bson update = Updates.set(Fields.multiEnvConfigs, configuration.getMultiEnvConfigs());
+
+    List<RecordServiceConfigCollection> configs = Optional.ofNullable(configuration.getMultiEnvConfigs())
+        .orElse(Collections.emptyList())
+        .stream().map(RecordServiceConfigMapper.INSTANCE::daoFromDto)
+        .collect(Collectors.toList());
+
+    Bson update = Updates.set(Fields.multiEnvConfigs, configs);
     return mongoCollection.updateOne(filter, update).getModifiedCount() > 0;
   }
 }