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

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/go/copyfrom.go

@@ -0,0 +1,79 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: copyfrom.go
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+// iteratorForInsertSingleValue implements pgx.CopyFromSource.
+type iteratorForInsertSingleValue struct {
+	rows                 []pgtype.Text
+	skippedFirstNextCall bool
+}
+
+func (r *iteratorForInsertSingleValue) 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 iteratorForInsertSingleValue) Values() ([]any, error) {
+	return []any{
+		r.rows[0],
+	}, nil
+}
+
+func (r iteratorForInsertSingleValue) Err() error {
+	return nil
+}
+
+// InsertSingleValue inserts a single value using copy.
+func (q *Queries) InsertSingleValue(ctx context.Context, a []pgtype.Text) (int64, error) {
+	return q.db.CopyFrom(ctx, []string{"myschema", "foo"}, []string{"a"}, &iteratorForInsertSingleValue{rows: a})
+}
+
+// iteratorForInsertValues implements pgx.CopyFromSource.
+type iteratorForInsertValues struct {
+	rows                 []InsertValuesParams
+	skippedFirstNextCall bool
+}
+
+func (r *iteratorForInsertValues) 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 iteratorForInsertValues) Values() ([]any, error) {
+	return []any{
+		r.rows[0].A,
+		r.rows[0].B,
+	}, nil
+}
+
+func (r iteratorForInsertValues) Err() error {
+	return nil
+}
+
+// InsertValues inserts multiple values using copy.
+func (q *Queries) InsertValues(ctx context.Context, arg []InsertValuesParams) (int64, error) {
+	return q.db.CopyFrom(ctx, []string{"myschema", "foo"}, []string{"a", "b"}, &iteratorForInsertValues{rows: arg})
+}

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/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/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
+	CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error)
+}
+
+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/copyfrom/postgresql/pgx/v5/go/models.go

@@ -0,0 +1,14 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type MyschemaFoo struct {
+	A pgtype.Text
+	B pgtype.Int4
+}

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/go/querier.go

@@ -0,0 +1,20 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type Querier interface {
+	// InsertSingleValue inserts a single value using copy.
+	InsertSingleValue(ctx context.Context, a []pgtype.Text) (int64, error)
+	// InsertValues inserts multiple values using copy.
+	InsertValues(ctx context.Context, arg []InsertValuesParams) (int64, error)
+}
+
+var _ Querier = (*Queries)(nil)

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/go/query.sql.go

@@ -0,0 +1,15 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type InsertValuesParams struct {
+	A pgtype.Text
+	B pgtype.Int4
+}

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/query.sql

@@ -0,0 +1,7 @@
+-- name: InsertValues :copyfrom
+-- InsertValues inserts multiple values using copy.
+INSERT INTO myschema.foo (a, b) VALUES ($1, $2);
+
+-- name: InsertSingleValue :copyfrom
+-- InsertSingleValue inserts a single value using copy.
+INSERT INTO myschema.foo (a) VALUES ($1);

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/schema.sql

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

internal/endtoend/testdata/copyfrom/postgresql/pgx/v5/sqlc.json

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