Function ReplaceTable

pkg/transform/tables.go:39–69

ReplaceTable returns a Rule that replaces all occurrences of a table name throughout a SELECT, UPDATE, or DELETE statement. The replacement covers: - FROM clause table references - JOIN clause left and right table references - Column qualifiers (table.column identifiers in SELECT list, WHERE, ORDER

(oldName, newName string)

//
// Returns ErrUnsupportedStatement for INSERT or DDL statements.
func ReplaceTable(oldName, newName string) Rule {
	return RuleFunc(func(stmt ast.Statement) error {
		switch s := stmt.(type) {
		case *ast.SelectStatement:
			replaceTableInFrom(s.From, oldName, newName)
			replaceTableInJoins(s.Joins, oldName, newName)
			for i, col := range s.Columns {
				s.Columns[i] = replaceTableInExpr(col, oldName, newName)
			}
			s.Where = replaceTableInExpr(s.Where, oldName, newName)
			for i, ob := range s.OrderBy {
				s.OrderBy[i].Expression = replaceTableInExpr(ob.Expression, oldName, newName)
			}
			return nil
		case *ast.UpdateStatement:
			if strings.EqualFold(s.TableName, oldName) {
				s.TableName = newName
			}
			s.Where = replaceTableInExpr(s.Where, oldName, newName)
			return nil
		case *ast.DeleteStatement:
			if strings.EqualFold(s.TableName, oldName) {
				s.TableName = newName
			}
			s.Where = replaceTableInExpr(s.Where, oldName, newName)
			return nil
		default:
			return &ErrUnsupportedStatement{Transform: "ReplaceTable", Got: stmtTypeName(stmt)}
		}
	})
}

// AddTableAlias returns a Rule that assigns an alias to a specific table in a
// SELECT, UPDATE, or DELETE statement. For SELECT statements the alias is applied