fix: map properties error

YongwuHe · YongwuHe · commit 6628de036189 · 2024-10-25T16:26:55.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
@@ -35,11 +35,11 @@ public class ArexMocker implements Mocker {
     public ArexMocker() {
     }
 
-    public ArexMocker(MockCategoryType categoryType) {
+    public ArexMocker(MockCategoryType categoryType, Map<String, String> arexMockerTags) {
         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());
+        this.tags = arexMockerTags == null ? Collections.emptyMap() : arexMockerTags;
     }
 
     /**
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 Map<String, String> arexMockerTags;
 
     Config(boolean enableDebug, String serviceName, List<DynamicClassEntity> dynamicClassList,
         Map<String, String> properties,
@@ -107,6 +108,14 @@ private void buildDynamicClassInfo() {
         this.dynamicAbstractClassList = list.toArray(StringUtil.EMPTY_STRING_ARRAY);
     }
 
+    public void setArexMockerTags(Map<String, String> arexMockerTags) {
+        this.arexMockerTags = arexMockerTags;
+    }
+
+    public Map<String, String> getArexMockerTags() {
+        return arexMockerTags;
+    }
+
     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
@@ -77,7 +77,7 @@ public static ArexMocker createNettyProvider(String pattern) {
     }
 
     public static ArexMocker create(MockCategoryType categoryType, String operationName) {
-        ArexMocker mocker = new ArexMocker(categoryType);
+        ArexMocker mocker = new ArexMocker(categoryType, Config.get().getArexMockerTags());
         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
@@ -7,6 +7,7 @@
 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;
 
@@ -20,6 +21,7 @@
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
 
+import java.util.Map;
 import java.util.function.Predicate;
 import java.util.stream.Stream;
 
@@ -147,4 +149,16 @@ void buildCoveragePackages() {
         System.clearProperty(ArexConstants.SPRING_SCAN_PACKAGES);
 
     }
+
+    @Test
+    void arexMockerTags() {
+        ConfigBuilder configBuilder = ConfigBuilder.create("mock");
+        configBuilder.build();
+        Config.get().setArexMockerTags(null);
+        assertNull(Config.get().getArexMockerTags());
+        Map<String, String> tags = new HashMap<>();
+        tags.put("key1", "value1");
+        Config.get().setArexMockerTags(tags);
+        assertEquals("value1", Config.get().getArexMockerTags().get("key1"));
+    }
 }
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
@@ -36,7 +36,7 @@ static void tearDown() {
 
     @Test
     void match() {
-        ArexMocker requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        ArexMocker requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         requestMocker.setOperationName("mock");
         requestMocker.setTargetRequest(new Mocker.Target());
         requestMocker.setTargetResponse(new Mocker.Target());
@@ -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
@@ -49,7 +49,7 @@ static void setUp() {
         Mockito.mockStatic(ContextManager.class);
         Mockito.mockStatic(Config.class);
         Mockito.when(Config.get()).thenReturn(Mockito.mock(Config.class));
-        requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         requestMocker.setOperationName("mock");
         requestMocker.setTargetRequest(new Mocker.Target());
         requestMocker.setTargetResponse(new Mocker.Target());
@@ -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/MockUtilsTest.java b/arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/MockUtilsTest.java
@@ -199,7 +199,7 @@ void methodSignatureHash() {
 
     @Test
     void methodRequestTypeHash() {
-        ArexMocker mocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        ArexMocker mocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         mocker.setTargetRequest(new Mocker.Target());
         mocker.getTargetRequest().setBody("mock");
         assertTrue(MockUtils.methodRequestTypeHash(mocker) > 0);
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
@@ -34,7 +34,7 @@ static void setUp() {
         Mockito.mockStatic(ContextManager.class);
         Mockito.mockStatic(Config.class);
         Mockito.when(Config.get()).thenReturn(Mockito.mock(Config.class));
-        requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        requestMocker = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         requestMocker.setOperationName("mock");
         requestMocker.setTargetRequest(new Mocker.Target());
         requestMocker.setTargetResponse(new Mocker.Target());
@@ -56,15 +56,15 @@ void replayAllMocker(Runnable mocker) {
     }
 
     static Stream<Arguments> replayAllMockerCase() {
-        ArexMocker recordMocker1 = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        ArexMocker recordMocker1 = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         recordMocker1.setOperationName(ArexConstants.MERGE_RECORD_NAME);
         recordMocker1.setRecordId("mock");
         recordMocker1.setReplayId("mock");
         recordMocker1.setTargetRequest(new Mocker.Target());
         recordMocker1.setTargetResponse(new Mocker.Target());
         recordMocker1.getTargetResponse().setBody("mock");
         recordMocker1.setCreationTime(System.currentTimeMillis());
-        ArexMocker recordMocker2 = new ArexMocker(MockCategoryType.DYNAMIC_CLASS);
+        ArexMocker recordMocker2 = new ArexMocker(MockCategoryType.DYNAMIC_CLASS, Collections.emptyMap());
         recordMocker2.setOperationName(ArexConstants.MERGE_RECORD_NAME);
         recordMocker2.setRecordId("mock");
         recordMocker2.setReplayId("mock");
@@ -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
@@ -10,6 +10,7 @@
 import io.arex.foundation.util.httpclient.AsyncHttpClientUtil;
 import io.arex.foundation.util.NetUtils;
 import io.arex.agent.bootstrap.util.StringUtil;
+import io.arex.inst.runtime.config.Config;
 
 import java.util.Collections;
 import java.util.HashMap;
@@ -28,6 +29,7 @@ public class ConfigService {
 
     private static final AgentLogger LOGGER = AgentLoggerFactory.getAgentLogger(ConfigService.class);
     private static final Map<String, String> TAGS_PROPERTIES = new HashMap<>();
+    private static final Map<String, String> AREX_MOCKER_TAGS = new HashMap<>();
 
     public static final ConfigService INSTANCE = new ConfigService();
     private static final String TAGS_PREFIX = "arex.tags.";
@@ -84,6 +86,7 @@ public void loadAgentConfig() {
                 return;
             }
             ConfigManager.INSTANCE.updateConfigFromService(configResponse.getBody());
+            Config.get().setArexMockerTags(Collections.unmodifiableMap(AREX_MOCKER_TAGS));
         } catch (Throwable e) {
             LOGGER.warn("[AREX] Load agent config error, pause recording. exception message: {}", e.getMessage(), e);
             ConfigManager.INSTANCE.setConfigInvalid();
@@ -129,24 +132,22 @@ AgentStatusEnum getAgentStatus() {
     public Map<String, String> getSystemProperties() {
         Properties properties = System.getProperties();
         Map<String, String> map = MapUtils.newHashMapWithExpectedSize(properties.size());
-        Map<String, String> mockerTags = new HashMap<>();
         for (Map.Entry<Object, Object> entry : properties.entrySet()) {
             String key = String.valueOf(entry.getKey());
             String value = String.valueOf(entry.getValue());
             map.put(key, value);
-            buildTags(mockerTags, key, value);
+            buildTags(key, value);
         }
-        properties.put(ConfigConstants.MOCKER_TAGS, Collections.unmodifiableMap(mockerTags));
         return map;
     }
 
     /**
      * ex: -Darex.tags.xxx=xxx
      */
-    private void buildTags(Map<String, String> mockerTags, String key, String value) {
+    private void buildTags(String key, String value) {
         if (StringUtil.startWith(key, TAGS_PREFIX)) {
             TAGS_PROPERTIES.put(key, value);
-            mockerTags.put(key.substring(TAGS_PREFIX.length()), value);
+            AREX_MOCKER_TAGS.put(key.substring(TAGS_PREFIX.length()), value);
         }
     }
 
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
`@@ -35,11 +35,11 @@ public class ArexMocker implements Mocker {`
`35`	`35`	`public ArexMocker() {`
`36`	`36`	`}`
`37`	`37`
`38`		`- public ArexMocker(MockCategoryType categoryType) {`
	`38`	`+ public ArexMocker(MockCategoryType categoryType, Map<String, String> arexMockerTags) {`
`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());`
	`42`	`+ this.tags = arexMockerTags == null ? Collections.emptyMap() : arexMockerTags;`
`43`	`43`	`}`
`44`	`44`
`45`	`45`	`/**`
Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ public static ArexMocker createNettyProvider(String pattern) {`
`77`	`77`	`}`
`78`	`78`
`79`	`79`	`public static ArexMocker create(MockCategoryType categoryType, String operationName) {`
`80`		`- ArexMocker mocker = new ArexMocker(categoryType);`
	`80`	`+ ArexMocker mocker = new ArexMocker(categoryType, Config.get().getArexMockerTags());`
`81`	`81`	`long createTime = System.currentTimeMillis();`
`82`	`82`	`ArexContext context = ContextManager.currentContext();`
`83`	`83`	`if (context != null) {`