Method parseMainStatementAfterWith

pkg/sql/parser/cte.go:227–242

parseMainStatementAfterWith parses the main statement after WITH clause. It supports SELECT, INSERT, UPDATE, and DELETE statements, routing them to the appropriate parsers while preserving set operation support for SELECT statements.

()

// It supports SELECT, INSERT, UPDATE, and DELETE statements, routing them to the appropriate
// parsers while preserving set operation support for SELECT statements.
func (p *Parser) parseMainStatementAfterWith() (ast.Statement, error) {
	if p.isType(models.TokenTypeSelect) {
		p.advance() // Consume SELECT
		return p.parseSelectWithSetOperations()
	} else if p.isType(models.TokenTypeInsert) {
		p.advance() // Consume INSERT
		return p.parseInsertStatement()
	} else if p.isType(models.TokenTypeUpdate) {
		p.advance() // Consume UPDATE
		return p.parseUpdateStatement()
	} else if p.isType(models.TokenTypeDelete) {
		p.advance() // Consume DELETE
		return p.parseDeleteStatement()
	}
	return nil, p.expectedError("SELECT, INSERT, UPDATE, or DELETE after WITH")
}