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

internal/endtoend/testdata/cte_recursive_subquery/issue.md

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

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/exec.json

@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/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/cte_recursive_subquery/postgresql/pgx/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 Version struct {
+	ID                int64
+	Name              pgtype.Text
+	PreviousVersionID int64
+}

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

@@ -0,0 +1,60 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+)
+
+const getLatestVersion = `-- name: GetLatestVersion :one
+WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE base.previous_version_id IS NULL
+  	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id
+)
+SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+FROM search_tree   
+ORDER BY search_tree.chain_id, chain_counter DESC
+`
+
+func (q *Queries) GetLatestVersion(ctx context.Context) (int64, error) {
+	row := q.db.QueryRow(ctx, getLatestVersion)
+	var id int64
+	err := row.Scan(&id)
+	return id, err
+}
+
+const getLatestVersionWithSubquery = `-- name: GetLatestVersionWithSubquery :one
+SELECT id
+FROM versions
+WHERE versions.id IN (
+  WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE versions.previous_version_id IS NULL
+	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id 
+  )
+  SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+  FROM search_tree   
+  ORDER BY search_tree.chain_id, chain_counter DESC
+)
+`
+
+func (q *Queries) GetLatestVersionWithSubquery(ctx context.Context) (int64, error) {
+	row := q.db.QueryRow(ctx, getLatestVersionWithSubquery)
+	var id int64
+	err := row.Scan(&id)
+	return id, err
+}

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/query.sql

@@ -0,0 +1,33 @@
+-- name: GetLatestVersion :one
+WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE base.previous_version_id IS NULL
+  	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id
+)
+SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+FROM search_tree   
+ORDER BY search_tree.chain_id, chain_counter DESC;
+
+-- name: GetLatestVersionWithSubquery :one
+SELECT id
+FROM versions
+WHERE versions.id IN (
+  WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE versions.previous_version_id IS NULL
+	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id 
+  )
+  SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+  FROM search_tree   
+  ORDER BY search_tree.chain_id, chain_counter DESC
+);

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/schema.sql

@@ -0,0 +1,5 @@
+CREATE TABLE versions (
+  id   BIGSERIAL PRIMARY KEY,
+  name TEXT,
+  previous_version_id bigint NOT NULL
+);

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/sqlc.yaml

@@ -0,0 +1,14 @@
+version: "2"
+cloud:
+  project: "01HAQMMECEYQYKFJN8MP16QC41"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"
+    database:
+      managed: true