Function splitSQLStatements

tests/integration/harness.go:763–818

(sql string)

}

func splitSQLStatements(sql string) []string {
	var statements []string
	var current strings.Builder
	inString := false
	stringChar := byte(0)
	inLineComment := false

	for i := 0; i < len(sql); i++ {
		c := sql[i]

		// Handle line comments (--) when not in a string
		if !inString && !inLineComment && c == '-' && i+1 < len(sql) && sql[i+1] == '-' {
			inLineComment = true
			i++ // skip second dash
			continue
		}

		// End line comment on newline
		if inLineComment {
			if c == '\n' {
				inLineComment = false
				current.WriteByte(' ') // replace comment with space
			}
			continue
		}

		// Handle string literals
		if (c == '\'' || c == '"') && (i == 0 || sql[i-1] != '\\') {
			if !inString {
				inString = true
				stringChar = c
			} else if c == stringChar {
				inString = false
			}
		}

		// Handle statement terminator
		if c == ';' && !inString {
			stmt := strings.TrimSpace(current.String())
			if stmt != "" {
				statements = append(statements, stmt)
			}
			current.Reset()
			continue
		}

		current.WriteByte(c)
	}

	// Add any remaining statement
	if stmt := strings.TrimSpace(current.String()); stmt != "" {
		statements = append(statements, stmt)
	}

	return statements
}

func truncate(s string, maxLen int) string {