feat: add createDatabase method

Author: yanni97

arex-instrumentation-api/src/main/java/io/arex/inst/runtime/util/MockUtils.java

@@ -1,12 +1,16 @@
 package io.arex.inst.runtime.util;
 
+import com.fasterxml.jackson.databind.JsonNode;
 import io.arex.agent.bootstrap.model.ArexMocker;
 import io.arex.agent.bootstrap.model.MockCategoryType;
 import io.arex.agent.bootstrap.model.MockStrategyEnum;
 import io.arex.agent.bootstrap.model.Mocker;
 import io.arex.agent.bootstrap.model.Mocker.Target;
 import io.arex.agent.bootstrap.util.MapUtils;
 import io.arex.agent.bootstrap.util.StringUtil;
+import io.arex.agent.thirdparty.util.parse.sqlparse.SqlParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.constants.DbParseConstants;
+import io.arex.agent.thirdparty.util.parse.sqlparse.util.ParseUtil;
 import io.arex.inst.runtime.log.LogManager;
 import io.arex.inst.runtime.config.Config;
 import io.arex.inst.runtime.context.ArexContext;
@@ -16,10 +20,12 @@
 import io.arex.inst.runtime.serializer.Serializer;
 import io.arex.inst.runtime.service.DataService;
 import io.arex.inst.runtime.util.sizeof.AgentSizeOf;
+import net.sf.jsqlparser.JSQLParserException;
 
 public final class MockUtils {
 
     private static final String EMPTY_JSON = "{}";
+    private static final SqlParse sqlParse = new SqlParse();
 
     private MockUtils() {
     }
@@ -48,6 +54,25 @@ public static ArexMocker createDatabase(String method) {
         return create(MockCategoryType.DATABASE, method);
     }
 
+    public static ArexMocker createDatabase(String method, String sql, String dbName) {
+        StringBuilder operationName = new StringBuilder();
+        try {
+            String[] splitSql = sql.split(";");
+            for (String s : splitSql) {
+                JsonNode parse = sqlParse.parse(s);
+                if (parse != null) {
+                    String action = parse.get(DbParseConstants.ACTION).asText();
+                    String tableName = ParseUtil.parseTable(parse);
+                    operationName.append(dbName).append("-").append(tableName).append("-").append(action).append(";");
+                }
+            }
+        } catch (JSQLParserException e) {
+            LogManager.warn("createDatabase", "parse sql error", e);
+            operationName = new StringBuilder(method);
+        }
+        return create(MockCategoryType.DATABASE, operationName.toString());
+    }
+
     public static ArexMocker createRedis(String method) {
         return create(MockCategoryType.REDIS, method);
     }

arex-instrumentation-api/src/test/java/io/arex/inst/runtime/util/MockUtilsTest.java

@@ -178,6 +178,9 @@ void createMocker() {
 
         actualResult = MockUtils.createNettyProvider("query");
         assertEquals(MockCategoryType.NETTY_PROVIDER, actualResult.getCategoryType());
+
+        actualResult = MockUtils.createDatabase("query", "select * from test", "testDB");
+        assertEquals(MockCategoryType.DATABASE, actualResult.getCategoryType());
     }
 
     @Test

arex-instrumentation/database/arex-database-common/src/main/java/io/arex/inst/database/common/DatabaseExtractor.java

@@ -116,7 +116,7 @@ public MockResult replay(String serializer) {
     }
 
     private Mocker makeMocker(Object response, String serializer) {
-        Mocker mocker = MockUtils.createDatabase(this.methodName);
+        Mocker mocker = MockUtils.createDatabase(this.methodName, this.sql, this.dbName);
         mocker.getTargetRequest().setBody(this.sql);
         mocker.getTargetRequest().setAttribute("dbName", this.dbName);
         mocker.getTargetRequest().setAttribute("parameters", this.parameters);

arex-third-party/src/main/java/io/arex/agent/thirdparty/util/parse/sqlparse/util/ParseUtil.java

@@ -15,13 +15,16 @@
  */
 public class ParseUtil {
 
+    public static final String EMPTY = "";
 
-    public static Object parseTable(JsonNode result) {
+
+    public static String parseTable(JsonNode result) {
         switch (result.get(DbParseConstants.ACTION).asText()) {
             case DbParseConstants.SELECT:
                 return parseSelectTableName(result);
             case DbParseConstants.DELETE:
-                return parseDeleteTable(result);
+                List<String> tableNames = parseDeleteTable(result);
+                return tableNames.isEmpty() ? EMPTY : String.join(",", tableNames);
             case DbParseConstants.INSERT:
                 return parseInsertTable(result);
             case DbParseConstants.REPLACE:
@@ -42,7 +45,7 @@ public static String parseSelectTableName(JsonNode result) {
             }
             return jsonNode.asText();
         }
-        return null;
+        return EMPTY;
     }
 
     public static List<String> parseDeleteTable(JsonNode result) {
@@ -76,7 +79,7 @@ public static String parseCommonTableName(JsonNode result) {
             }
             return jsonNode.asText();
         }
-        return null;
+        return EMPTY;
     }
 
 }