fix(compiler): Search SELECT and UPDATE the same way (#2841)

* fix(compiler): Search SELECT and UPDATE the same way

* Update sqlc.yaml

Author: kyleconroy

internal/compiler/output_columns.go

@@ -507,9 +507,10 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 			return ok
 		})
 	case *ast.UpdateStmt:
-		list = &ast.List{
-			Items: append(n.FromClause.Items, n.Relations.Items...),
-		}
+		var tv tableVisitor
+		astutils.Walk(&tv, n.FromClause)
+		astutils.Walk(&tv, n.Relations)
+		list = &tv.list
 	case *ast.DoStmt:
 		list = &ast.List{}
 	case *ast.CallStmt:

internal/endtoend/testdata/join_update/issue.md

@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/1100

internal/endtoend/testdata/join_update/postgresql/pgx/go/db.go

@@ -0,0 +1,6 @@
+-- name: Percentile :exec
+UPDATE group_calc_totals gct 
+SET npn = nem.npn 
+FROM producer_group_attribute ga 
+JOIN npn_external_map nem ON ga.npn_external_map_id = nem.id 
+WHERE gct.group_id = ga.group_id;

internal/endtoend/testdata/join_update/postgresql/pgx/go/models.go

@@ -0,0 +1,14 @@
+CREATE TABLE group_calc_totals (
+  npn  text,
+  group_id text
+);
+
+CREATE TABLE producer_group_attribute (
+  npn_external_map_id text,
+  group_id text
+);
+
+CREATE TABLE npn_external_map (
+  id   text,
+  npn  text
+);

internal/endtoend/testdata/join_update/postgresql/pgx/go/query.sql.go

@@ -0,0 +1,10 @@
+version: "2"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"