add parser

Author: hasa1K

parser/Test0.java

@@ -0,0 +1,54 @@
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+
+public class DatabaseConnection {
+    private Connection connection;
+
+
+    public Connection getConnection() {
+        return connection;
+    }
+
+    public ResultSet executeQuery(String sqlQuery, String sqlQuery1) throws SQLException {
+        Statement statement = connection.createStatement();
+        String a = sqlQuery+sqlQuery1;
+        ResultSet resultSet = statement.executeQuery(a);
+
+        String formattedString = String.format("%s %s", sqlQuery, sqlQuery1);
+
+            ResultSet resultSet1 = statement.executeQuery(formattedString);
+        return resultSet;
+    }
+}
+
+public class DatabaseQuery {
+    private Connection connection;
+
+    public DatabaseQuery(Connection connection) {
+        this.connection = connection;
+    }
+
+
+
+    public static void main(String[] args) {
+        String jdbcUrl = "jdbc:mysql://localhost:3306/mydatabase";
+        String username = "yourUsername";
+        String password = "yourPassword";
+
+        try {
+            DatabaseConnection dbQuery = new DatabaseConnection();
+
+            String sqlQuery = "SELECT * FROM employees";
+            String sqlQuery1 = " limit 1";
+            ResultSet resultSet = dbQuery.executeQuery(sqlQuery, sqlQuery1);
+
+
+
+            connection.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+}

parser/parser.go

@@ -1,6 +1,8 @@
 package parser
 
 import (
+	"strings"
+
 	"github.com/antlr4-go/antlr/v4"
 
 	javaAntlr "github.com/actiontech/java-sql-extractor/java_antlr"
@@ -12,20 +14,56 @@ type JavaVisitor struct {
 	ExecSqlExpressions []*Expression
 
 	currentNode  interface{}
+	currentFile  *FileBlock
 	currentClass *ClassBlock
 	currentFunc  *FuncBlock
+	currentExpr  *Expression
 
 	isStatic bool
 }
 
+// 获取变量类型，并判断变量类型是否为import引入的类
+func (v *JavaVisitor) getClassFromVar(variableName string) (string, bool) {
+	var variableType string
+	if v.currentFunc != nil {
+		for _, vari := range v.currentFunc.Variables {
+			if vari.Name == variableName {
+				variableType = vari.Type
+			}
+		}
+	}
+	if variableType == "" {
+		return variableType, false
+	}
+	for _, importStr := range v.currentFile.ImportClass {
+		strSlice := strings.Split(importStr, ".")
+		if strSlice[len(strSlice)-1] == variableName {
+			return importStr, true
+		}
+	}
+	return variableType, false
+}
+
+func (v *JavaVisitor) getClassFromSingleFile(className string) *ClassBlock {
+	for _, classBlock := range v.currentFile.ClassBlocks {
+		if classBlock.Name == className {
+			return classBlock
+		}
+	}
+	return nil
+}
+
 type FileBlock struct {
 	ClassBlocks []*ClassBlock
+	ImportClass []string
 }
 
 type ClassBlock struct {
 	Name       string
 	FuncBlocks []*FuncBlock
 	Variables  []*Variable // 静态变量
+
+	isDeclare bool // 判断该类是在申明状态还是解析状态
 }
 
 func (v *ClassBlock) UpdateVariable(variable *Variable) {
@@ -36,10 +74,24 @@ func (v *ClassBlock) UpdateVariable(variable *Variable) {
 	}
 }
 
+func (c *ClassBlock) getFuncFromClass(funcName string) *FuncBlock {
+	for _, funcBlock := range c.FuncBlocks {
+		if funcBlock.Name == funcName {
+			return funcBlock
+		}
+	}
+	return nil
+}
+
 type FuncBlock struct {
 	Name      string
 	CodeBlock *CodeBlock
 	Variables []*Variable // 函数中所有的变量
+	Pointer   *javaAntlr.MethodDeclarationContext
+	isDeclare bool // 判断该方法是在申明状态还是解析状态
+	Params    []*Param // 函数参数列表
+
+	CalledExprs []*Expression // 调用该函数的表达式列表
 
 	Parent interface{}
 }
@@ -52,6 +104,11 @@ func (v *FuncBlock) UpdateVariable(variable *Variable) {
 	}
 }
 
+type Param struct {
+	Content string
+	Type    string
+}
+
 type CodeBlock struct {
 	Variables  []*Variable
 	CodeBlocks []*CodeBlock
@@ -61,16 +118,26 @@ type CodeBlock struct {
 
 type Variable struct {
 	Name  string
-	Value *Primary
+	Value *Expression
 	Level string
 	Type  string
+
+	Pointer *javaAntlr.VariableDeclaratorContext
 }
 
 type Expression struct {
-	SubExpr    *Expression
-	Content    *Primary
-	MethodCall string
+	Next       *Expression // 子节点
+	Before     *Expression // 父节点
+	Content    string
+	RuleIndex  int
 	Arguments  []*Expression
+	MethodCall *FuncBlock
+	Depth      int    // 调用深度;例如a.b.c  a的调用深度为1 b的调用深度为2 c的调用深度为3
+	Symbol     string // 表达式后面跟的符号；例如+和.
+
+	IsImport   bool
+	ImportName string
+	LocalClass *ClassBlock // 指向表达式a.c()中a的class
 
 	Node interface{}
 }

parser/parser_test.go

@@ -64,14 +64,15 @@ func TestJavaFiles(t *testing.T) {
 }
 
 func TestSingleJavaFile(t *testing.T) {
-	sqls, err := GetSqlFromJavaFile("/root/javaexample/test/Test6.java")
+	javaFile := "/root/javaexample/parser/Test0.java"
+	sqls, err := GetSqlFromJavaFile(javaFile)
 	if err != nil {
 		t.Error(err)
 	}
-	sqlFileSqls := getSqlsFromSqlFile("/root/javaexample/test/Test6.java" + ".sql")
+	sqlFileSqls := getSqlsFromSqlFile(javaFile + ".sql")
 	for i, sql := range sqls {
 		if sql != sqlFileSqls[i] {
-			t.Error(fmt.Errorf("sql parser failed, java file: %s", "/root/javaexample/test/Test6.java"))
+			t.Error(fmt.Errorf("sql parser failed, java file: %s", javaFile))
 		}
 	}
 }

parser/symbol.go

@@ -0,0 +1,11 @@
+package parser
+
+const STATIC = "static"
+
+
+const ASSIGN = "="
+const DOT = "."
+const PLUS = "+"
+
+// 自定义 . 和 = 符号的ruleindex
+const TerminalRuleIndex = -1