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

internal/endtoend/testdata/output_file_names/pgx/v4/go/batch_gen.go

@@ -0,0 +1,76 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: batch_gen.go
+
+package querytest
+
+import (
+	"context"
+	"errors"
+
+	"github.com/jackc/pgx/v4"
+)
+
+var (
+	ErrBatchAlreadyClosed = errors.New("batch already closed")
+)
+
+const usersB = `-- name: UsersB :batchmany
+SELECT id FROM "user"
+WHERE id = $1
+`
+
+type UsersBBatchResults struct {
+	br     pgx.BatchResults
+	tot    int
+	closed bool
+}
+
+func (q *Queries) UsersB(ctx context.Context, id []int64) *UsersBBatchResults {
+	batch := &pgx.Batch{}
+	for _, a := range id {
+		vals := []any{
+			a,
+		}
+		batch.Queue(usersB, vals...)
+	}
+	br := q.db.SendBatch(ctx, batch)
+	return &UsersBBatchResults{br, len(id), false}
+}
+
+func (b *UsersBBatchResults) Query(f func(int, []int64, error)) {
+	defer b.br.Close()
+	for t := 0; t < b.tot; t++ {
+		var items []int64
+		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 id int64
+				if err := rows.Scan(&id); err != nil {
+					return err
+				}
+				items = append(items, id)
+			}
+			return rows.Err()
+		}()
+		if f != nil {
+			f(t, items, err)
+		}
+	}
+}
+
+func (b *UsersBBatchResults) Close() error {
+	b.closed = true
+	return b.br.Close()
+}

internal/endtoend/testdata/output_file_names/pgx/v4/go/copyfrom_gen.go

@@ -0,0 +1,42 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: copyfrom_gen.go
+
+package querytest
+
+import (
+	"context"
+)
+
+// iteratorForUsersC implements pgx.CopyFromSource.
+type iteratorForUsersC struct {
+	rows                 []int64
+	skippedFirstNextCall bool
+}
+
+func (r *iteratorForUsersC) Next() bool {
+	if len(r.rows) == 0 {
+		return false
+	}
+	if !r.skippedFirstNextCall {
+		r.skippedFirstNextCall = true
+		return true
+	}
+	r.rows = r.rows[1:]
+	return len(r.rows) > 0
+}
+
+func (r iteratorForUsersC) Values() ([]any, error) {
+	return []any{
+		r.rows[0],
+	}, nil
+}
+
+func (r iteratorForUsersC) Err() error {
+	return nil
+}
+
+func (q *Queries) UsersC(ctx context.Context, id []int64) (int64, error) {
+	return q.db.CopyFrom(ctx, []string{"user"}, []string{"id"}, &iteratorForUsersC{rows: id})
+}

internal/endtoend/testdata/output_file_names/pgx/v4/go/db_gen.go

@@ -0,0 +1,34 @@
+// 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
+	CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error)
+	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/output_file_names/pgx/v4/go/models_gen.go

@@ -0,0 +1,9 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+type User struct {
+	ID int64
+}

internal/endtoend/testdata/output_file_names/pgx/v4/go/querier_gen.go

@@ -0,0 +1,17 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+)
+
+type Querier interface {
+	User(ctx context.Context) ([]int64, error)
+	UsersB(ctx context.Context, id []int64) *UsersBBatchResults
+	UsersC(ctx context.Context, id []int64) (int64, error)
+}
+
+var _ Querier = (*Queries)(nil)

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

@@ -0,0 +1,34 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+)
+
+const user = `-- name: User :many
+SELECT "user".id FROM "user"
+`
+
+func (q *Queries) User(ctx context.Context) ([]int64, error) {
+	rows, err := q.db.Query(ctx, user)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []int64
+	for rows.Next() {
+		var id int64
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		items = append(items, id)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}

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

@@ -0,0 +1,12 @@
+-- name: User :many
+SELECT "user".* FROM "user";
+
+-- name: UsersB :batchmany
+SELECT * FROM "user"
+WHERE id = $1;
+
+-- name: UsersC :copyfrom
+INSERT INTO "user"
+(id)
+VALUES
+($1);

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

@@ -0,0 +1,2 @@
+CREATE TABLE "user" (id bigserial not null);
+

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

@@ -0,0 +1,19 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v4",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql",
+      "emit_interface": true,
+      "output_batch_file_name": "batch_gen.go",
+      "output_db_file_name": "db_gen.go",
+      "output_models_file_name": "models_gen.go",
+      "output_querier_file_name": "querier_gen.go",
+      "output_copyfrom_file_name": "copyfrom_gen.go"
+    }
+  ]
+}