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

internal/endtoend/testdata/batch_imports/postgresql/pgx/v4/go/batch.go

@@ -0,0 +1,131 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: batch.go
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+
+	"github.com/jackc/pgx/v4"
+)
+
+var (
+	ErrBatchAlreadyClosed = errors.New("batch already closed")
+)
+
+const getValues = `-- name: GetValues :batchmany
+SELECT a, b
+FROM myschema.foo
+WHERE b = $1
+`
+
+type GetValuesBatchResults struct {
+	br     pgx.BatchResults
+	tot    int
+	closed bool
+}
+
+func (q *Queries) GetValues(ctx context.Context, b []sql.NullInt32) *GetValuesBatchResults {
+	batch := &pgx.Batch{}
+	for _, a := range b {
+		vals := []any{
+			a,
+		}
+		batch.Queue(getValues, vals...)
+	}
+	br := q.db.SendBatch(ctx, batch)
+	return &GetValuesBatchResults{br, len(b), false}
+}
+
+func (b *GetValuesBatchResults) Query(f func(int, []MyschemaFoo, error)) {
+	defer b.br.Close()
+	for t := 0; t < b.tot; t++ {
+		var items []MyschemaFoo
+		if b.closed {
+			if f != nil {
+				f(t, items, ErrBatchAlreadyClosed)
+			}
+			continue
+		}
+		err := func() error {
+			rows, err := b.br.Query()
+			if err != nil {
+				return err
+			}
+			defer rows.Close()
+			for rows.Next() {
+				var i MyschemaFoo
+				if err := rows.Scan(&i.A, &i.B); err != nil {
+					return err
+				}
+				items = append(items, i)
+			}
+			return rows.Err()
+		}()
+		if f != nil {
+			f(t, items, err)
+		}
+	}
+}
+
+func (b *GetValuesBatchResults) Close() error {
+	b.closed = true
+	return b.br.Close()
+}
+
+const insertValues = `-- name: InsertValues :batchone
+INSERT INTO myschema.foo (a, b)
+VALUES ($1, $2)
+RETURNING a
+`
+
+type InsertValuesBatchResults struct {
+	br     pgx.BatchResults
+	tot    int
+	closed bool
+}
+
+type InsertValuesParams struct {
+	A sql.NullString
+	B sql.NullInt32
+}
+
+func (q *Queries) InsertValues(ctx context.Context, arg []InsertValuesParams) *InsertValuesBatchResults {
+	batch := &pgx.Batch{}
+	for _, a := range arg {
+		vals := []any{
+			a.A,
+			a.B,
+		}
+		batch.Queue(insertValues, vals...)
+	}
+	br := q.db.SendBatch(ctx, batch)
+	return &InsertValuesBatchResults{br, len(arg), false}
+}
+
+func (b *InsertValuesBatchResults) QueryRow(f func(int, sql.NullString, error)) {
+	defer b.br.Close()
+	for t := 0; t < b.tot; t++ {
+		var a sql.NullString
+		if b.closed {
+			if f != nil {
+				f(t, a, ErrBatchAlreadyClosed)
+			}
+			continue
+		}
+		row := b.br.QueryRow()
+		err := row.Scan(&a)
+		if f != nil {
+			f(t, a, err)
+		}
+	}
+}
+
+func (b *InsertValuesBatchResults) Close() error {
+	b.closed = true
+	return b.br.Close()
+}

internal/endtoend/testdata/batch_imports/postgresql/pgx/v4/go/db.go

@@ -0,0 +1,33 @@
+// 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
+	SendBatch(context.Context, *pgx.Batch) pgx.BatchResults
+}
+
+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/batch_imports/postgresql/pgx/v4/go/models.go

@@ -0,0 +1,14 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"database/sql"
+)
+
+type MyschemaFoo struct {
+	A sql.NullString
+	B sql.NullInt32
+}

internal/endtoend/testdata/batch_imports/postgresql/pgx/v4/go/query.sql.go

@@ -0,0 +1,25 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+const updateValues = `-- name: UpdateValues :exec
+UPDATE myschema.foo SET a = $1, b = $2
+`
+
+type UpdateValuesParams struct {
+	A sql.NullString
+	B sql.NullInt32
+}
+
+func (q *Queries) UpdateValues(ctx context.Context, arg UpdateValuesParams) error {
+	_, err := q.db.Exec(ctx, updateValues, arg.A, arg.B)
+	return err
+}

internal/endtoend/testdata/batch_imports/postgresql/pgx/v4/query.sql

@@ -0,0 +1,12 @@
+-- name: InsertValues :batchone
+INSERT INTO myschema.foo (a, b)
+VALUES ($1, $2)
+RETURNING a;
+
+-- name: GetValues :batchmany
+SELECT *
+FROM myschema.foo
+WHERE b = $1;
+
+-- name: UpdateValues :exec
+UPDATE myschema.foo SET a = $1, b = $2;

internal/endtoend/testdata/batch_imports/postgresql/pgx/v4/schema.sql

@@ -0,0 +1,3 @@
+CREATE SCHEMA myschema;
+CREATE TABLE myschema.foo (a text, b integer);
+

internal/endtoend/testdata/batch_imports/postgresql/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"
+    }
+  ]
+}