postgres

type Config struct {
	URL                      string
	MaxConnection            int32
	MinConnection            int32
	MaxConnectionIdleTime    time.Duration
	MaxConnectionLifetime    time.Duration
	HealthCheckPeriod        time.Duration
	ConnectTimeout           time.Duration
	LogLevel                 tracelog.LogLevel
	StatementTimeout         time.Duration
	LockTimeout              time.Duration
	IdleInTransactionTimeout time.Duration
}

type DB interface {
	Executor
	Health
	Lifecycle
	TxRunner
	GetPoolStats() (*PoolStats, error)
}

type DBPool interface {
	pgxscan.Querier
	SendBatch(ctx context.Context, b *pgx.Batch) pgx.BatchResults
	Begin(ctx context.Context) (pgx.Tx, error)
	BeginTx(ctx context.Context, txOptions pgx.TxOptions) (pgx.Tx, error)
	Exec(ctx context.Context, sql string, arguments ...any) (pgconn.CommandTag, error)
	CopyFrom(ctx context.Context, tableName pgx.Identifier, columnNames []string, rowSrc pgx.CopyFromSource) (int64, error)
	Ping(ctx context.Context) error
	Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
	QueryRow(ctx context.Context, sql string, args ...any) pgx.Row
	Close()
}

type Executor interface {
	DBPool
	BulkInsert(ctx context.Context, tableName string, columns []string, rows [][]any) (int64, error)
}

type Health interface {
	IsHealthy(ctx context.Context) bool
	HealthCheck(ctx context.Context, opts ...HealthCheckOption) error
}

type HealthCheckOption func(*HealthCheckOptions)

type HealthCheckOptions struct {
	Timeout              time.Duration
	RequireActiveConns   bool
	MinIdleConns         int32
	CheckQueryExecution  bool
	CustomHealthCheckSQL string
}

type Lifecycle interface {
	Name() string
	Start(ctx context.Context) error
	Stop() error
}

type MigrationVersion struct {
	Version uint
	Dirty   bool
}

type PoolStats struct {
	AcquireCount            int64
	AcquireDuration         time.Duration
	AcquiredConns           int32
	CanceledAcquireCount    int64
	ConstructingConns       int32
	EmptyAcquireCount       int64
	IdleConns               int32
	MaxConns                int32
	TotalConns              int32
	NewConnsCount           int64
	MaxLifetimeDestroyCount int64
	MaxIdleDestroyCount     int64
}

type Postgres struct {
	DBPool
}

type RetryConfig struct {
	MaxRetries    int
	InitialDelay  time.Duration
	MaxDelay      time.Duration
	Multiplier    float64
	RetryableErrs []string
}

type RetryableFunc func(ctx context.Context) error

type TxFunc func(ctx context.Context, tx pgx.Tx) error

type TxRunner interface {
	WithTransaction(ctx context.Context, fn TxFunc) error
	WithTransactionOptions(ctx context.Context, txOptions pgx.TxOptions, fn TxFunc) error
	WithReadOnlyTransaction(ctx context.Context, fn TxFunc) error
	WithSerializableTransaction(ctx context.Context, fn TxFunc) error
	WithRepeatableReadTransaction(ctx context.Context, fn TxFunc) error
	WithRetryTx(ctx context.Context, config RetryConfig, fn TxFunc) error
	WithRetryTxDefault(ctx context.Context, fn TxFunc) error
}