Function splitStatements

pkg/sql/parser/corpus_test.go:102–137

splitStatements splits SQL content by semicolons, respecting comments. This is a simple splitter - not a full tokenizer-aware one.

(content string)

// splitStatements splits SQL content by semicolons, respecting comments.
// This is a simple splitter - not a full tokenizer-aware one.
func splitStatements(content string) []string {
	var statements []string
	var current strings.Builder
	lines := strings.Split(content, "\n")

	for _, line := range lines {
		trimmed := strings.TrimSpace(line)
		// Skip pure comment lines for splitting purposes, but include them in current statement
		if strings.HasPrefix(trimmed, "--") {
			current.WriteString(line)
			current.WriteString("\n")
			continue
		}

		// Check if line ends with semicolon (simple heuristic)
		if strings.HasSuffix(trimmed, ";") {
			current.WriteString(strings.TrimSuffix(line, ";"))
			stmt := strings.TrimSpace(current.String())
			if stmt != "" && !isOnlyComments(stmt) {
				statements = append(statements, stmt)
			}
			current.Reset()
		} else {
			current.WriteString(line)
			current.WriteString("\n")
		}
	}

	// Handle trailing statement without semicolon
	remaining := strings.TrimSpace(current.String())
	if remaining != "" && !isOnlyComments(remaining) {
		statements = append(statements, remaining)
	}

	return statements
}

// isOnlyComments returns true if the string contains only comment lines and whitespace.
func isOnlyComments(s string) bool {