Function ParseSQLWithDialect

pkg/transform/transform.go:258–280

ParseSQLWithDialect parses a SQL string using dialect-specific tokenization and parsing rules. This enables correct handling of dialect-specific syntax such as SQL Server TOP, MySQL backtick identifiers, and Snowflake QUALIFY. Use this when the input SQL uses dialect-specific constructs that the ge

(sql string, dialect keywords.SQLDialect)

//
//	tree, err := transform.ParseSQLWithDialect("SELECT TOP 10 * FROM users", keywords.DialectSQLServer)
func ParseSQLWithDialect(sql string, dialect keywords.SQLDialect) (*ast.AST, error) {
	tkz := tokenizer.GetTokenizer()
	defer tokenizer.PutTokenizer(tkz)

	if dialect != "" {
		tkz.SetDialect(dialect)
	}

	tokens, err := tkz.Tokenize([]byte(sql))
	if err != nil {
		return nil, fmt.Errorf("tokenize: %w", err)
	}

	p := parser.NewParser(parser.WithDialect(string(dialect)))
	defer p.Release()

	tree, err := p.ParseFromModelTokens(tokens)
	if err != nil {
		return nil, fmt.Errorf("parse: %w", err)
	}

	return tree, nil
}