fix splitting multiple SQLs

weed/command/sql.go

@@ -18,43 +18,124 @@ import (
 )
 
 // splitSQLStatements splits a query string into individual SQL statements
-// This is a simple implementation that splits on semicolons outside of quoted strings
+// This robust implementation handles SQL comments, quoted strings, and escaped characters
 func splitSQLStatements(query string) []string {
 	var statements []string
 	var current strings.Builder
-	inSingleQuote := false
-	inDoubleQuote := false
-
+	
 	query = strings.TrimSpace(query)
 	if query == "" {
 		return []string{}
 	}
 
-	for _, char := range query {
-		switch char {
-		case '\'':
-			if !inDoubleQuote {
-				inSingleQuote = !inSingleQuote
+	runes := []rune(query)
+	i := 0
+	
+	for i < len(runes) {
+		char := runes[i]
+		
+		// Handle single-line comments (-- comment)
+		if char == '-' && i+1 < len(runes) && runes[i+1] == '-' {
+			// Include the entire comment in the current statement
+			for i < len(runes) && runes[i] != '\n' && runes[i] != '\r' {
+				current.WriteRune(runes[i])
+				i++
 			}
-			current.WriteRune(char)
-		case '"':
-			if !inSingleQuote {
-				inDoubleQuote = !inDoubleQuote
+			// Include the newline if present
+			if i < len(runes) {
+				current.WriteRune(runes[i])
+				i++
 			}
-			current.WriteRune(char)
-		case ';':
-			if !inSingleQuote && !inDoubleQuote {
-				stmt := strings.TrimSpace(current.String())
-				if stmt != "" {
-					statements = append(statements, stmt)
+			continue
+		}
+		
+		// Handle multi-line comments (/* comment */)
+		if char == '/' && i+1 < len(runes) && runes[i+1] == '*' {
+			current.WriteRune(char) // Include the /*
+			i++
+			if i < len(runes) {
+				current.WriteRune(runes[i])
+				i++
+			}
+			
+			// Skip to end of comment or end of input
+			for i < len(runes) {
+				current.WriteRune(runes[i])
+				if runes[i] == '*' && i+1 < len(runes) && runes[i+1] == '/' {
+					i++
+					current.WriteRune(runes[i]) // Include the */
+					i++
+					break
 				}
-				current.Reset()
-			} else {
-				current.WriteRune(char)
+				i++
 			}
-		default:
+			continue
+		}
+		
+		// Handle single-quoted strings
+		if char == '\'' {
+			current.WriteRune(char)
+			i++
+			
+			for i < len(runes) {
+				char = runes[i]
+				current.WriteRune(char)
+				
+				if char == '\'' {
+					// Check if it's an escaped quote
+					if i+1 < len(runes) && runes[i+1] == '\'' {
+						i++ // Skip the next quote (it's escaped)
+						if i < len(runes) {
+							current.WriteRune(runes[i])
+						}
+					} else {
+						break // End of string
+					}
+				}
+				i++
+			}
+			i++
+			continue
+		}
+		
+		// Handle double-quoted identifiers
+		if char == '"' {
+			current.WriteRune(char)
+			i++
+			
+			for i < len(runes) {
+				char = runes[i]
+				current.WriteRune(char)
+				
+				if char == '"' {
+					// Check if it's an escaped quote
+					if i+1 < len(runes) && runes[i+1] == '"' {
+						i++ // Skip the next quote (it's escaped)
+						if i < len(runes) {
+							current.WriteRune(runes[i])
+						}
+					} else {
+						break // End of identifier
+					}
+				}
+				i++
+			}
+			i++
+			continue
+		}
+		
+		// Handle semicolon (statement separator)
+		if char == ';' {
+			stmt := strings.TrimSpace(current.String())
+			if stmt != "" {
+				statements = append(statements, stmt)
+			}
+			current.Reset()
+		} else {
 			current.WriteRune(char)
 		}
+		
+		i++
 	}
 
 	// Add any remaining statement