Function ReplaceSetClause

pkg/transform/set.go:138–159

ReplaceSetClause returns a Rule that completely replaces all SET assignments with the ones provided in the map. Keys are column names, values are SQL expression strings. This is useful for wholesale rewrites of the SET clause. Returns ErrUnsupportedStatement for non-UPDATE statements. Example: t

(assignments map[string]string)

//	    "updated_at": "NOW()",
//	}))
func ReplaceSetClause(assignments map[string]string) Rule {
	return RuleFunc(func(stmt ast.Statement) error {
		upd, ok := stmt.(*ast.UpdateStatement)
		if !ok {
			return &ErrUnsupportedStatement{Transform: "ReplaceSetClause", Got: stmtTypeName(stmt)}
		}

		newAssignments := make([]ast.UpdateExpression, 0, len(assignments))
		for col, valueSQL := range assignments {
			valueExpr, err := parseValueExpr(valueSQL)
			if err != nil {
				return fmt.Errorf("ReplaceSetClause: column %q: %w", col, err)
			}
			newAssignments = append(newAssignments, ast.UpdateExpression{
				Column: &ast.Identifier{Name: col},
				Value:  valueExpr,
			})
		}
		upd.Assignments = newAssignments
		return nil
	})
}