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

internal/endtoend/testdata/cte_update/issue.md

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

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

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

internal/endtoend/testdata/cte_update/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_update/postgresql/pgx/go/models.go

@@ -0,0 +1,16 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+type Attribute struct {
+	ID   int64
+	Name string
+}
+
+type AttributeValue struct {
+	ID        int64
+	Val       string
+	Attribute int64
+}

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

@@ -0,0 +1,42 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+const updateAttribute = `-- name: UpdateAttribute :one
+with updated_attribute as (UPDATE attribute_value
+    SET
+        val = CASE WHEN $1::bool THEN $2 ELSE val END
+    WHERE attribute_value.id = $3
+    RETURNING id,attribute,val)
+select updated_attribute.id, val, name
+from updated_attribute
+         left join attribute on updated_attribute.attribute = attribute.id
+`
+
+type UpdateAttributeParams struct {
+	FilterValue pgtype.Bool
+	Value       pgtype.Text
+	ID          pgtype.Int8
+}
+
+type UpdateAttributeRow struct {
+	ID   int64
+	Val  string
+	Name pgtype.Text
+}
+
+func (q *Queries) UpdateAttribute(ctx context.Context, arg UpdateAttributeParams) (UpdateAttributeRow, error) {
+	row := q.db.QueryRow(ctx, updateAttribute, arg.FilterValue, arg.Value, arg.ID)
+	var i UpdateAttributeRow
+	err := row.Scan(&i.ID, &i.Val, &i.Name)
+	return i, err
+}

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

@@ -0,0 +1,9 @@
+-- name: UpdateAttribute :one
+with updated_attribute as (UPDATE attribute_value
+    SET
+        val = CASE WHEN @filter_value::bool THEN @value ELSE val END
+    WHERE attribute_value.id = @id
+    RETURNING id,attribute,val)
+select updated_attribute.id, val, name
+from updated_attribute
+         left join attribute on updated_attribute.attribute = attribute.id;

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

@@ -0,0 +1,12 @@
+create table attribute_value
+(
+    id              bigserial not null,
+    val             text      not null,
+    attribute       bigint    not null
+);
+
+create table attribute
+(
+    id              bigserial      not null,
+    name            text           not null
+);

internal/endtoend/testdata/cte_update/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"