Index CTEs inside resolveCatalogRefs

internal/compiler/resolve.go

@@ -67,10 +67,14 @@ func (comp *Compiler) resolveCatalogRefs(qc *QueryCatalog, rvs []*ast.RangeVar,
 				continue
 			}
 			// If the table name doesn't exist, first check if it's a CTE
-			if _, qcerr := qc.GetTable(fqn); qcerr != nil {
+			qctab, qcerr := qc.GetTable(fqn)
+			if qcerr != nil {
 				return nil, err
 			}
-			continue
+			table = catalog.Table{
+				Rel:     qctab.Rel,
+				Columns: convertCompilerColumnsToCatalogColumns(qctab.Columns),
+			}
 		}
 		err = indexTable(table)
 		if err != nil {
@@ -635,3 +639,32 @@ func (comp *Compiler) resolveCatalogRefs(qc *QueryCatalog, rvs []*ast.RangeVar,
 	}
 	return a, nil
 }
+
+func convertCompilerColumnsToCatalogColumns(qccols []*Column) []*catalog.Column {
+	ret := make([]*catalog.Column, len(qccols))
+	for i, v := range qccols {
+		ret[i] = convertCompilerColumnToCatalogColumn(v)
+	}
+	return ret
+}
+
+func convertCompilerColumnToCatalogColumn(qccol *Column) *catalog.Column {
+
+	var tn ast.TypeName
+	if qccol.Type != nil {
+		tn = *qccol.Type
+	} else {
+		tn = ast.TypeName{Name: qccol.DataType}
+	}
+
+	return &catalog.Column{
+		Name:       qccol.Name,
+		Type:       tn,
+		IsNotNull:  qccol.NotNull,
+		IsUnsigned: qccol.Unsigned,
+		IsArray:    qccol.IsArray,
+		ArrayDims:  qccol.ArrayDims,
+		Comment:    qccol.Comment,
+		Length:     qccol.Length,
+	}
+}

internal/endtoend/testdata/cte_recursive_join/issue.md

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

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

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

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

@@ -0,0 +1,20 @@
+-- name: GetReferralChain :many
+WITH RECURSIVE referral_chain AS (
+    SELECT user_id,
+           referrer_id,
+           1 AS level
+    FROM user_referrals r
+    WHERE r.user_id = $1
+
+    UNION ALL
+
+    SELECT r.user_id,
+           r.referrer_id,
+           rc.level + 1
+    FROM user_referrals r
+             JOIN referral_chain rc ON r.user_id = rc.referrer_id
+    WHERE rc.level < @level
+)
+SELECT *
+FROM referral_chain
+;

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

@@ -0,0 +1,10 @@
+CREATE TABLE user_referrals
+(
+    user_id     bigint                                 NOT NULL
+        CONSTRAINT user_referrals_pk
+            PRIMARY KEY,
+    referrer_id bigint                                 NOT NULL,
+    balance     numeric(38, 9)           DEFAULT 0.00  NOT NULL,
+    created_at  timestamp with time zone DEFAULT NOW() NOT NULL,
+    updated_at  timestamp with time zone DEFAULT NOW() NOT NULL
+);

internal/endtoend/testdata/cte_recursive_join/postgresql/pgx/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"