fix: issue gogf#4106

sanrentai · Jan 9, 2025 · 972734b · 972734b
1 parent a233eca
commit 972734b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/contrib/drivers/mssql/mssql_do_filter.go b/contrib/drivers/mssql/mssql_do_filter.go
@@ -88,6 +88,10 @@ func (d *Driver) parseSql(toBeCommittedSql string) (string, error) {
 }
 
 func (d *Driver) handleSelectSqlReplacement(toBeCommittedSql string) (newSql string, err error) {
+	toBeCommittedSql, err = gdb.FormatMultiLineSqlToSingle(toBeCommittedSql)
+	if err != nil {
+		return "", err
+	}
 	// SELECT * FROM USER WHERE ID=1 LIMIT 1
 	match, err := gregex.MatchString(`^SELECT(.+?)LIMIT\s+1$`, toBeCommittedSql)
 	if err != nil {

diff --git a/contrib/drivers/mssql/mssql_do_filter_test.go b/contrib/drivers/mssql/mssql_do_filter_test.go
@@ -58,8 +58,10 @@ func TestDriver_handleSelectSqlReplacement(t *testing.T) {
 		d := &Driver{}
 
 		// LIMIT 1
-		inputSql := "SELECT * FROM User WHERE ID = 1 LIMIT 1"
-		expectedSql := "SELECT TOP 1 * FROM User WHERE ID = 1"
+		inputSql := `SELECT a,
+b
+FROM User WHERE ID = 1 LIMIT 1`
+		expectedSql := `SELECT TOP 1 a, b FROM User WHERE ID = 1`
 		resultSql, err := d.handleSelectSqlReplacement(inputSql)
 		t.AssertNil(err)
 		t.Assert(resultSql, expectedSql)