fix: table alias can not find column

Author: nusr

.gitignore

@@ -2,6 +2,7 @@
 __pycache__
 .DS_Store
 .*.swp
+.vscode
 
 # Devenv
 .envrc

internal/compiler/output_columns.go

@@ -620,9 +620,10 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 			}
 			if n.Alias != nil {
 				table.Rel = &ast.TableName{
-					Catalog: table.Rel.Catalog,
-					Schema:  table.Rel.Schema,
-					Name:    *n.Alias.Aliasname,
+					Catalog:      table.Rel.Catalog,
+					Schema:       table.Rel.Schema,
+					Name:         *n.Alias.Aliasname,
+					OriginalName: table.Rel.Name,
 				}
 			}
 			tables = append(tables, table)
@@ -656,7 +657,7 @@ func outputColumnRefs(res *ast.ResTarget, tables []*Table, node *ast.ColumnRef)
 		if schema != "" && t.Rel.Schema != schema {
 			continue
 		}
-		if alias != "" && t.Rel.Name != alias {
+		if alias != "" && t.Rel.Name != alias && t.Rel.OriginalName != alias {
 			continue
 		}
 		for _, c := range t.Columns {

internal/endtoend/testdata/alias_join/mysql/go/db.go

@@ -0,0 +1,19 @@
+-- name: RetrieveAllWithDeprecation :many
+select
+  arn,
+  name,
+  aws_rds_databases.engine
+from
+  aws_rds_databases r
+  natural join aws_rds_databases_engines
+;
+
+-- name: RetrieveAllWithDeprecationOther :many
+select
+  arn,
+  name,
+  r.engine
+from
+  aws_rds_databases r
+  natural join aws_rds_databases_engines
+;

internal/endtoend/testdata/alias_join/mysql/go/models.go

@@ -0,0 +1,21 @@
+CREATE TABLE IF NOT EXISTS aws_rds_databases (
+  account_id VARCHAR(20) NOT NULL,
+  region VARCHAR(20) NOT NULL,
+  arn VARCHAR(20) NOT NULL,
+  name TEXT NOT NULL,
+  engine TEXT NOT NULL,
+  engine_version TEXT NOT NULL,
+
+  -- tags is a JSON object
+  tags TEXT NOT NULL,
+
+  UNIQUE (account_id, region, arn) 
+);
+
+CREATE TABLE IF NOT EXISTS aws_rds_databases_engines (
+  engine VARCHAR(20) NOT NULL,
+  engine_version VARCHAR(20) NOT NULL,
+  deprecation TEXT NOT NULL,
+
+  UNIQUE (engine, engine_version) 
+);

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

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "mysql",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

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

@@ -0,0 +1,19 @@
+-- name: RetrieveAllWithDeprecation :many
+select
+  arn,
+  name,
+  aws_rds_databases.engine
+from
+  aws_rds_databases r
+  natural join aws_rds_databases_engines
+;
+
+-- name: RetrieveAllWithDeprecationOther :many
+select
+  arn,
+  name,
+  r.engine
+from
+  aws_rds_databases r
+  natural join aws_rds_databases_engines
+;