package database

import (
	"gitee.com/taotechip/modules/logger"
	"github.com/jinzhu/configor"
	"gorm.io/gorm"
)

var connMaps = make(map[string]*gorm.DB)

var dialectors = map[string]func(map[string]string) (gorm.Dialector, error){
	"mysql":  MysqlDialector,
	"oracle": OracleDialector,
}

var configs = make([]struct {
	Name    string
	Type    string
	Default bool
	Config  map[string]string
}, 0)

func init() {
	configor.Load(&configs, "conf/database.yml")
	logger.Info("读取%d个数据源", len(configs))
	var defaultConn *gorm.DB
	for i, cfg := range configs {
		dialector, ok := dialectors[cfg.Type]
		if !ok {
			logger.Warn("不支持的数据库类型[%s]", cfg.Type)
			continue
		}
		dialect, err := dialector(cfg.Config)
		if err != nil {
			logger.Error("[%s]%s", cfg.Name, err.Error())
			continue
		}

		dbConn, err := gorm.Open(dialect, &gorm.Config{})
		if err != nil {
			logger.Error("[%s]数据库连接错误 %s", cfg.Name, err.Error())
			continue
		}
		connMaps[cfg.Name] = dbConn
		if cfg.Default {
			defaultConn = dbConn
		}
		if i == 0 {
			defaultConn = dbConn
		}
	}
	connMaps["default"] = defaultConn
}

func GetConn() (conn *gorm.DB) {
	return GetConnWithName("default")
}

func GetConnWithName(name string) (conn *gorm.DB) {
	return connMaps[name].Session(&gorm.Session{})
}