Function extractQueries

pkg/sql/parser/integration_test.go:144–200

extractQueries extracts individual SQL queries from a file Queries are separated by semicolons, and comments are handled

(content string)

// extractQueries extracts individual SQL queries from a file
// Queries are separated by semicolons, and comments are handled
func extractQueries(content string) []string {
	queries := []string{}
	currentQuery := strings.Builder{}
	inBlockComment := false

	lines := strings.Split(content, "\n")

	for _, line := range lines {
		trimmed := strings.TrimSpace(line)

		// Handle line comments
		if strings.HasPrefix(trimmed, "--") {
			continue
		}

		// Handle block comments
		if strings.Contains(line, "/*") {
			inBlockComment = true
		}
		if strings.Contains(line, "*/") {
			inBlockComment = false
			continue
		}
		if inBlockComment {
			continue
		}

		// Skip empty lines
		if trimmed == "" {
			continue
		}

		// Add line to current query
		currentQuery.WriteString(line)
		currentQuery.WriteString("\n")

		// Check if query ends with semicolon
		if strings.HasSuffix(trimmed, ";") {
			query := strings.TrimSpace(currentQuery.String())
			// Remove trailing semicolon
			query = strings.TrimSuffix(query, ";")
			query = strings.TrimSpace(query)
			if query != "" {
				queries = append(queries, query)
			}
			currentQuery.Reset()
		}
	}

	// Add last query if it doesn't end with semicolon
	lastQuery := strings.TrimSpace(currentQuery.String())
	if lastQuery != "" && lastQuery != ";" {
		queries = append(queries, lastQuery)
	}

	return queries
}