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

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

@@ -0,0 +1,78 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: copyfrom.go
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+// iteratorForInsertSingleValue implements pgx.CopyFromSource.
+type iteratorForInsertSingleValue struct {
+	rows                 []sql.NullString
+	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 []sql.NullString) (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/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
+	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/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/copyfrom/postgresql/pgx/v4/go/querier.go

@@ -0,0 +1,19 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+type Querier interface {
+	// InsertSingleValue inserts a single value using copy.
+	InsertSingleValue(ctx context.Context, a []sql.NullString) (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/v4/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 (
+	"database/sql"
+)
+
+type InsertValuesParams struct {
+	A sql.NullString
+	B sql.NullInt32
+}