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

internal/endtoend/testdata/enum/postgresql/pgx/v4/go/db.go

@@ -0,0 +1,25 @@
+// 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
+}
+
+func New() *Queries {
+	return &Queries{}
+}
+
+type Queries struct {
+}

internal/endtoend/testdata/enum/postgresql/pgx/v4/go/models.go

@@ -0,0 +1,65 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"database/sql"
+	"database/sql/driver"
+	"fmt"
+)
+
+type Size string
+
+const (
+	SizeXSmall Size = "x-small"
+	SizeSmall  Size = "small"
+	SizeMedium Size = "medium"
+	SizeLarge  Size = "large"
+	SizeXLarge Size = "x-large"
+)
+
+func (e *Size) Scan(src any) error {
+	switch s := src.(type) {
+	case []byte:
+		*e = Size(s)
+	case string:
+		*e = Size(s)
+	default:
+		return fmt.Errorf("unsupported scan type for Size: %T", src)
+	}
+	return nil
+}
+
+type NullSize struct {
+	Size  Size
+	Valid bool // Valid is true if Size is not NULL
+}
+
+// Scan implements the Scanner interface.
+func (ns *NullSize) Scan(value any) error {
+	if value == nil {
+		ns.Size, ns.Valid = "", false
+		return nil
+	}
+	ns.Valid = true
+	return ns.Size.Scan(value)
+}
+
+// Value implements the driver Valuer interface.
+func (ns NullSize) Value() (driver.Value, error) {
+	if !ns.Valid {
+		return nil, nil
+	}
+	return string(ns.Size), nil
+}
+
+type User struct {
+	ID        int32
+	FirstName string
+	LastName  sql.NullString
+	Age       int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}

internal/endtoend/testdata/enum/postgresql/pgx/v4/go/query.sql.go

@@ -0,0 +1,104 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+	"database/sql"
+)
+
+const deleteBySize = `-- name: DeleteBySize :exec
+DELETE FROM users
+WHERE shoe_size = $1 AND shirt_size = $2
+`
+
+type DeleteBySizeParams struct {
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) DeleteBySize(ctx context.Context, db DBTX, arg DeleteBySizeParams) error {
+	_, err := db.Exec(ctx, deleteBySize, arg.ShoeSize, arg.ShirtSize)
+	return err
+}
+
+const getAll = `-- name: GetAll :many
+SELECT id, first_name, last_name, age, shoe_size, shirt_size FROM users
+`
+
+func (q *Queries) GetAll(ctx context.Context, db DBTX) ([]User, error) {
+	rows, err := db.Query(ctx, getAll)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []User
+	for rows.Next() {
+		var i User
+		if err := rows.Scan(
+			&i.ID,
+			&i.FirstName,
+			&i.LastName,
+			&i.Age,
+			&i.ShoeSize,
+			&i.ShirtSize,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const newUser = `-- name: NewUser :exec
+INSERT INTO users (
+    first_name,
+    last_name,
+    age,
+    shoe_size,
+    shirt_size
+) VALUES
+($1, $2, $3, $4, $5)
+`
+
+type NewUserParams struct {
+	FirstName string
+	LastName  sql.NullString
+	Age       int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) NewUser(ctx context.Context, db DBTX, arg NewUserParams) error {
+	_, err := db.Exec(ctx, newUser,
+		arg.FirstName,
+		arg.LastName,
+		arg.Age,
+		arg.ShoeSize,
+		arg.ShirtSize,
+	)
+	return err
+}
+
+const updateSizes = `-- name: UpdateSizes :exec
+UPDATE users
+SET shoe_size = $2, shirt_size = $3
+WHERE id = $1
+`
+
+type UpdateSizesParams struct {
+	ID        int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) UpdateSizes(ctx context.Context, db DBTX, arg UpdateSizesParams) error {
+	_, err := db.Exec(ctx, updateSizes, arg.ID, arg.ShoeSize, arg.ShirtSize)
+	return err
+}

internal/endtoend/testdata/enum/postgresql/pgx/v4/query.sql

@@ -0,0 +1,21 @@
+-- name: GetAll :many
+SELECT * FROM users;
+
+-- name: NewUser :exec
+INSERT INTO users (
+    first_name,
+    last_name,
+    age,
+    shoe_size,
+    shirt_size
+) VALUES
+($1, $2, $3, $4, $5);
+
+-- name: UpdateSizes :exec
+UPDATE users
+SET shoe_size = $2, shirt_size = $3
+WHERE id = $1;
+
+-- name: DeleteBySize :exec
+DELETE FROM users
+WHERE shoe_size = $1 AND shirt_size = $2;

internal/endtoend/testdata/enum/postgresql/pgx/v4/schema.sql

@@ -0,0 +1,11 @@
+CREATE TYPE size AS ENUM('x-small', 'small', 'medium', 'large', 'x-large');
+
+
+CREATE TABLE users (
+    id integer NOT NULL GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+    first_name varchar(255) NOT NULL,
+    last_name varchar(255),
+    age integer NOT NULL,
+    shoe_size size NOT NULL,
+    shirt_size size
+);

internal/endtoend/testdata/enum/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_methods_with_db_argument": true
+    }
+  ]
+}

