Method parseReturningColumns

pkg/sql/parser/dml_insert.go:234–259

parseReturningColumns parses the columns in a RETURNING clause Supports: column names, *, qualified names (table.column), expressions

()

// parseReturningColumns parses the columns in a RETURNING clause
// Supports: column names, *, qualified names (table.column), expressions
func (p *Parser) parseReturningColumns() ([]ast.Expression, error) {
	var columns []ast.Expression

	for {
		// Check for * (return all columns)
		if p.isType(models.TokenTypeMul) {
			columns = append(columns, &ast.Identifier{Name: "*"})
			p.advance()
		} else {
			// Parse expression (can be column name, qualified name, or expression)
			expr, err := p.parseExpression()
			if err != nil {
				return nil, fmt.Errorf("failed to parse RETURNING column: %w", err)
			}
			columns = append(columns, expr)
		}

		// Check for comma to continue parsing more columns
		if !p.isType(models.TokenTypeComma) {
			break
		}
		p.advance() // Consume comma
	}

	return columns, nil
}

// parseOnConflictClause parses the ON CONFLICT clause (PostgreSQL UPSERT)
// Syntax: ON CONFLICT [(columns)] | ON CONSTRAINT name DO NOTHING | DO UPDATE SET ...