feat: support sqlParse

Author: yanni97

arex-instrumentation-api/pom.xml

@@ -32,7 +32,6 @@
             <groupId>com.fasterxml.jackson.core</groupId>
             <artifactId>jackson-databind</artifactId>
             <version>${jackson.version}</version>
-            <scope>test</scope>
         </dependency>
         <dependency>
             <groupId>com.google.code.gson</groupId>

arex-third-party/pom.xml

@@ -7,6 +7,18 @@
         <groupId>io.arex</groupId>
         <version>${revision}</version>
     </parent>
+    <dependencies>
+        <dependency>
+            <groupId>com.fasterxml.jackson.core</groupId>
+            <artifactId>jackson-databind</artifactId>
+            <version>${jackson.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>com.github.jsqlparser</groupId>
+            <artifactId>jsqlparser</artifactId>
+            <version>${jsqlparser.version}</version>
+        </dependency>
+    </dependencies>
     <modelVersion>4.0.0</modelVersion>
 
     <artifactId>arex-third-party</artifactId>

arex-third-party/src/main/java/io/arex/agent/thirdparty/util/JacksonHelperUtil.java

@@ -0,0 +1,23 @@
+package io.arex.agent.thirdparty.util;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class JacksonHelperUtil {
+
+    public static ObjectMapper objectMapper = new ObjectMapper();
+
+    public static ObjectNode getObjectNode() {
+        return objectMapper.createObjectNode();
+    }
+
+    public static ArrayNode getArrayNode() {
+        return objectMapper.createArrayNode();
+    }
+}

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

@@ -0,0 +1,13 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse;
+
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public interface Parse<T> {
+    ObjectNode parse(T parseObj);
+}

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

@@ -0,0 +1,26 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import io.arex.agent.thirdparty.util.parse.sqlparse.action.ActionFactory;
+import net.sf.jsqlparser.JSQLParserException;
+import net.sf.jsqlparser.parser.CCJSqlParserUtil;
+import net.sf.jsqlparser.statement.Statement;
+
+/**
+ * sql parse
+ *
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class SqlParse {
+    public JsonNode parse(String sql) throws JSQLParserException {
+        if (sql == null || sql.isEmpty()) {
+            return null;
+        }
+
+        Statement statement = CCJSqlParserUtil.parse(sql);
+        Parse matchParse = ActionFactory.selectParse(statement);
+        return matchParse.parse(statement);
+    }
+}

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

@@ -0,0 +1,37 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse.action;
+
+import io.arex.agent.thirdparty.util.parse.sqlparse.Parse;
+import net.sf.jsqlparser.statement.delete.Delete;
+import net.sf.jsqlparser.statement.execute.Execute;
+import net.sf.jsqlparser.statement.insert.Insert;
+import net.sf.jsqlparser.statement.replace.Replace;
+import net.sf.jsqlparser.statement.select.Select;
+import net.sf.jsqlparser.statement.update.Update;
+
+import net.sf.jsqlparser.statement.Statement;
+
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class ActionFactory {
+    public static Parse selectParse(Statement statement) {
+        if (statement instanceof Select) {
+            return new SelectParse();
+        } else if (statement instanceof Execute) {
+            return new ExecuteParse();
+        } else if (statement instanceof Delete) {
+            return new DeleteParse();
+        } else if (statement instanceof Insert) {
+            return new InsertParse();
+        } else if (statement instanceof Replace) {
+            return new ReplaceParse();
+        } else if (statement instanceof Update) {
+            return new UpdateParse();
+        } else {
+            throw new UnsupportedOperationException("not support");
+        }
+    }
+}

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

@@ -0,0 +1,47 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse.action;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import io.arex.agent.thirdparty.util.JacksonHelperUtil;
+import io.arex.agent.thirdparty.util.parse.sqlparse.Parse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.CommonParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.ExpressionParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.JoinParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.OrderByParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.TableParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.constants.DbParseConstants;
+import net.sf.jsqlparser.statement.delete.Delete;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class DeleteParse implements Parse<Delete> {
+
+    private final ObjectNode sqlObjectNode;
+    public DeleteParse() {
+        sqlObjectNode = JacksonHelperUtil.getObjectNode();
+        // action parse
+        sqlObjectNode.put(DbParseConstants.ACTION, DbParseConstants.DELETE);
+    }
+    @Override
+    public ObjectNode parse(Delete parseObj) {
+
+        // tables parse
+        TableParse.parseDelTable(parseObj.getTables(), parseObj.getTable(), sqlObjectNode);
+
+        // join parse
+        JoinParse.parse(parseObj.getJoins(), sqlObjectNode);
+
+        // where parse
+        ExpressionParse.parseWhere(parseObj.getWhere(), sqlObjectNode);
+
+        // orderBy parse
+        OrderByParse.parse(parseObj.getOrderByElements(), sqlObjectNode);
+
+        // limit parse
+        CommonParse.parseLimit(parseObj.getLimit(), sqlObjectNode);
+
+        return sqlObjectNode;
+    }
+}

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

@@ -0,0 +1,36 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse.action;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import io.arex.agent.thirdparty.util.JacksonHelperUtil;
+import io.arex.agent.thirdparty.util.parse.sqlparse.Parse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.CommonParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.ExpressionParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.constants.DbParseConstants;
+import net.sf.jsqlparser.statement.execute.Execute;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class ExecuteParse implements Parse<Execute> {
+
+    private final ObjectNode executeObj;
+    public ExecuteParse() {
+        executeObj = JacksonHelperUtil.getObjectNode();
+        // action parse
+        executeObj.put(DbParseConstants.ACTION, DbParseConstants.EXECUTE);
+    }
+
+    @Override
+    public ObjectNode parse(Execute parseObj) {
+
+        // execute name parse
+        CommonParse.parseName(parseObj.getName(), executeObj);
+
+        // expressions parse
+        ExpressionParse.parse(parseObj.getExprList(), executeObj);
+
+        return executeObj;
+    }
+}

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

@@ -0,0 +1,37 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse.action;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import io.arex.agent.thirdparty.util.JacksonHelperUtil;
+import io.arex.agent.thirdparty.util.parse.sqlparse.Parse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.ColumnParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.TableParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.constants.DbParseConstants;
+import net.sf.jsqlparser.statement.insert.Insert;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class InsertParse implements Parse<Insert> {
+
+    private final ObjectNode insertObj;
+    public InsertParse() {
+        insertObj = JacksonHelperUtil.getObjectNode();
+        // action parse
+        insertObj.put(DbParseConstants.ACTION, DbParseConstants.INSERT);
+    }
+    @Override
+    public ObjectNode parse(Insert parseObj) {
+        // table parse
+        TableParse.parse(parseObj.getTable(), insertObj);
+
+        // columns parse
+        ColumnParse.parse(parseObj.getColumns(), parseObj.getItemsList(), insertObj);
+
+        // setColumns parse
+        ColumnParse.parseSetColumns(parseObj.getSetColumns(), parseObj.getSetExpressionList(), insertObj);
+
+        return insertObj;
+    }
+}

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

@@ -0,0 +1,38 @@
+package io.arex.agent.thirdparty.util.parse.sqlparse.action;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import io.arex.agent.thirdparty.util.JacksonHelperUtil;
+import io.arex.agent.thirdparty.util.parse.sqlparse.Parse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.ColumnParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.ExpressionParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.parse.TableParse;
+import io.arex.agent.thirdparty.util.parse.sqlparse.constants.DbParseConstants;
+import net.sf.jsqlparser.statement.replace.Replace;
+
+/**
+ * @author niyan
+ * @date 2024/4/3
+ * @since 1.0.0
+ */
+public class ReplaceParse implements Parse<Replace> {
+
+    private final ObjectNode replaceObj;
+    public ReplaceParse() {
+        replaceObj = JacksonHelperUtil.getObjectNode();
+        // action parse
+        replaceObj.put(DbParseConstants.ACTION, DbParseConstants.REPLACE);
+    }
+    @Override
+    public ObjectNode parse(Replace parseObj) {
+        // table parse
+        TableParse.parse(parseObj.getTable(), replaceObj);
+
+        // columns parse
+        ColumnParse.parse(parseObj.getColumns(), parseObj.getItemsList(), replaceObj);
+
+        // expressions parse
+        ExpressionParse.parse(parseObj.getExpressions(), parseObj.getColumns(), replaceObj);
+
+        return replaceObj;
+    }
+}