package DbUtil

import (
	"dsSdsf/Utils/ConfigUtil"
	"fmt"
	_ "github.com/go-sql-driver/mysql"
	"github.com/xormplus/xorm"
	"github.com/xormplus/xorm/log"
	"os"
	"time"
)

var Engine *xorm.Engine

func init() {
	host := ConfigUtil.MysqlIp
	port := ConfigUtil.MysqlPort
	user := ConfigUtil.MysqlUser
	dbname := ConfigUtil.MysqlDbName
	password := ConfigUtil.MysqlPwd
	//mysql
	master := fmt.Sprintf("%s:%s@%s(%s:%s)/%s?charset=utf8", user, password, "tcp", host, port, dbname)
	var err error
	Engine, err = xorm.NewEngine("mysql", master)
	if err != nil {
		fmt.Println(err)
	}
	//注册SqlMap配置，可选功能，如应用中无需使用SqlMap，可无需初始化
	err = Engine.RegisterSqlMap(xorm.Xml("./Sql", ".xml"))
	if err != nil {
		fmt.Println(err)
	}

	//设置数据库连接池
	Engine.SetMaxOpenConns(100) //设置打开数据库的最大连接数，包含正在使用的连接和连接池的连接。
	Engine.SetMaxIdleConns(10)  //设置连接池中的保持连接的最大连接数。
	Engine.SetConnMaxLifetime(time.Second * 30)

	//调用第一次
	Engine.Ping()

	//这段代码是黄海后加上的 2020-04-16
	//通过定时ping保持鲜活
	// 创建完成engine之后，并没有立即连接数据库，此时可以通过engine.Ping()来进行数据库的连接测试是否可以连接到数据库。
	// 另外对于某些数据库有连接超时设置的，可以通过起一个定期Ping的Go程来保持连接鲜活。
	go func() {
		for {
			//https://studygolang.com/articles/12617
			time.Sleep(time.Duration(10) * time.Second)
			Engine.Ping()
		}
	}()

	//设置数据时区
	var location *time.Location
	location, err = time.LoadLocation("Asia/Shanghai")
	Engine.TZLocation = location

	//显示+记录SQL日志
	f, _ := os.Create(ConfigUtil.DistributeRemotePath + "sql.log")
	Engine.SetLogger(log.NewSimpleLogger(f))

	Engine.ShowSQL(true)                    // 则会在控制台打印出生成的SQL语句
	Engine.Logger().SetLevel(log.LOG_DEBUG) //则会在控制台打印info及以上的信息
}