60 lines
1.0 KiB
Go
60 lines
1.0 KiB
Go
package database
|
|
|
|
import (
|
|
"github.com/mitchellh/mapstructure"
|
|
"matrix-middle-service/pkg/config"
|
|
"matrix-middle-service/pkg/logger"
|
|
)
|
|
|
|
var SupportDataSources = make(map[string]InitDSFn)
|
|
var drivers []DataBase
|
|
|
|
type InitDSFn = func(conf config.DataSourcesType[any]) DataBase
|
|
|
|
type DataBase interface {
|
|
Open() error
|
|
Close() error
|
|
}
|
|
|
|
func Init(conf []interface{}) {
|
|
drivers = make([]DataBase, 0, len(SupportDataSources))
|
|
|
|
logger.Log().Debug("[数据源] 初始化")
|
|
for _, item := range conf {
|
|
dst := config.DataSourcesType[any]{}
|
|
_ = mapstructure.Decode(item, &dst)
|
|
|
|
fn, ok := SupportDataSources[dst.Type]
|
|
if !ok {
|
|
logger.Log().Fatalf("不支持的数据源类型 %s", dst.Type)
|
|
}
|
|
|
|
if !dst.Enable {
|
|
continue
|
|
}
|
|
|
|
drivers = append(drivers, fn(dst))
|
|
}
|
|
}
|
|
|
|
func Register(name string, fn InitDSFn) {
|
|
SupportDataSources[name] = fn
|
|
}
|
|
|
|
func Open() (err error) {
|
|
for _, driver := range drivers {
|
|
err = driver.Open()
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func Close() {
|
|
for _, driver := range drivers {
|
|
_ = driver.Close()
|
|
}
|
|
return
|
|
}
|