Function AddSetClause

pkg/transform/set.go:60–87

AddSetClause returns a Rule that appends a new assignment to the SET clause of an UPDATE statement. If a column with the same name already exists (case- insensitive), its value is overwritten rather than duplicated. Parameters: - column: the column name to set - valueSQL: a SQL expression string fo

(column, valueSQL string)

//
//	transform.Apply(stmt, transform.AddSetClause("updated_at", "NOW()"))
func AddSetClause(column, valueSQL string) Rule {
	return RuleFunc(func(stmt ast.Statement) error {
		upd, ok := stmt.(*ast.UpdateStatement)
		if !ok {
			return &ErrUnsupportedStatement{Transform: "AddSetClause", Got: stmtTypeName(stmt)}
		}

		valueExpr, err := parseValueExpr(valueSQL)
		if err != nil {
			return fmt.Errorf("AddSetClause: %w", err)
		}

		// Replace existing assignment for the same column (case-insensitive).
		for i, a := range upd.Assignments {
			if strings.EqualFold(identifierName(a.Column), column) {
				upd.Assignments[i].Value = valueExpr
				return nil
			}
		}

		// Append a new assignment.
		upd.Assignments = append(upd.Assignments, ast.UpdateExpression{
			Column: &ast.Identifier{Name: column},
			Value:  valueExpr,
		})
		return nil
	})
}

// SetClause is an alias for AddSetClause. It sets the value of a column in the
// UPDATE SET clause, adding it if not present or replacing it if already there.