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

internal/endtoend/testdata/update_cte/pgx/v4/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/pgconn"
+	"github.com/jackc/pgx/v4"
+)
+
+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/update_cte/pgx/v4/go/models.go

@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"database/sql"
+	"time"
+)
+
+type Td3Code struct {
+	ID        int32
+	TsCreated time.Time
+	TsUpdated time.Time
+	CreatedBy string
+	UpdatedBy string
+	Code      sql.NullString
+	Hash      sql.NullString
+	IsPrivate sql.NullBool
+}
+
+type Td3TestCode struct {
+	ID        int32
+	TsCreated time.Time
+	TsUpdated time.Time
+	CreatedBy string
+	UpdatedBy string
+	TestID    int32
+	CodeHash  string
+}

internal/endtoend/testdata/update_cte/pgx/v4/go/query.sql.go

@@ -0,0 +1,72 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+	"time"
+)
+
+const updateCode = `-- name: UpdateCode :one
+WITH cc AS (
+            UPDATE td3.codes
+            SET
+                created_by = $1,
+                updated_by  = $1,
+                code = $2,
+                hash = $3,
+                is_private = false
+            RETURNING hash
+)
+UPDATE td3.test_codes
+SET
+    created_by = $1,
+    updated_by  = $1,
+    test_id = $4,
+    code_hash = cc.hash
+    FROM cc
+RETURNING hash, id, ts_created, ts_updated, created_by, updated_by, test_id, code_hash
+`
+
+type UpdateCodeParams struct {
+	CreatedBy string
+	Code      sql.NullString
+	Hash      sql.NullString
+	TestID    int32
+}
+
+type UpdateCodeRow struct {
+	Hash      sql.NullString
+	ID        int32
+	TsCreated time.Time
+	TsUpdated time.Time
+	CreatedBy string
+	UpdatedBy string
+	TestID    int32
+	CodeHash  string
+}
+
+func (q *Queries) UpdateCode(ctx context.Context, arg UpdateCodeParams) (UpdateCodeRow, error) {
+	row := q.db.QueryRow(ctx, updateCode,
+		arg.CreatedBy,
+		arg.Code,
+		arg.Hash,
+		arg.TestID,
+	)
+	var i UpdateCodeRow
+	err := row.Scan(
+		&i.Hash,
+		&i.ID,
+		&i.TsCreated,
+		&i.TsUpdated,
+		&i.CreatedBy,
+		&i.UpdatedBy,
+		&i.TestID,
+		&i.CodeHash,
+	)
+	return i, err
+}

internal/endtoend/testdata/update_cte/pgx/v4/query.sql

@@ -0,0 +1,19 @@
+-- name: UpdateCode :one
+WITH cc AS (
+            UPDATE td3.codes
+            SET
+                created_by = $1,
+                updated_by  = $1,
+                code = $2,
+                hash = $3,
+                is_private = false
+            RETURNING hash
+)
+UPDATE td3.test_codes
+SET
+    created_by = $1,
+    updated_by  = $1,
+    test_id = $4,
+    code_hash = cc.hash
+    FROM cc
+RETURNING *;

internal/endtoend/testdata/update_cte/pgx/v4/schema.sql

@@ -0,0 +1,31 @@
+-- FILE: schema.sql
+
+DROP SCHEMA IF EXISTS td3 CASCADE;
+CREATE SCHEMA td3;
+
+CREATE TABLE td3.codes (
+            id SERIAL PRIMARY KEY,
+            ts_created timestamptz DEFAULT now() NOT NULL,
+            ts_updated timestamptz DEFAULT now() NOT NULL,
+            created_by text NOT NULL,
+            updated_by text NOT NULL,
+            
+            code text,
+            hash text,
+            is_private boolean
+);
+
+
+CREATE TABLE td3.test_codes (
+            id SERIAL PRIMARY KEY,
+            ts_created timestamptz DEFAULT now() NOT NULL,
+            ts_updated timestamptz DEFAULT now() NOT NULL,
+            created_by text NOT NULL,
+            updated_by text NOT NULL,
+
+            test_id integer NOT NULL,
+            code_hash text NOT NULL
+);
+
+-- FILE: query.sql
+

