Method parseArrayConstructor

pkg/sql/parser/expressions_complex.go:281–348

parseArrayConstructor parses PostgreSQL ARRAY constructor syntax. Supports both ARRAY[...] (square bracket) and ARRAY(...) (subquery) forms. Examples: ARRAY[1, 2, 3] - Array literal with square brackets ARRAY['a', 'b', 'c'] - String array ARRAY[x, y, z]

()

//	ARRAY[x, y, z]                   - Array from expressions
//	ARRAY(SELECT id FROM users)      - Array from subquery
func (p *Parser) parseArrayConstructor() (*ast.ArrayConstructorExpression, error) {
	p.advance() // Consume ARRAY

	arrayExpr := ast.GetArrayConstructor()

	// Check for square bracket syntax: ARRAY[...]
	if p.isType(models.TokenTypeLBracket) {
		p.advance() // Consume [

		// Parse comma-separated list of expressions (can be empty)
		if !p.isType(models.TokenTypeRBracket) {
			for {
				elem, err := p.parseExpression()
				if err != nil {
					return nil, err
				}
				arrayExpr.Elements = append(arrayExpr.Elements, elem)

				if p.isType(models.TokenTypeComma) {
					p.advance() // Consume comma
				} else if p.isType(models.TokenTypeRBracket) {
					break
				} else {
					return nil, p.expectedError(", or ]")
				}
			}
		}

		// Expect closing bracket
		if !p.isType(models.TokenTypeRBracket) {
			return nil, p.expectedError("]")
		}
		p.advance() // Consume ]

		return arrayExpr, nil
	}

	// Check for parenthesis syntax: ARRAY(SELECT ...)
	if p.isType(models.TokenTypeLParen) {
		p.advance() // Consume (

		// Expect a subquery
		if !p.isType(models.TokenTypeSelect) && !p.isType(models.TokenTypeWith) {
			return nil, p.expectedError("SELECT in ARRAY subquery")
		}

		subquery, err := p.parseSubquery()
		if err != nil {
			return nil, err
		}

		selectStmt, ok := subquery.(*ast.SelectStatement)
		if !ok {
			return nil, p.expectedError("SELECT statement in ARRAY subquery")
		}
		arrayExpr.Subquery = selectStmt

		// Expect closing parenthesis