Function QualifyColumns

pkg/transform/tables.go:119–133

QualifyColumns returns a Rule that prefixes every unqualified column reference in a SELECT statement's column list and WHERE clause with tableName. Only identifiers that have no existing table qualifier and whose name is not the wildcard "*" are modified. This is useful when merging standalone colu

(tableName string)

//
// Returns ErrUnsupportedStatement for non-SELECT statements.
func QualifyColumns(tableName string) Rule {
	return RuleFunc(func(stmt ast.Statement) error {
		sel, err := getSelect(stmt, "QualifyColumns")
		if err != nil {
			return err
		}
		for i, col := range sel.Columns {
			sel.Columns[i] = qualifyExpr(col, tableName)
		}
		if sel.Where != nil {
			sel.Where = qualifyExpr(sel.Where, tableName)
		}
		return nil
	})
}

func replaceTableInFrom(from []ast.TableReference, old, new string) {
	for i := range from {