sqlc-dev · kyleconroy · Nov 25, 2024 · Oct 14, 2024
diff --git a/internal/compiler/find_params.go b/internal/compiler/find_params.go
@@ -195,7 +195,7 @@ func (p paramSearch) Visit(node ast.Node) astutils.Visitor {
 		if n.Sel == nil {
 			p.parent = node
 		} else {
-			if sel, ok := n.Sel.(*ast.SelectStmt); ok && sel.FromClause != nil {
+			if sel, ok := n.Sel.(*ast.SelectStmt); ok && sel.FromClause != nil && len(sel.FromClause.Items) > 0 {
 				from := sel.FromClause
 				if schema, ok := from.Items[0].(*ast.RangeVar); ok && schema != nil {
 					p.rangeVar = &ast.RangeVar{

diff --git a/internal/endtoend/testdata/in_union/mysql/go/db.go b/internal/endtoend/testdata/in_union/mysql/go/db.go
diff --git a/internal/endtoend/testdata/in_union/mysql/go/models.go b/internal/endtoend/testdata/in_union/mysql/go/models.go
diff --git a/internal/endtoend/testdata/in_union/mysql/go/query.sql.go b/internal/endtoend/testdata/in_union/mysql/go/query.sql.go
diff --git a/internal/endtoend/testdata/in_union/mysql/query.sql b/internal/endtoend/testdata/in_union/mysql/query.sql
@@ -0,0 +1,3 @@
+-- name: GetAuthors :many
+SELECT * FROM authors
+WHERE author_id IN (SELECT author_id FROM book1 UNION SELECT author_id FROM book2);
diff --git a/internal/endtoend/testdata/in_union/mysql/schema.sql b/internal/endtoend/testdata/in_union/mysql/schema.sql
@@ -0,0 +1,15 @@
+CREATE TABLE authors (
+  id   int PRIMARY KEY,
+  name text      NOT NULL,
+  bio  text
+);
+CREATE TABLE book1 (
+  author_id  int PRIMARY KEY,
+  name text,
+  FOREIGN KEY (`author_id`) REFERENCES `authors` (`id`) ON DELETE CASCADE
+);
+CREATE TABLE book2 (
+  author_id  int PRIMARY KEY,
+  name text,
+  FOREIGN KEY (`author_id`) REFERENCES `authors` (`id`) ON DELETE CASCADE
+);
diff --git a/internal/endtoend/testdata/in_union/mysql/sqlc.json b/internal/endtoend/testdata/in_union/mysql/sqlc.json
@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "mysql",
+      "path": "go",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}