fix: map properties error (#580)

YongwuHe · web-flow · commit 9d6488feab1f · 2024-10-28T14:14:42.000+08:00
diff --git a/arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/model/ArexMocker.java b/arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/model/ArexMocker.java
@@ -39,7 +39,6 @@ public ArexMocker(MockCategoryType categoryType) {
         this.categoryType = categoryType;
         this.appId = System.getProperty(ConfigConstants.SERVICE_NAME);
         this.recordVersion = System.getProperty(ConfigConstants.AGENT_VERSION);
-        this.tags = (Map<String, String>) System.getProperties().getOrDefault(ConfigConstants.MOCKER_TAGS, Collections.emptyMap());
     }
 
     /**
diff --git a/arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/util/StringUtil.java b/arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/util/StringUtil.java
@@ -106,6 +106,20 @@ private static void indent(StringBuilder sb, int indent) {
         }
     }
 
+    public static String mapToString(Map<String, String> map) {
+        if (MapUtils.isEmpty(map)) {
+            return StringUtil.EMPTY;
+        }
+        StringBuilder stringBuilder = new StringBuilder();
+        for (Map.Entry<String, String> entry : map.entrySet()) {
+            stringBuilder.append(entry.getKey())
+                    .append("=")
+                    .append(entry.getValue())
+                    .append(";");
+        }
+        return stringBuilder.toString();
+    }
+
     public static Map<String, String> asMap(String content) {
         if (isEmpty(content)) {
             return Collections.emptyMap();
diff --git a/arex-agent-bootstrap/src/test/java/io/arex/agent/bootstrap/util/StringUtilTest.java b/arex-agent-bootstrap/src/test/java/io/arex/agent/bootstrap/util/StringUtilTest.java
@@ -425,4 +425,12 @@ void splitByLastSeparator() {
         actualResult = StringUtil.splitByLastSeparator(val, ',');
         assertArrayEquals(new String[] {"x,,y", "z"}, actualResult);
     }
+
+    @Test
+    void mapToString() {
+        Map<String, String> map = new HashMap<>();
+        map.put("a", "b");
+        map.put("c", "d");
+        assertEquals("a=b;c=d;", StringUtil.mapToString(map));
+    }
 }
diff --git a/arex-instrumentation-api/src/main/java/io/arex/inst/runtime/config/Config.java b/arex-instrumentation-api/src/main/java/io/arex/inst/runtime/config/Config.java
@@ -47,6 +47,7 @@ public static Config get() {
     private final String recordVersion;
     private final Set<String> includeServiceOperations;
     private final Set<String> coveragePackages = new ConcurrentHashSet<>();
+    private final Map<String, String> mockerTags;
 
     Config(boolean enableDebug, String serviceName, List<DynamicClassEntity> dynamicClassList,
         Map<String, String> properties,
@@ -60,6 +61,7 @@ public static Config get() {
         this.recordRate = recordRate;
         this.recordVersion = properties.get("arex.agent.version");
         this.includeServiceOperations = StringUtil.splitToSet(properties.get("includeServiceOperations"), SEPARATOR);
+        this.mockerTags = StringUtil.asMap(System.getProperty(ConfigConstants.MOCKER_TAGS));
         buildCoveragePackages(properties);
         buildDynamicClassInfo();
     }
@@ -107,6 +109,10 @@ private void buildDynamicClassInfo() {
         this.dynamicAbstractClassList = list.toArray(StringUtil.EMPTY_STRING_ARRAY);
     }
 
+    public Map<String, String> getMockerTags() {
+        return mockerTags;
+    }
+
     public Set<String> getCoveragePackages() {
         return coveragePackages;
     }
diff --git a/arex-instrumentation-api/src/main/java/io/arex/inst/runtime/util/MockUtils.java b/arex-instrumentation-api/src/main/java/io/arex/inst/runtime/util/MockUtils.java
@@ -78,6 +78,7 @@ public static ArexMocker createNettyProvider(String pattern) {
 
     public static ArexMocker create(MockCategoryType categoryType, String operationName) {
         ArexMocker mocker = new ArexMocker(categoryType);
+        mocker.setTags(Config.get().getMockerTags());
         long createTime = System.currentTimeMillis();
         ArexContext context = ContextManager.currentContext();
         if (context != null) {
diff --git a/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/config/ConfigTest.java b/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/config/ConfigTest.java
@@ -2,11 +2,14 @@
 
 import io.arex.agent.bootstrap.constants.ConfigConstants;
 import io.arex.agent.bootstrap.util.ConcurrentHashSet;
+import io.arex.agent.bootstrap.util.MapUtils;
+import io.arex.agent.bootstrap.util.StringUtil;
 import io.arex.inst.runtime.context.RecordLimiter;
 import io.arex.inst.runtime.listener.EventProcessor;
 import io.arex.inst.runtime.model.ArexConstants;
 import io.arex.inst.runtime.model.DynamicClassEntity;
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 
@@ -17,9 +20,9 @@
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.MethodSource;
-import org.mockito.MockedStatic;
 import org.mockito.Mockito;
 
+import java.util.Map;
 import java.util.function.Predicate;
 import java.util.stream.Stream;
 
@@ -147,4 +150,18 @@ void buildCoveragePackages() {
         System.clearProperty(ArexConstants.SPRING_SCAN_PACKAGES);
 
     }
+
+    @Test
+    void arexMockerTags() {
+        ConfigBuilder configBuilder = ConfigBuilder.create("mock");
+        configBuilder.build();
+        assertTrue(MapUtils.isEmpty(Config.get().getMockerTags()));
+        Map<String, String> tags = new HashMap<>();
+        tags.put("key1", "value1");
+        tags.put("key2", "value2");
+        System.setProperty(ConfigConstants.MOCKER_TAGS, StringUtil.mapToString(tags));
+        configBuilder.build();
+        assertEquals("value1", Config.get().getMockerTags().get("key1"));
+        assertEquals("value2", Config.get().getMockerTags().get("key2"));
+    }
 }
diff --git a/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/match/ReplayMatcherTest.java b/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/match/ReplayMatcherTest.java
@@ -57,4 +57,4 @@ void match() {
         Mockito.when(MockUtils.methodRequestTypeHash(requestMocker)).thenReturn(2);
         assertNull(ReplayMatcher.match(requestMocker, MockStrategyEnum.FIND_LAST));
     }
-}
+}
diff --git a/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/MergeRecordUtilTest.java b/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/MergeRecordUtilTest.java
@@ -141,4 +141,4 @@ static Stream<Arguments> recordRemainCase() {
                 arguments(contextSupplier2.get(), asserts2)
         );
     }
-}
+}
diff --git a/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/ReplayUtilTest.java b/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/ReplayUtilTest.java
@@ -99,4 +99,4 @@ static Stream<Arguments> replayAllMockerCase() {
                 arguments(mocker2)
         );
     }
-}
+}
diff --git a/arex-instrumentation-foundation/src/main/java/io/arex/foundation/services/ConfigService.java b/arex-instrumentation-foundation/src/main/java/io/arex/foundation/services/ConfigService.java
@@ -11,7 +11,6 @@
 import io.arex.foundation.util.NetUtils;
 import io.arex.agent.bootstrap.util.StringUtil;
 
-import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Objects;
@@ -136,7 +135,7 @@ public Map<String, String> getSystemProperties() {
             map.put(key, value);
             buildTags(mockerTags, key, value);
         }
-        properties.put(ConfigConstants.MOCKER_TAGS, Collections.unmodifiableMap(mockerTags));
+        System.setProperty(ConfigConstants.MOCKER_TAGS, StringUtil.mapToString(mockerTags));
         return map;
     }
 
diff --git a/arex-instrumentation/httpclient/arex-httpclient-feign/src/test/java/io/arex/inst/httpclient/feign/FeignClientInstrumentationTest.java b/arex-instrumentation/httpclient/arex-httpclient-feign/src/test/java/io/arex/inst/httpclient/feign/FeignClientInstrumentationTest.java
@@ -6,6 +6,7 @@
 import feign.Response;
 import io.arex.agent.bootstrap.model.MockResult;
 import io.arex.inst.httpclient.common.HttpClientExtractor;
+import io.arex.inst.runtime.config.ConfigBuilder;
 import io.arex.inst.runtime.context.ContextManager;
 import io.arex.inst.runtime.context.RepeatedCollectManager;
 import io.arex.inst.runtime.util.IgnoreUtils;
@@ -24,6 +25,7 @@ static void setUp() {
         Mockito.mockStatic(IgnoreUtils.class);
         Mockito.mockStatic(ContextManager.class);
         Mockito.mockStatic(RepeatedCollectManager.class);
+        ConfigBuilder.create("test").build();
     }
 
     @AfterAll

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,6 @@ public ArexMocker(MockCategoryType categoryType) {`
`39`	`39`	`this.categoryType = categoryType;`
`40`	`40`	`this.appId = System.getProperty(ConfigConstants.SERVICE_NAME);`
`41`	`41`	`this.recordVersion = System.getProperty(ConfigConstants.AGENT_VERSION);`
`42`		`- this.tags = (Map<String, String>) System.getProperties().getOrDefault(ConfigConstants.MOCKER_TAGS, Collections.emptyMap());`
`43`	`42`	`}`
`44`	`43`
`45`	`44`	`/**`
Original file line number	Diff line number	Diff line change
`@@ -425,4 +425,12 @@ void splitByLastSeparator() {`
`425`	`425`	`actualResult = StringUtil.splitByLastSeparator(val, ',');`
`426`	`426`	`assertArrayEquals(new String[] {"x,,y", "z"}, actualResult);`
`427`	`427`	`}`
	`428`	`+`
	`429`	`+ @Test`
	`430`	`+ void mapToString() {`
	`431`	`+ Map<String, String> map = new HashMap<>();`
	`432`	`+ map.put("a", "b");`
	`433`	`+ map.put("c", "d");`
	`434`	`+ assertEquals("a=b;c=d;", StringUtil.mapToString(map));`
	`435`	`+ }`
`428`	`436`	`}`
Original file line number	Diff line number	Diff line change
`@@ -57,4 +57,4 @@ void match() {`
`57`	`57`	`Mockito.when(MockUtils.methodRequestTypeHash(requestMocker)).thenReturn(2);`
`58`	`58`	`assertNull(ReplayMatcher.match(requestMocker, MockStrategyEnum.FIND_LAST));`
`59`	`59`	`}`
`60`		`-}`
	`60`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -141,4 +141,4 @@ static Stream<Arguments> recordRemainCase() {`
`141`	`141`	`arguments(contextSupplier2.get(), asserts2)`
`142`	`142`	`);`
`143`	`143`	`}`
`144`		`-}`
	`144`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -99,4 +99,4 @@ static Stream<Arguments> replayAllMockerCase() {`
`99`	`99`	`arguments(mocker2)`
`100`	`100`	`);`
`101`	`101`	`}`
`102`		`-}`
	`102`	`+}`
Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,6 @@`
`11`	`11`	`import io.arex.foundation.util.NetUtils;`
`12`	`12`	`import io.arex.agent.bootstrap.util.StringUtil;`
`13`	`13`
`14`		`-import java.util.Collections;`
`15`	`14`	`import java.util.HashMap;`
`16`	`15`	`import java.util.Map;`
`17`	`16`	`import java.util.Objects;`
`@@ -136,7 +135,7 @@ public Map<String, String> getSystemProperties() {`
`136`	`135`	`map.put(key, value);`
`137`	`136`	`buildTags(mockerTags, key, value);`
`138`	`137`	`}`
`139`		`- properties.put(ConfigConstants.MOCKER_TAGS, Collections.unmodifiableMap(mockerTags));`
	`138`	`+ System.setProperty(ConfigConstants.MOCKER_TAGS, StringUtil.mapToString(mockerTags));`
`140`	`139`	`return map;`
`141`	`140`	`}`
`142`	`141`