internal/endtoend/testdata/enum/postgresql/pgx/v5/go/db.go

@@ -0,0 +1,25 @@
+// 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
+}
+
+func New() *Queries {
+	return &Queries{}
+}
+
+type Queries struct {
+}

internal/endtoend/testdata/enum/postgresql/pgx/v5/go/models.go

@@ -0,0 +1,66 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+
+package querytest
+
+import (
+	"database/sql/driver"
+	"fmt"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+type Size string
+
+const (
+	SizeXSmall Size = "x-small"
+	SizeSmall  Size = "small"
+	SizeMedium Size = "medium"
+	SizeLarge  Size = "large"
+	SizeXLarge Size = "x-large"
+)
+
+func (e *Size) Scan(src any) error {
+	switch s := src.(type) {
+	case []byte:
+		*e = Size(s)
+	case string:
+		*e = Size(s)
+	default:
+		return fmt.Errorf("unsupported scan type for Size: %T", src)
+	}
+	return nil
+}
+
+type NullSize struct {
+	Size  Size
+	Valid bool // Valid is true if Size is not NULL
+}
+
+// Scan implements the Scanner interface.
+func (ns *NullSize) Scan(value any) error {
+	if value == nil {
+		ns.Size, ns.Valid = "", false
+		return nil
+	}
+	ns.Valid = true
+	return ns.Size.Scan(value)
+}
+
+// Value implements the driver Valuer interface.
+func (ns NullSize) Value() (driver.Value, error) {
+	if !ns.Valid {
+		return nil, nil
+	}
+	return string(ns.Size), nil
+}
+
+type User struct {
+	ID        int32
+	FirstName string
+	LastName  pgtype.Text
+	Age       int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}

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

@@ -0,0 +1,105 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: query.sql
+
+package querytest
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+const deleteBySize = `-- name: DeleteBySize :exec
+DELETE FROM users
+WHERE shoe_size = $1 AND shirt_size = $2
+`
+
+type DeleteBySizeParams struct {
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) DeleteBySize(ctx context.Context, db DBTX, arg DeleteBySizeParams) error {
+	_, err := db.Exec(ctx, deleteBySize, arg.ShoeSize, arg.ShirtSize)
+	return err
+}
+
+const getAll = `-- name: GetAll :many
+SELECT id, first_name, last_name, age, shoe_size, shirt_size FROM users
+`
+
+func (q *Queries) GetAll(ctx context.Context, db DBTX) ([]User, error) {
+	rows, err := db.Query(ctx, getAll)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []User
+	for rows.Next() {
+		var i User
+		if err := rows.Scan(
+			&i.ID,
+			&i.FirstName,
+			&i.LastName,
+			&i.Age,
+			&i.ShoeSize,
+			&i.ShirtSize,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const newUser = `-- name: NewUser :exec
+INSERT INTO users (
+    first_name,
+    last_name,
+    age,
+    shoe_size,
+    shirt_size
+) VALUES
+($1, $2, $3, $4, $5)
+`
+
+type NewUserParams struct {
+	FirstName string
+	LastName  pgtype.Text
+	Age       int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) NewUser(ctx context.Context, db DBTX, arg NewUserParams) error {
+	_, err := db.Exec(ctx, newUser,
+		arg.FirstName,
+		arg.LastName,
+		arg.Age,
+		arg.ShoeSize,
+		arg.ShirtSize,
+	)
+	return err
+}
+
+const updateSizes = `-- name: UpdateSizes :exec
+UPDATE users
+SET shoe_size = $2, shirt_size = $3
+WHERE id = $1
+`
+
+type UpdateSizesParams struct {
+	ID        int32
+	ShoeSize  Size
+	ShirtSize NullSize
+}
+
+func (q *Queries) UpdateSizes(ctx context.Context, db DBTX, arg UpdateSizesParams) error {
+	_, err := db.Exec(ctx, updateSizes, arg.ID, arg.ShoeSize, arg.ShirtSize)
+	return err
+}

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

@@ -0,0 +1,21 @@
+-- name: GetAll :many
+SELECT * FROM users;
+
+-- name: NewUser :exec
+INSERT INTO users (
+    first_name,
+    last_name,
+    age,
+    shoe_size,
+    shirt_size
+) VALUES
+($1, $2, $3, $4, $5);
+
+-- name: UpdateSizes :exec
+UPDATE users
+SET shoe_size = $2, shirt_size = $3
+WHERE id = $1;
+
+-- name: DeleteBySize :exec
+DELETE FROM users
+WHERE shoe_size = $1 AND shirt_size = $2;

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

@@ -0,0 +1,11 @@
+CREATE TYPE size AS ENUM('x-small', 'small', 'medium', 'large', 'x-large');
+
+
+CREATE TABLE users (
+    id integer NOT NULL GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+    first_name varchar(255) NOT NULL,
+    last_name varchar(255),
+    age integer NOT NULL,
+    shoe_size size NOT NULL,
+    shirt_size size
+);

internal/endtoend/testdata/enum/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_methods_with_db_argument": true
+    }
+  ]
+}