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

internal/endtoend/testdata/copyfrom/mysql/go/copyfrom.go

@@ -0,0 +1,88 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: copyfrom.go
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"io"
+	"sync/atomic"
+
+	"github.com/go-sql-driver/mysql"
+	"github.com/hexon/mysqltsv"
+)
+
+var readerHandlerSequenceForInsertSingleValue uint32 = 1
+
+func convertRowsForInsertSingleValue(w *io.PipeWriter, a []sql.NullString) {
+	e := mysqltsv.NewEncoder(w, 1, nil)
+	for _, row := range a {
+		e.AppendValue(row)
+	}
+	w.CloseWithError(e.Close())
+}
+
+// InsertSingleValue uses MySQL's LOAD DATA LOCAL INFILE and is not atomic.
+//
+// Errors and duplicate keys are treated as warnings and insertion will
+// continue, even without an error for some cases.  Use this in a transaction
+// and use SHOW WARNINGS to check for any problems and roll back if you want to.
+//
+// Check the documentation for more information:
+// https://dev.mysql.com/doc/refman/8.0/en/load-data.html#load-data-error-handling
+func (q *Queries) InsertSingleValue(ctx context.Context, a []sql.NullString) (int64, error) {
+	pr, pw := io.Pipe()
+	defer pr.Close()
+	rh := fmt.Sprintf("InsertSingleValue_%d", atomic.AddUint32(&readerHandlerSequenceForInsertSingleValue, 1))
+	mysql.RegisterReaderHandler(rh, func() io.Reader { return pr })
+	defer mysql.DeregisterReaderHandler(rh)
+	go convertRowsForInsertSingleValue(pw, a)
+	// The string interpolation is necessary because LOAD DATA INFILE requires
+	// the file name to be given as a literal string.
+	result, err := q.db.ExecContext(ctx, fmt.Sprintf("LOAD DATA LOCAL INFILE '%s' INTO TABLE `foo` %s (a)", "Reader::"+rh, mysqltsv.Escaping))
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}
+
+var readerHandlerSequenceForInsertValues uint32 = 1
+
+func convertRowsForInsertValues(w *io.PipeWriter, arg []InsertValuesParams) {
+	e := mysqltsv.NewEncoder(w, 4, nil)
+	for _, row := range arg {
+		e.AppendValue(row.A)
+		e.AppendValue(row.B)
+		e.AppendValue(row.C)
+		e.AppendValue(row.D)
+	}
+	w.CloseWithError(e.Close())
+}
+
+// InsertValues uses MySQL's LOAD DATA LOCAL INFILE and is not atomic.
+//
+// Errors and duplicate keys are treated as warnings and insertion will
+// continue, even without an error for some cases.  Use this in a transaction
+// and use SHOW WARNINGS to check for any problems and roll back if you want to.
+//
+// Check the documentation for more information:
+// https://dev.mysql.com/doc/refman/8.0/en/load-data.html#load-data-error-handling
+func (q *Queries) InsertValues(ctx context.Context, arg []InsertValuesParams) (int64, error) {
+	pr, pw := io.Pipe()
+	defer pr.Close()
+	rh := fmt.Sprintf("InsertValues_%d", atomic.AddUint32(&readerHandlerSequenceForInsertValues, 1))
+	mysql.RegisterReaderHandler(rh, func() io.Reader { return pr })
+	defer mysql.DeregisterReaderHandler(rh)
+	go convertRowsForInsertValues(pw, arg)
+	// The string interpolation is necessary because LOAD DATA INFILE requires
+	// the file name to be given as a literal string.
+	result, err := q.db.ExecContext(ctx, fmt.Sprintf("LOAD DATA LOCAL INFILE '%s' INTO TABLE `foo` %s (a, b, c, d)", "Reader::"+rh, mysqltsv.Escaping))
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected()
+}

internal/endtoend/testdata/copyfrom/mysql/go/db.go

@@ -0,0 +1,31 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+type DBTX interface {
+	ExecContext(context.Context, string, ...any) (sql.Result, error)
+	PrepareContext(context.Context, string) (*sql.Stmt, error)
+	QueryContext(context.Context, string, ...any) (*sql.Rows, error)
+	QueryRowContext(context.Context, string, ...any) *sql.Row
+}
+
+func New(db DBTX) *Queries {
+	return &Queries{db: db}
+}
+
+type Queries struct {
+	db DBTX
+}
+
+func (q *Queries) WithTx(tx *sql.Tx) *Queries {
+	return &Queries{
+		db: tx,
+	}
+}

internal/endtoend/testdata/copyfrom/mysql/go/models.go

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

internal/endtoend/testdata/copyfrom/mysql/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 (
+	"database/sql"
+)
+
+const insertSingleValue = `-- name: InsertSingleValue :copyfrom
+INSERT INTO foo (a) VALUES (?)
+`
+
+const insertValues = `-- name: InsertValues :copyfrom
+INSERT INTO foo (a, b, c, d) VALUES (?, ?, ?, ?)
+`
+
+type InsertValuesParams struct {
+	A sql.NullString
+	B sql.NullInt32
+	C sql.NullTime
+	D sql.NullTime
+}

internal/endtoend/testdata/copyfrom/mysql/query.sql

@@ -0,0 +1,5 @@
+-- name: InsertValues :copyfrom
+INSERT INTO foo (a, b, c, d) VALUES (?, ?, ?, ?);
+
+-- name: InsertSingleValue :copyfrom
+INSERT INTO foo (a) VALUES (?);

internal/endtoend/testdata/copyfrom/mysql/schema.sql

@@ -0,0 +1,2 @@
+CREATE TABLE foo (a text, b integer, c DATETIME, d DATE);
+

internal/endtoend/testdata/copyfrom/mysql/sqlc.json

@@ -0,0 +1,14 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "sql_package": "database/sql",
+      "sql_driver": "github.com/go-sql-driver/mysql",
+      "engine": "mysql",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

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
+}

internal/endtoend/testdata/copyfrom/postgresql/pgx/v4/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/v4/schema.sql

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

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

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

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
+    }
+  ]
+}