Method parseAlterTableUpdate

parser/parser_alter.go:886–931

Syntax: ALTER TABLE UPDATE column1 = expr1 [, column2 = expr2, ...] [IN PARTITION partition_id] WHERE condition

(pos Pos)

// Syntax: ALTER TABLE UPDATE column1 = expr1 [, column2 = expr2, ...] [IN PARTITION partition_id] WHERE condition
func (p *Parser) parseAlterTableUpdate(pos Pos) (AlterTableClause, error) {
	if err := p.expectKeyword(KeywordUpdate); err != nil {
		return nil, err
	}

	// Parse at least one assignment
	assignments := make([]*UpdateAssignment, 0)
	assignment, err := p.parseUpdateAssignment(p.Pos())
	if err != nil {
		return nil, err
	}
	assignments = append(assignments, assignment)

	// Parse additional comma-separated assignments
	for p.tryConsumeTokenKind(TokenKindComma) != nil {
		assignment, err = p.parseUpdateAssignment(p.Pos())
		if err != nil {
			return nil, err
		}
		assignments = append(assignments, assignment)
	}

	var inPartition *PartitionClause
	if p.tryConsumeKeywords(KeywordIn) {
		inPartition, err = p.parsePartitionClause(p.Pos())
		if err != nil {
			return nil, err
		}
	}
	if err := p.expectKeyword(KeywordWhere); err != nil {
		return nil, err
	}

	whereExpr, err := p.parseExpr(p.Pos())
	if err != nil {
		return nil, err
	}

	return &AlterTableUpdate{
		UpdatePos:    pos,
		StatementEnd: whereExpr.End(),
		Assignments:  assignments,
		InPartition:  inPartition,
		WhereClause:  whereExpr,
	}, nil
}

// Parse column = expression assignment
func (p *Parser) parseUpdateAssignment(pos Pos) (*UpdateAssignment, error) {