driver

type BatchItem struct {
	SQL  string
	Args []any
}

type BatchResults interface {
	// Query returns the rows for the next queued query.
	Query() (Rows, error)
	// Close releases the batch results.
	Close() error
}

type BulkCopier interface {
	CopyFrom(ctx context.Context, table string, columns []string, rows [][]any) (int64, error)
}

type Driver interface {
	QueryRow(ctx context.Context, sql string, args ...any) Row
	Query(ctx context.Context, sql string, args ...any) (Rows, error)
	Exec(ctx context.Context, sql string, args ...any) (int64, error)
	Begin(ctx context.Context) (Tx, error)
	BeginTx(ctx context.Context, opts TxOptions) (Tx, error)
	// Ping verifies a working connection to the database, used by health probes.
	Ping(ctx context.Context) error
	// Stat returns a snapshot of connection-pool utilisation.
	Stat() PoolStat
	Close()
}

type IsolationLevel int

type Pipeliner interface {
	SendBatch(ctx context.Context, items []BatchItem) (BatchResults, error)
}

type PoolConfig struct {
	MaxConns        int           // maximum open connections
	ConnMaxLifetime time.Duration // recycle connections older than this
	ConnMaxIdleTime time.Duration // close connections idle longer than this
	// SimpleProtocol, when true, makes the Postgres driver use pgx's simple
	// query protocol (no server-side prepared statements) for compatibility with
	// PgBouncer transaction/statement pooling. Ignored by SQLite/libSQL.
	SimpleProtocol bool
}

type PoolStat struct {
	MaxConns      int32 // configured maximum open connections (0 = driver default / unlimited)
	AcquiredConns int32 // connections currently checked out / in use
	IdleConns     int32 // connections currently idle in the pool
	TotalConns    int32 // total connections currently held (acquired + idle + constructing)
}

type Row interface {
	Scan(dest ...any) error
}

type Rows interface {
	Next() bool
	Scan(dest ...any) error
	Close()
	Err() error
}

type Tx interface {
	QueryRow(ctx context.Context, sql string, args ...any) Row
	Query(ctx context.Context, sql string, args ...any) (Rows, error)
	Exec(ctx context.Context, sql string, args ...any) (int64, error)
	Commit(ctx context.Context) error
	Rollback(ctx context.Context) error
}

type TxBulkCopier interface {
	CopyFrom(ctx context.Context, table string, columns []string, rows [][]any) (int64, error)
}

type TxOptions struct {
	Isolation IsolationLevel
	ReadOnly  bool
}