package cDB

import (
	"gitee.com/csingo/cHelper"
	"gitee.com/csingo/cLog"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
	"time"
)

type MysqlConnector struct{}

func (i *MysqlConnector) Connect(conf DatabaseConfDriverInterface) *gorm.DB {
	var driverConf *DatabaseConf_MySql
	var ok bool

	if driverConf, ok = conf.(*DatabaseConf_MySql); !ok {
		return nil
	}

	dsn := driverConf.Username + ":" +
		driverConf.Password + "@tcp(" +
		driverConf.Host + ":" +
		cHelper.ToString(driverConf.Port) + ")/" +
		driverConf.Database + "?charset=" +
		driverConf.Charset + "&parseTime=True"
	cLog.WithContext(nil, map[string]interface{}{
		"source": "cDB.MysqlConnector.Connect",
		"dsn":    dsn,
	}).Debug("创建数据库连接")

	conn := mysql.Open(dsn)
	db, err := gorm.Open(conn, &gorm.Config{
		//Logger: logger.Default.LogMode(logger.Info),
		NamingStrategy: schema.NamingStrategy{
			SingularTable: true, // 使用单数表名
		},
	})
	if err != nil {
		cLog.WithContext(nil, map[string]interface{}{
			"source": "cDB.MysqlConnector.Connect",
			"err":    err.Error(),
		}).Panic("数据库连接异常")
	}

	sqlDB, err := db.DB()
	if err != nil {
		cLog.WithContext(nil, map[string]interface{}{
			"source": "cDB.MysqlConnector.Connect",
			"err":    err.Error(),
		}).Panic("数据库连接异常")
	}

	sqlDB.SetMaxIdleConns(driverConf.Pool.MaxIdleConnections)
	sqlDB.SetMaxOpenConns(driverConf.Pool.MaxConnections)
	sqlDB.SetConnMaxLifetime(time.Duration(driverConf.Pool.ConnectTimeout) * time.Second)
	sqlDB.SetConnMaxIdleTime(time.Duration(driverConf.Pool.MaxIdleTime) * time.Second)

	return db
}