Method parseUpdateStatement

pkg/sql/parser/dml_update.go:28–133

parseUpdateStatement parses an UPDATE statement

()

// parseUpdateStatement parses an UPDATE statement
func (p *Parser) parseUpdateStatement() (ast.Statement, error) {
	// We've already consumed the UPDATE token in matchType

	// Parse table name (supports schema.table qualification and double-quoted identifiers)
	tableName, err := p.parseQualifiedName()
	if err != nil {
		return nil, p.expectedError("table name")
	}

	// Parse SET
	if !p.isType(models.TokenTypeSet) {
		return nil, p.expectedError("SET")
	}
	p.advance() // Consume SET

	// Parse assignments
	updates := make([]ast.UpdateExpression, 0)
	for {
		// Parse column name (supports double-quoted identifiers)
		if !p.isIdentifier() {
			return nil, p.expectedError("column name")
		}
		columnName := p.currentToken.Token.Value
		p.advance()

		if !p.isType(models.TokenTypeEq) {
			return nil, p.expectedError("=")
		}
		p.advance() // Consume =

		// Parse value expression
		var expr ast.Expression
		if p.isStringLiteral() {
			expr = &ast.LiteralValue{Value: p.currentToken.Token.Value, Type: "string"}
			p.advance()
		} else if p.isNumericLiteral() {
			litType := "int"
			if strings.ContainsAny(p.currentToken.Token.Value, ".eE") {
				litType = "float"
			}
			expr = &ast.LiteralValue{Value: p.currentToken.Token.Value, Type: litType}
			p.advance()
		} else if p.isBooleanLiteral() {
			expr = &ast.LiteralValue{Value: p.currentToken.Token.Value, Type: "bool"}
			p.advance()
		} else {
			var err error
			expr, err = p.parseExpression()
			if err != nil {
				return nil, err
			}
		}

		// Create update expression
		columnExpr := &ast.Identifier{Name: columnName}
		updateExpr := ast.UpdateExpression{
			Column: columnExpr,
			Value:  expr,