truncate colons at the end of queries

coroot · Aug 7, 2023 · b8a372a · b8a372a
1 parent 833e286
commit b8a372a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/obfuscate/sql.go b/obfuscate/sql.go
@@ -33,7 +33,8 @@ func Sql(query string) string {
 	query = rePunctuation.ReplaceAllString(query, " $1 ")
 	query = reWhitespace.ReplaceAllString(query, " ")
 	query = strings.ReplaceAll(query, " ,", ",")
-	query = strings.Trim(query, " ")
+	query = strings.TrimLeft(query, " ")
+	query = strings.TrimRight(query, "; ")
 	return query
 }
 

diff --git a/obfuscate/sql_test.go b/obfuscate/sql_test.go
@@ -242,6 +242,10 @@ func TestSql(t *testing.T) {
 			in:  "select lower('DdD'), cast(f as text)",
 			out: "select lower ( ? ), cast ( f as text )",
 		},
+		{
+			in:  "  select 1 ; ",
+			out: "select ?",
+		},
 	} {
 		assert.Equal(t, c.out, Sql(c.in), c.in)
 	}