pgdb

type Config struct {
	// If set, URL will override other connection parameters
	//
	// Use this parameter to specify more advanced connection parameters such as SSL
	URL                              string    `yaml:"URL" env:"PG_URL"`
	Host                             string    `yaml:"Host" env:"PG_HOST"`
	Port                             int       `yaml:"Port" env:"PG_PORT"`
	Database                         string    `yaml:"Database" env:"PG_DATABASE"`
	Username                         string    `yaml:"Username" env:"PG_USERNAME"`
	Password                         string    `yaml:"Password" env:"PG_PASSWORD"`
	SSL                              string    `yaml:"SSL" env:"PG_SSL"`
	PoolParam                        PoolParam `yaml:"PoolParam"`
	MigrationStatementTimeoutSeconds int       `yaml:"MigrationStatementTimeoutSeconds" env:"PG_MIGRATION_STATEMENT_TIMEOUT_SECONDS"`
	MigrationFromEmbedded            bool      `yaml:"MigrationFromEmbedded" env:"PG_MIGRATION_FROM_EMBEDDED"`
	MigrationDirectory               string    `yaml:"MigrationDirectory" env:"PG_MIGRATION_DIRECTORY"`
}

type EnhancedDbPool struct{ *pgxpool.Pool }

type PoolParam struct {
	// Non-zero values will be passed to DbPool initialization. Defaults to 0.
	MinConns int `yaml:"MinConns" env:"pg_min_conns"`
	// Non-zero values will be passed to DbPool initialization. Defaults to max(numCPU, 4)
	MaxConns int `yaml:"MaxConns" env:"pg_max_conns"`
	// Non-zero values will be passed to DbPool initialization. Set with time units (s, m, h). Defaults to 2 mins.
	ConnectTimeout string `yaml:"ConnectTimeout" env:"pg_connect_timeout"`
	// Non-zero values will be passed to DbPool initialization. Set with time units (s, m, h). Defaults to 30 mins.
	MaxConnIdleTime string `yaml:"MaxConnIdleTime" env:"pg_max_conn_idle_time"`
	// Non-zero values will be passed to DbPool initialization. Set with time units (s, m, h). Defaults to 1 hour.
	MaxConnLifetime string `yaml:"MaxConnLifetime" env:"pg_max_conn_lifetime"`
	// Non-zero values will be passed to DbPool initialization. Set with time units (s, m, h). Defaults to 60s.
	HealthCheckPeriod string `yaml:"HealthCheckPeriod" env:"pg_health_check_period"`
}

type QueryExecer interface {
	Exec(ctx context.Context, sql string, arguments ...any) (commandTag pgconn.CommandTag, err error)
	Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
	QueryRow(ctx context.Context, sql string, args ...any) pgx.Row
}