internal/endtoend/testdata/update_cte/pgx/v4/sqlc.json

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v4",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/update_cte/pgx/v5/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/update_cte/pgx/v5/go/models.go

@@ -0,0 +1,30 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type Td3Code struct {
+	ID        int32
+	TsCreated pgtype.Timestamptz
+	TsUpdated pgtype.Timestamptz
+	CreatedBy string
+	UpdatedBy string
+	Code      pgtype.Text
+	Hash      pgtype.Text
+	IsPrivate pgtype.Bool
+}
+
+type Td3TestCode struct {
+	ID        int32
+	TsCreated pgtype.Timestamptz
+	TsUpdated pgtype.Timestamptz
+	CreatedBy string
+	UpdatedBy string
+	TestID    int32
+	CodeHash  string
+}

internal/endtoend/testdata/update_cte/pgx/v5/go/query.sql.go

@@ -0,0 +1,72 @@
+// 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 updateCode = `-- name: UpdateCode :one
+WITH cc AS (
+            UPDATE td3.codes
+            SET
+                created_by = $1,
+                updated_by  = $1,
+                code = $2,
+                hash = $3,
+                is_private = false
+            RETURNING hash
+)
+UPDATE td3.test_codes
+SET
+    created_by = $1,
+    updated_by  = $1,
+    test_id = $4,
+    code_hash = cc.hash
+    FROM cc
+RETURNING hash, id, ts_created, ts_updated, created_by, updated_by, test_id, code_hash
+`
+
+type UpdateCodeParams struct {
+	CreatedBy string
+	Code      pgtype.Text
+	Hash      pgtype.Text
+	TestID    int32
+}
+
+type UpdateCodeRow struct {
+	Hash      pgtype.Text
+	ID        int32
+	TsCreated pgtype.Timestamptz
+	TsUpdated pgtype.Timestamptz
+	CreatedBy string
+	UpdatedBy string
+	TestID    int32
+	CodeHash  string
+}
+
+func (q *Queries) UpdateCode(ctx context.Context, arg UpdateCodeParams) (UpdateCodeRow, error) {
+	row := q.db.QueryRow(ctx, updateCode,
+		arg.CreatedBy,
+		arg.Code,
+		arg.Hash,
+		arg.TestID,
+	)
+	var i UpdateCodeRow
+	err := row.Scan(
+		&i.Hash,
+		&i.ID,
+		&i.TsCreated,
+		&i.TsUpdated,
+		&i.CreatedBy,
+		&i.UpdatedBy,
+		&i.TestID,
+		&i.CodeHash,
+	)
+	return i, err
+}

internal/endtoend/testdata/update_cte/pgx/v5/query.sql

@@ -0,0 +1,19 @@
+-- name: UpdateCode :one
+WITH cc AS (
+            UPDATE td3.codes
+            SET
+                created_by = $1,
+                updated_by  = $1,
+                code = $2,
+                hash = $3,
+                is_private = false
+            RETURNING hash
+)
+UPDATE td3.test_codes
+SET
+    created_by = $1,
+    updated_by  = $1,
+    test_id = $4,
+    code_hash = cc.hash
+    FROM cc
+RETURNING *;

internal/endtoend/testdata/update_cte/pgx/v5/schema.sql

@@ -0,0 +1,31 @@
+-- FILE: schema.sql
+
+DROP SCHEMA IF EXISTS td3 CASCADE;
+CREATE SCHEMA td3;
+
+CREATE TABLE td3.codes (
+            id SERIAL PRIMARY KEY,
+            ts_created timestamptz DEFAULT now() NOT NULL,
+            ts_updated timestamptz DEFAULT now() NOT NULL,
+            created_by text NOT NULL,
+            updated_by text NOT NULL,
+            
+            code text,
+            hash text,
+            is_private boolean
+);
+
+
+CREATE TABLE td3.test_codes (
+            id SERIAL PRIMARY KEY,
+            ts_created timestamptz DEFAULT now() NOT NULL,
+            ts_updated timestamptz DEFAULT now() NOT NULL,
+            created_by text NOT NULL,
+            updated_by text NOT NULL,
+
+            test_id integer NOT NULL,
+            code_hash text NOT NULL
+);
+
+-- FILE: query.sql
+

internal/endtoend/testdata/update_cte/pgx/v5/sqlc.json

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v5",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}