Merge pull request #9 from lufeee/feat/check-indent

if query has leading whitespace delete indent rune

execinquery.go

@@ -3,6 +3,7 @@ package execinquery
 import (
 	"go/ast"
 	"strings"
+	"unicode"
 
 	"golang.org/x/tools/go/analysis"
 	"golang.org/x/tools/go/analysis/passes/inspect"
@@ -76,7 +77,10 @@ func run(pass *analysis.Pass) (interface{}, error) {
 						return
 					}
 
-					s = strings.Replace(basicLit.Value, "\"", "", -1)
+					s = strings.TrimLeftFunc(basicLit.Value, func(r rune) bool {
+						return !unicode.IsLetter(r) && !unicode.IsNumber(r)
+					})
+					s = strings.Replace(s, "\"", "", -1)
 				}
 
 			default:

testdata/src/a/a.go

@@ -6,6 +6,10 @@ import (
 	"testing"
 )
 
+const cquery = `
+	UPDATE * FROM test where test=?
+`
+
 func setup() *sql.DB {
 	db, _ := sql.Open("mysql", "test:test@tcp(test:3306)/test")
 	return db
@@ -31,4 +35,5 @@ func f(t *testing.T) {
 	var query1 string = "UPDATE * FROM test where test=?"
 	_, _ = db.Query(query, s)  // want "It\\'s better to use Execute method instead of Query method to execute `UPDATE` query"
 	_, _ = db.Query(query1, s) // want "It\\'s better to use Execute method instead of Query method to execute `UPDATE` query"
+	_, _ = db.Query(cquery, s) // want "It\\'s better to use Execute method instead of Query method to execute `UPDATE` query"
 }