feat: add unit test

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/model/ArexMocker.java

@@ -16,18 +16,18 @@ public class ArexMocker implements Mocker {
     private long creationTime;
     private Mocker.Target targetRequest;
     private Mocker.Target targetResponse;
-    private transient boolean needMerge;
     private String operationName;
-    private Map<String, String> tags;
+    private transient boolean needMerge;
     private final transient AtomicBoolean matched = new AtomicBoolean(false);
-    /**
-     * replay match need
-     */
     private transient int fuzzyMatchKey;
-    /**
-     * replay match need
-     */
     private transient int accurateMatchKey;
+    private transient Map<Integer, Long> eigenMap;
+    private Map<String, String> tags;
+
+    // original compressed text for request
+    private transient String request;
+    // original compressed text for response
+    private transient String response;
 
     /**
      * The default constructor is for deserialization
@@ -103,6 +103,7 @@ public String getOperationName() {
         return this.operationName;
     }
 
+
     public void setId(String id) {
         this.id = id;
     }
@@ -178,4 +179,30 @@ public int getFuzzyMatchKey() {
     public void setFuzzyMatchKey(int fuzzyMatchKey) {
         this.fuzzyMatchKey = fuzzyMatchKey;
     }
+
+    @Override
+    public Map<Integer, Long> getEigenMap() {
+        return eigenMap;
+    }
+
+    @Override
+    public void setEigenMap(Map<Integer, Long> eigenMap) {
+        this.eigenMap = eigenMap;
+    }
+
+    public String getRequest() {
+        return request;
+    }
+
+    public void setRequest(String request) {
+        this.request = request;
+    }
+
+    public String getResponse() {
+        return response;
+    }
+
+    public void setResponse(String response) {
+        this.response = response;
+    }
 }

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/model/MockCategoryType.java

@@ -15,7 +15,7 @@ public class MockCategoryType implements Serializable {
     public static final MockCategoryType DYNAMIC_CLASS = createSkipComparison("DynamicClass");
     public static final MockCategoryType REDIS = createSkipComparison("Redis");
     public static final MockCategoryType MESSAGE_PRODUCER = createDependency("QMessageProducer");
-    public static final MockCategoryType MESSAGE_CONSUMER = createEntryPoint("QMessageConsumer");
+    public static final MockCategoryType MESSAGE_CONSUMER = create("QMessageConsumer",true,true);
     public static final MockCategoryType DUBBO_CONSUMER = createDependency("DubboConsumer");
     public static final MockCategoryType DUBBO_PROVIDER = createEntryPoint("DubboProvider");
     public static final MockCategoryType DUBBO_STREAM_PROVIDER = createDependency("DubboStreamProvider");

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/model/Mocker.java

@@ -7,6 +7,8 @@
 
 public interface Mocker extends Serializable {
 
+    void setAppId(String appId);
+
     String getAppId();
 
     String getReplayId();
@@ -33,10 +35,16 @@ public interface Mocker extends Serializable {
 
     String getOperationName();
 
+    void setOperationName(String operationName);
+
     Target getTargetRequest();
 
     Target getTargetResponse();
 
+    void setTargetRequest(Target targetRequest);
+
+    void setTargetResponse(Target targetResponse);
+
     public static class Target implements Serializable {
 
         private String body;
@@ -132,4 +140,16 @@ default String replayLogTitle() {
     int getFuzzyMatchKey();
 
     void setFuzzyMatchKey(int fuzzyMatchKey);
+
+    Map<Integer, Long> getEigenMap();
+
+    void setEigenMap(Map<Integer, Long> eigenMap);
+
+    String getRequest();
+
+    void setRequest(String request);
+
+    String getResponse();
+
+    void setResponse(String response);
 }

arex-agent-bootstrap/src/main/java/io/arex/agent/bootstrap/util/StringUtil.java

@@ -283,11 +283,11 @@ public static String[] splitByLastSeparator(String str, char separator) {
         return new String[]{str.substring(0, index), str.substring(index + 1)};
     }
 
-    public static int encodeAndHash(String str){
-        if (isBlank(str)) {
+    public static int encodeAndHash(String... str){
+        if (ArrayUtils.isEmpty(str)) {
             return 0;
         }
-        return Base64.getEncoder().encodeToString(str.getBytes(StandardCharsets.UTF_8)).hashCode();
+        return Base64.getEncoder().encodeToString(String.join("_", str).getBytes(StandardCharsets.UTF_8)).hashCode();
     }
 
     public static String replace(final String text, final String searchString, final String replacement) {

arex-agent-core/src/main/java/io/arex/agent/instrumentation/BaseAgentInstaller.java

@@ -112,7 +112,7 @@ private void initDependentComponents() {
         initDataCollector();
     }
     private void initDataCollector() {
-        List<DataCollector> collectorList = ServiceLoader.load(DataCollector.class, getClassLoader());
+        List<DataCollector> collectorList = ServiceLoader.load(DataCollector.class, Thread.currentThread().getContextClassLoader());
         DataService.setDataCollector(collectorList);
     }
 

arex-agent/pom.xml

@@ -33,6 +33,10 @@
             <groupId>${project.groupId}</groupId>
             <artifactId>arex-third-party</artifactId>
         </dependency>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>arex-compare</artifactId>
+        </dependency>
 
         <!--Agent instrumentation start-->
         <dependency>

arex-compare/pom.xml

@@ -0,0 +1,63 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>arex-agent-parent</artifactId>
+        <groupId>io.arex</groupId>
+        <version>${revision}</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>arex-compare</artifactId>
+
+    <dependencies>
+        <dependency>
+            <groupId>${project.groupId}</groupId>
+            <artifactId>arex-agent-bootstrap</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.fasterxml.jackson.core</groupId>
+            <artifactId>jackson-databind</artifactId>
+            <version>${jackson.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.slf4j</groupId>
+            <artifactId>slf4j-api</artifactId>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-shade-plugin</artifactId>
+                <version>${maven-shade-plugin.version}</version>
+                <executions>
+                    <execution>
+                        <phase>package</phase>
+                        <goals>
+                            <goal>shade</goal>
+                        </goals>
+                        <configuration>
+                            <relocations>
+                                <relocation>
+                                    <pattern>com.fasterxml.jackson</pattern>
+                                    <shadedPattern>io.arex.shaded.com.fasterxml.jackson</shadedPattern>
+                                </relocation>
+                            </relocations>
+                        </configuration>
+                    </execution>
+                </executions>
+                <configuration>
+                    <artifactSet>
+                        <includes>
+                            <include>com.fasterxml.jackson.core:**</include>
+                            <include>io.arex:arex-compare</include>
+                        </includes>
+                    </artifactSet>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+</project>

arex-compare/src/main/java/io/arex/agent/compare/eigen/EigenCalculateHandler.java

@@ -0,0 +1,41 @@
+package io.arex.agent.compare.eigen;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import io.arex.agent.bootstrap.model.MockCategoryType;
+import io.arex.agent.compare.handler.parse.JSONParse;
+import io.arex.agent.compare.handler.parse.ObjectParse;
+import io.arex.agent.compare.handler.parse.sqlparse.SqlParse;
+import io.arex.agent.compare.model.RulesConfig;
+import io.arex.agent.compare.model.eigen.EigenResult;
+
+import java.util.Objects;
+
+public class EigenCalculateHandler {
+    private static ObjectParse objectParse = new ObjectParse();
+    private static JSONParse jsonParse = new JSONParse();
+    private static SqlParse sqlParse = new SqlParse();
+
+    private static EigenMapCalculate eigenMapCalculate = new EigenMapCalculate();
+
+    public EigenResult doHandler(RulesConfig rulesConfig) {
+        Object obj;
+        try {
+            // if it is not json, it will return null
+            obj = objectParse.msgToObj(rulesConfig.getBaseMsg());
+
+            if (obj instanceof JsonNode) {
+                jsonParse.getJSONParseResult(obj, rulesConfig);
+            }
+
+            if (Objects.equals(rulesConfig.getCategoryType(), MockCategoryType.DATABASE.getName())
+                    && obj instanceof ObjectNode) {
+                sqlParse.sqlParse((ObjectNode) obj, rulesConfig.isNameToLower());
+            }
+        } catch (Throwable e) {
+            obj = null;
+        }
+
+        return eigenMapCalculate.doCalculate(obj, rulesConfig);
+    }
+}