Function RemoveSetClause

pkg/transform/set.go:108–124

RemoveSetClause returns a Rule that removes a column from the UPDATE SET clause. If the column is not found the statement is left unchanged (no error). The comparison is case-insensitive. Returns ErrUnsupportedStatement for non-UPDATE statements. Example: transform.Apply(stmt, transform.RemoveSe

(column string)

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

		filtered := upd.Assignments[:0]
		for _, a := range upd.Assignments {
			if !strings.EqualFold(identifierName(a.Column), column) {
				filtered = append(filtered, a)
			}
		}
		upd.Assignments = filtered
		return nil
	})
}

// ReplaceSetClause returns a Rule that completely replaces all SET assignments
// with the ones provided in the map. Keys are column names, values are SQL