fix(normalizer): INSERT stmt space normalization

traP-jp · Feb 3, 2025 · d31012f · d31012f
1 parent ed8d415
commit d31012f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/normalizer/query_normalizer.go b/normalizer/query_normalizer.go
@@ -6,6 +6,7 @@ import (
 )
 
 var spaceRegex = regexp.MustCompile(`\s+`)
+var insertRegex = regexp.MustCompile(`INSERT INTO (\w+)\s*\(`)
 var inRegex = regexp.MustCompile(`IN\s*\((\?,\s*)+\?\)`)
 var valuesRegex = regexp.MustCompile(`VALUES\s*\((\?,\s*)+\?\)`)
 
@@ -20,6 +21,9 @@ func NormalizeQuery(query string) string {
 	// remove backquotes
 	query = strings.ReplaceAll(query, "`", "")
 
+	// INSERT INTO table(... -> INSERT INTO table (...
+	query = insertRegex.ReplaceAllString(query, "INSERT INTO $1 (")
+
 	// IN (?, ?, ?) -> IN (?)
 	query = inRegex.ReplaceAllString(query, "IN (?)")
 

diff --git a/normalizer/query_normalizer_test.go b/normalizer/query_normalizer_test.go
@@ -24,7 +24,7 @@ func TestNormalizeQuery(t *testing.T) {
 			expected: "SELECT id from table;",
 		},
 		{
-			query:    "INSERT INTO table (name, col) VALUES (?, ?);",
+			query:    "INSERT INTO table(name, col) VALUES (?, ?);",
 			expected: "INSERT INTO table (name, col) VALUES (?);",
 		},
 		{