fix(compiler): support identifiers with schema (#2579)

* fix(compiler): support identifiers with schema

close #1858

* test: update endtoend

Author: orisano

internal/compiler/output_columns.go

@@ -612,19 +612,26 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 
 func outputColumnRefs(res *ast.ResTarget, tables []*Table, node *ast.ColumnRef) ([]*Column, error) {
 	parts := stringSlice(node.Fields)
-	var name, alias string
+	var schema, name, alias string
 	switch {
 	case len(parts) == 1:
 		name = parts[0]
 	case len(parts) == 2:
 		alias = parts[0]
 		name = parts[1]
+	case len(parts) == 3:
+		schema = parts[0]
+		alias = parts[1]
+		name = parts[2]
 	default:
 		return nil, fmt.Errorf("unknown number of fields: %d", len(parts))
 	}
 	var cols []*Column
 	var found int
 	for _, t := range tables {
+		if schema != "" && t.Rel.Schema != schema {
+			continue
+		}
 		if alias != "" && t.Rel.Name != alias {
 			continue
 		}

internal/endtoend/testdata/schema_scoped_list/mysql/go/query.sql.go

@@ -3,3 +3,6 @@ CREATE TABLE foo.bar (id serial not null);
 
 -- name: SchemaScopedList :many
 SELECT * FROM foo.bar;
+
+-- name: SchemaScopedColList :many
+SELECT foo.bar.id FROM foo.bar;

internal/endtoend/testdata/schema_scoped_list/mysql/query.sql

@@ -3,3 +3,6 @@ CREATE TABLE foo.bar (id serial not null);
 
 -- name: SchemaScopedList :many
 SELECT * FROM foo.bar;
+
+-- name: SchemaScopedColList :many
+SELECT foo.bar.id FROM foo.bar;

internal/endtoend/testdata/schema_scoped_list/postgresql/pgx/v4/go/query.sql.go

@@ -3,3 +3,6 @@ CREATE TABLE foo.bar (id serial not null);
 
 -- name: SchemaScopedList :many
 SELECT * FROM foo.bar;
+
+-- name: SchemaScopedColList :many
+SELECT foo.bar.id FROM foo.bar;

internal/endtoend/testdata/schema_scoped_list/postgresql/pgx/v4/query.sql

@@ -3,3 +3,6 @@ CREATE TABLE foo.bar (id serial not null);
 
 -- name: SchemaScopedList :many
 SELECT * FROM foo.bar;
+
+-- name: SchemaScopedColList :many
+SELECT foo.bar.id FROM foo.bar;