snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/endtoend/testdata/select_union_subquery/issue.md

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

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

@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+type DBTX interface {
+	ExecContext(context.Context, string, ...any) (sql.Result, error)
+	PrepareContext(context.Context, string) (*sql.Stmt, error)
+	QueryContext(context.Context, string, ...any) (*sql.Rows, error)
+	QueryRowContext(context.Context, string, ...any) *sql.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx *sql.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

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

@@ -0,0 +1,15 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"database/sql"
+)
+
+type Author struct {
+	ID   int32
+	Name string
+	Bio  sql.NullString
+}

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

@@ -0,0 +1,41 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+)
+
+const testSubqueryUnion = `-- name: TestSubqueryUnion :many
+SELECT tmp.id, tmp.name, tmp.bio FROM (
+  SELECT id, name, bio FROM authors
+  UNION
+  SELECT id, name, bio FROM authors
+) tmp LIMIT 5
+`
+
+func (q *Queries) TestSubqueryUnion(ctx context.Context) ([]Author, error) {
+	rows, err := q.db.QueryContext(ctx, testSubqueryUnion)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []Author
+	for rows.Next() {
+		var i Author
+		if err := rows.Scan(&i.ID, &i.Name, &i.Bio); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Close(); err != nil {
+		return nil, err
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}

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

@@ -0,0 +1,6 @@
+-- name: TestSubqueryUnion :many
+SELECT tmp.* FROM (
+  SELECT * FROM authors
+  UNION
+  SELECT * FROM authors
+) tmp LIMIT 5;

internal/endtoend/testdata/select_union_subquery/mysql/schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE authors (
+  id   int PRIMARY KEY,
+  name text      NOT NULL,
+  bio  text
+);

internal/endtoend/testdata/select_union_subquery/mysql/sqlc.yaml

@@ -0,0 +1,9 @@
+version: "2"
+sql:
+  - engine: "mysql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"

internal/endtoend/testdata/select_union_subquery/postgresql/go/db.go

@@ -0,0 +1,32 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5"
+	"github.com/jackc/pgx/v5/pgconn"
+)
+
+type DBTX interface {
+	Exec(context.Context, string, ...any) (pgconn.CommandTag, error)
+	Query(context.Context, string, ...any) (pgx.Rows, error)
+	QueryRow(context.Context, string, ...any) pgx.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx pgx.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

internal/endtoend/testdata/select_union_subquery/postgresql/go/models.go

@@ -0,0 +1,15 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type Author struct {
+	ID   int32
+	Name string
+	Bio  pgtype.Text
+}

internal/endtoend/testdata/select_union_subquery/postgresql/go/query.sql.go

@@ -0,0 +1,38 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+)
+
+const testSubqueryUnion = `-- name: TestSubqueryUnion :many
+SELECT tmp.id, tmp.name, tmp.bio FROM (
+  SELECT id, name, bio FROM authors
+  UNION
+  SELECT id, name, bio FROM authors
+) tmp LIMIT 5
+`
+
+func (q *Queries) TestSubqueryUnion(ctx context.Context) ([]Author, error) {
+	rows, err := q.db.Query(ctx, testSubqueryUnion)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []Author
+	for rows.Next() {
+		var i Author
+		if err := rows.Scan(&i.ID, &i.Name, &i.Bio); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}

internal/endtoend/testdata/select_union_subquery/postgresql/query.sql

@@ -0,0 +1,6 @@
+-- name: TestSubqueryUnion :many
+SELECT tmp.* FROM (
+  SELECT * FROM authors
+  UNION
+  SELECT * FROM authors
+) tmp LIMIT 5;

internal/endtoend/testdata/select_union_subquery/postgresql/schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE authors (
+  id   int PRIMARY KEY,
+  name text      NOT NULL,
+  bio  text
+);

internal/endtoend/testdata/select_union_subquery/postgresql/sqlc.yaml

@@ -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"