db

package

v0.0.1 Latest Latest Go to latest Published: Dec 19, 2025 License: MIT Imports: 14 Imported by: 0

Details

Package db provides database connection management.

This section is empty.

This section is empty.

func NewMongoDB(ctx context.Context, cfg config.MongoConfig) (*mongo.Database, error)

NewMongoDB creates a MongoDB connection. ctx is used for connection timeout.

func NewMySQL(cfg config.MysqlConfig, logOutput io.Writer) (*gorm.DB, error)

NewMySQL creates a MySQL connection with connection pooling. logOutput is the writer for SQL logs (use io.Discard to suppress).

func NewRedis(cfg config.RedisConfig) *redis.Client

NewRedis creates a Redis client.

type KafkaConfig struct {
	config.KafkaConfig
	ClientID string `json:"client_id" yaml:"client_id" toml:"client_id"`
}

KafkaConfig extends the base config for Kafka.

type Manager struct {
	MySQL   *gorm.DB
	Redis   *redis.Client
	MongoDB *mongo.Database
	Config  *config.BaseConfig
}

Manager manages all database connections.

func NewManager(ctx context.Context, cfg config.BaseConfig) (*Manager, error)

NewManager creates a new database manager. ctx is used for connection timeouts during initialization.

func (m *Manager) Health(ctx context.Context) map[string]error

Health returns health status of all components.

func (m *Manager) IsEnabled(component string) bool

IsEnabled checks if a component is enabled.

func (m *Manager) MySQLWithContext(ctx context.Context) *gorm.DB

MySQLWithContext returns MySQL with context.

func (m *Manager) Start(ctx context.Context) error

Start checks connectivity of all databases.

func (m *Manager) Stop(ctx context.Context) error

Stop closes all database connections.