diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/go/db.go b/internal/endtoend/testdata/unnest/postgresql/pgx/go/db.go new file mode 100644 index 0000000000..6c6a7eb822 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/go/db.go @@ -0,0 +1,32 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 + +package querytest + +import ( + "context" + + "github.com/jackc/pgconn" + "github.com/jackc/pgx/v4" +) + +type DBTX interface { + Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error) + Query(context.Context, string, ...interface{}) (pgx.Rows, error) + QueryRow(context.Context, string, ...interface{}) pgx.Row +} + +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, + } +} diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/go/models.go b/internal/endtoend/testdata/unnest/postgresql/pgx/go/models.go new file mode 100644 index 0000000000..70640b2121 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/go/models.go @@ -0,0 +1,19 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 + +package querytest + +import ( + "database/sql" + "time" + + "github.com/google/uuid" +) + +type Memory struct { + ID uuid.UUID + VampireID uuid.UUID + CreatedAt time.Time + UpdatedAt sql.NullTime +} diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/go/query.sql.go b/internal/endtoend/testdata/unnest/postgresql/pgx/go/query.sql.go new file mode 100644 index 0000000000..1956798983 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/go/query.sql.go @@ -0,0 +1,45 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 +// source: query.sql + +package querytest + +import ( + "context" + + "github.com/google/uuid" +) + +const createMemories = `-- name: CreateMemories :many +INSERT INTO memories (vampire_id) +SELECT + unnest($1::uuid[]) AS vampire_id +RETURNING + id, vampire_id, created_at, updated_at +` + +func (q *Queries) CreateMemories(ctx context.Context, vampireID []uuid.UUID) ([]Memory, error) { + rows, err := q.db.Query(ctx, createMemories, vampireID) + if err != nil { + return nil, err + } + defer rows.Close() + var items []Memory + for rows.Next() { + var i Memory + if err := rows.Scan( + &i.ID, + &i.VampireID, + &i.CreatedAt, + &i.UpdatedAt, + ); err != nil { + return nil, err + } + items = append(items, i) + } + if err := rows.Err(); err != nil { + return nil, err + } + return items, nil +} diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/query.sql b/internal/endtoend/testdata/unnest/postgresql/pgx/query.sql new file mode 100644 index 0000000000..86deda3df4 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/query.sql @@ -0,0 +1,6 @@ +-- name: CreateMemories :many +INSERT INTO memories (vampire_id) +SELECT + unnest(@vampire_id::uuid[]) AS vampire_id +RETURNING + *; diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/schema.sql b/internal/endtoend/testdata/unnest/postgresql/pgx/schema.sql new file mode 100644 index 0000000000..e85e75f37a --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/schema.sql @@ -0,0 +1,7 @@ +CREATE TABLE memories ( + id uuid PRIMARY KEY DEFAULT gen_random_uuid (), + vampire_id uuid REFERENCES vampires (id) NOT NULL, + created_at timestamp NOT NULL DEFAULT NOW(), + updated_at timestamp +); + diff --git a/internal/endtoend/testdata/unnest/postgresql/pgx/sqlc.json b/internal/endtoend/testdata/unnest/postgresql/pgx/sqlc.json new file mode 100644 index 0000000000..d1244c9e7a --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/pgx/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" + } + ] +} diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/go/db.go b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/db.go new file mode 100644 index 0000000000..19557a6a5a --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/db.go @@ -0,0 +1,31 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 + +package querytest + +import ( + "context" + "database/sql" +) + +type DBTX interface { + ExecContext(context.Context, string, ...interface{}) (sql.Result, error) + PrepareContext(context.Context, string) (*sql.Stmt, error) + QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error) + QueryRowContext(context.Context, string, ...interface{}) *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, + } +} diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/go/models.go b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/models.go new file mode 100644 index 0000000000..70640b2121 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/models.go @@ -0,0 +1,19 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 + +package querytest + +import ( + "database/sql" + "time" + + "github.com/google/uuid" +) + +type Memory struct { + ID uuid.UUID + VampireID uuid.UUID + CreatedAt time.Time + UpdatedAt sql.NullTime +} diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/go/query.sql.go b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/query.sql.go new file mode 100644 index 0000000000..cac7ab4e55 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/go/query.sql.go @@ -0,0 +1,49 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.14.0 +// source: query.sql + +package querytest + +import ( + "context" + + "github.com/google/uuid" + "github.com/lib/pq" +) + +const createMemories = `-- name: CreateMemories :many +INSERT INTO memories (vampire_id) +SELECT + unnest($1::uuid[]) AS vampire_id +RETURNING + id, vampire_id, created_at, updated_at +` + +func (q *Queries) CreateMemories(ctx context.Context, vampireID []uuid.UUID) ([]Memory, error) { + rows, err := q.db.QueryContext(ctx, createMemories, pq.Array(vampireID)) + if err != nil { + return nil, err + } + defer rows.Close() + var items []Memory + for rows.Next() { + var i Memory + if err := rows.Scan( + &i.ID, + &i.VampireID, + &i.CreatedAt, + &i.UpdatedAt, + ); err != nil { + return nil, err + } + items = append(items, i) + } + if err := rows.Close(); err != nil { + return nil, err + } + if err := rows.Err(); err != nil { + return nil, err + } + return items, nil +} diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/query.sql b/internal/endtoend/testdata/unnest/postgresql/stdlib/query.sql new file mode 100644 index 0000000000..86deda3df4 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/query.sql @@ -0,0 +1,6 @@ +-- name: CreateMemories :many +INSERT INTO memories (vampire_id) +SELECT + unnest(@vampire_id::uuid[]) AS vampire_id +RETURNING + *; diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/schema.sql b/internal/endtoend/testdata/unnest/postgresql/stdlib/schema.sql new file mode 100644 index 0000000000..1780ad7ad3 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/schema.sql @@ -0,0 +1,6 @@ +CREATE TABLE memories ( + id uuid PRIMARY KEY DEFAULT gen_random_uuid (), + vampire_id uuid REFERENCES vampires (id) NOT NULL, + created_at timestamp NOT NULL DEFAULT NOW(), + updated_at timestamp +); diff --git a/internal/endtoend/testdata/unnest/postgresql/stdlib/sqlc.json b/internal/endtoend/testdata/unnest/postgresql/stdlib/sqlc.json new file mode 100644 index 0000000000..f717ca2e66 --- /dev/null +++ b/internal/endtoend/testdata/unnest/postgresql/stdlib/sqlc.json @@ -0,0 +1,12 @@ +{ + "version": "1", + "packages": [ + { + "path": "go", + "engine": "postgresql", + "name": "querytest", + "schema": "schema.sql", + "queries": "query.sql" + } + ] +}