Merge pull request #2950 from actiontech/fix_sql_line_comment

fix: scanner扫描带注释的xml文件报错

Author: winfredLIN

sqle/cmd/scannerd/scanners/common/common_test.go

@@ -111,3 +111,26 @@ func TestClearComments(t *testing.T) {
 		}
 	}
 }
+
+func TestClearLineComments(t *testing.T) {
+	testCases := []struct {
+		input    string
+		expected string
+	}{
+		{
+			"SELECT -- id,\n -- name,\n id, name FROM table;",
+			"SELECT id, name FROM table",
+		},
+		{
+			"SELECT -- id,\n -- name,\n -- age,\n id, name, age FROM table;",
+			"SELECT id, name, age FROM table",
+		},
+	}
+
+	for _, tc := range testCases {
+		actual := clearComments(tc.input)
+		if actual != tc.expected {
+			t.Errorf("Expected %s, got %s", tc.expected, actual)
+		}
+	}
+}

sqle/cmd/scannerd/scanners/common/parse.go

@@ -46,6 +46,7 @@ func Parse(_ context.Context, sqlText string) ([]driverV2.Node, error) {
 func clearComments(sqlText string) string {
 	// 将注释替换为一个空格，防止语句粘连
 	sqlText = regexp.MustCompile(`(?s)/\*.*?\*/`).ReplaceAllString(sqlText, " ")
+	sqlText = regexp.MustCompile(`--.*`).ReplaceAllString(sqlText, " ")
 	// 去除结尾分号后的内容
 	idx := strings.Index(sqlText, ";")
 	if idx >= 0 {