package mysql

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	. "gorm.io/gorm/logger"
	"strings"
	"time"
)

type GormInstance struct {
	DB *gorm.DB
	Config DataSource
}

func InitGormDatasource(datasource []DataSource, isUseSSH bool, loggingLevel string) []GormInstance {
	dbCh := make(chan DBInstance, len(datasource))
	InitDatasource(datasource, isUseSSH, dbCh)
	var gormInstance []GormInstance
	for ins := range dbCh {
		ins.DB.SetConnMaxLifetime(time.Second * 120)
		gormDB, err := gorm.Open(mysql.New(mysql.Config{
			Conn: ins.DB,
		}), &gorm.Config{
			Logger: Default.LogMode(gormLogLevelAdapter(loggingLevel)),
		})
		if err != nil {
			log.Error("Failed to facade db to gormDB, caused by:", err)
		}else {
			gormInstance = append(gormInstance, GormInstance{
				DB:     gormDB,
				Config: ins.Config,
			})
		}
	}
	return gormInstance
}

func gormLogLevelAdapter(level string) LogLevel {
	switch strings.ToLower(level) {
	case "panic":
		return Silent
	case "fatal", "error":
		return Error
	case "warn", "warning":
		return Warn
	default:
		return Info
	}
}