// Tool Url: https://github.com/go-gorm/gorm
// Tool Guide: https://gorm.io/docs/

package database

import (
	"context"
	"dataapps/commons/config"
	"fmt"
	"time"

	"github.com/tidbcloud/tiinf/monitor/tilog"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

// Mysql handler infomation
type MysqlInfo struct {
	DB *gorm.DB
}

var DBConn = &MysqlInfo{}

func Connect(ctx context.Context, config *config.CentralConfig) {
	// Params
	url := fmt.Sprintf("%s:%s@tcp(%s:%v)/%s?charset=%s&parseTime=True&loc=%s",
		config.MysqlConfig.User,
		config.MysqlConfig.Password,
		config.MysqlConfig.Host,
		config.MysqlConfig.Port,
		config.MysqlConfig.Database,
		config.MysqlConfig.CharSet,
		config.MysqlConfig.TimeZone,
	)
	tilog.Info(ctx, "db sqleditor connection begin!")
	// Connect
	conn, err := gorm.Open(mysql.Open(url), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Error),
	})
	if err != nil {
		panic(err.Error())
	}
	sqlDB, err := conn.DB()
	if err != nil {
		panic(err.Error())
	}
	sqlDB.SetMaxIdleConns(20)
	sqlDB.SetMaxOpenConns(100)
	sqlDB.SetConnMaxLifetime(time.Second * 600)

	DBConn.DB = conn
}