restore order of IndexFunc and TrimFunc->TrimLeftFunc

Alex Charis · bramp · commit 789ee6d442e6 · 2018-06-06T07:45:08.000-07:00
Signed-off-by: Alex Charis &lt;acharis@hubspot.com&gt;
diff --git a/analyzer.go b/analyzer.go
@@ -53,10 +53,11 @@ const (
 func Preview(sql string) int {
 	trimmed := StripLeadingComments(sql)
 
-	firstWord := strings.TrimLeftFunc(trimmed, func(r rune) bool { return !unicode.IsLetter(r) })
-	if end := strings.IndexFunc(firstWord, unicode.IsSpace); end != -1 {
-		firstWord = firstWord[:end]
+	firstWord := trimmed
+	if end := strings.IndexFunc(trimmed, unicode.IsSpace); end != -1 {
+		firstWord = trimmed[:end]
 	}
+	firstWord = strings.TrimLeftFunc(firstWord, func(r rune) bool { return !unicode.IsLetter(r) })
 	// Comparison is done in order of priority.
 	loweredFirstWord := strings.ToLower(firstWord)
 	switch loweredFirstWord {
diff --git a/analyzer_test.go b/analyzer_test.go
@@ -70,6 +70,7 @@ func TestPreview(t *testing.T) {
 		{"unknown", StmtUnknown},
 
 		{"/* leading comment */ select ...", StmtSelect},
+		{"/* leading comment */ (select ...", StmtSelect},
 		{"/* leading comment */ /* leading comment 2 */ select ...", StmtSelect},
 		{"/*! MySQL-specific comment */", StmtComment},
 		{"/*!50708 MySQL-version comment */", StmtComment},
