db

package

v1.0.5 Latest Latest Go to latest Published: Sep 28, 2025 License: Apache-2.0 Imports: 16 Imported by: 0

Details

This section is empty.

This section is empty.

func AfterCreate(db *gorm.DB)

func MongoClient(cfg configs.Mongodb) *mongo.Database

func NewKafkaProducer(cfg configs.KafkaConfig) (sarama.SyncProducer, error)

func NewMysql(cfg configs.MysqlConfig) *gorm.DB

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

type ComponentStatus struct {
	Enabled bool
	Error   error
}

ComponentStatus 组件状态

type DataSource struct {
	Mysql   *gorm.DB
	Mongodb *mongo.Database
	Redis   *redis.Client
	Kafka   sarama.SyncProducer
}

DataSource 统一的数据源，包含所有数据库组件

func NewDataSource(lc fx.Lifecycle, cfg configs.Config) *DataSource

NewDataSource 根据配置初始化数据源

func (ds *DataSource) GetComponentStatus(ctx context.Context) map[string]ComponentStatus

GetComponentStatus 获取所有组件状态

func (ds *DataSource) Stop(ctx context.Context) error

Stop 停止所有组件