database

type BaseBeginner interface {
	Begin(context.Context) (sql.Tx, error)
}

type BaseQuerier interface {
	WithTx(tx *sql.Tx) BaseQuerier
}

type BaseQuerierTx interface {
	BaseQuerier
	Begin(ctx context.Context) (BaseQuerierTx, error)
	Commit(ctx context.Context) error
	Rollback(ctx context.Context) error
}

type BaseTx interface {
	BaseBeginner
	Commit(ctx context.Context) error
	Rollback(ctx context.Context) error
}

type Config struct {
	// MySQL configuration (use this OR Postgres, not both)
	MySQL *MySQLConfig `yaml:"mysql,omitempty"`
	// Postgres configuration (use this OR MySQL, not both)
	Postgres *PostgresConfig `yaml:"postgres,omitempty"`

	// Legacy flat PostgreSQL format (deprecated, for backward compatibility only)
	// If neither MySQL nor Postgres is set, these fields will be used for PostgreSQL
	User    string `yaml:"user,omitempty"`
	Pass    string `yaml:"pass,omitempty"`
	Host    string `yaml:"host,omitempty"`
	Port    uint16 `yaml:"port,omitempty"`
	Name    string `yaml:"name,omitempty"`
	SSLMode string `yaml:"sslmode,omitempty"`
}

type ConfigOptions struct {
	// ConfigFiles is a list of config file paths to search for database configuration
	ConfigFiles []string

	// EnablePoolMonitoring enables connection pool metrics collection
	EnablePoolMonitoring bool

	// PrometheusRegisterer for metrics collection. If nil, no metrics are collected.
	PrometheusRegisterer prometheus.Registerer

	// Connection pool settings
	MaxOpenConns    int
	MaxIdleConns    int
	ConnMaxLifetime time.Duration
}

type CreateConnectorFunc func() (driver.Connector, error)

type DB[Q any] interface {
	Begin(ctx context.Context) (Q, error)
}

type DBConfig = MySQLConfig

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
}

type DatabaseMetrics struct {
	ConnectionsOpen         prometheus.Gauge
	ConnectionsIdle         prometheus.Gauge
	ConnectionsInUse        prometheus.Gauge
	ConnectionsWaitCount    prometheus.Counter
	ConnectionsWaitDuration prometheus.Histogram
	// contains filtered or unexported fields
}

type Driver struct {
	CreateConnectorFunc CreateConnectorFunc
}

type MySQLConfig struct {
	DSN    string `yaml:"dsn" default:"" flag:"dsn" usage:"Database DSN"`
	User   string `yaml:"user" default:"" flag:"user"`
	Pass   string `yaml:"pass" default:"" flag:"pass"`
	DBName string `yaml:"name,omitempty"` // Optional database name override
}

type PostgresConfig struct {
	User    string `yaml:"user"`
	Pass    string `yaml:"pass"`
	Host    string `yaml:"host"`
	Port    uint16 `yaml:"port"`
	Name    string `yaml:"name"`
	SSLMode string `yaml:"sslmode"`
}

type TX interface {
	Commit(ctx context.Context) error
	Rollback(ctx context.Context) error
}

type TransactionFunc[Q any] func(ctx context.Context, q Q) error