package model

import (
	"git.hpds.cc/Component/logging"
	_ "github.com/go-sql-driver/mysql"
	"go.uber.org/zap"
	"os"

	"xorm.io/xorm"
	"xorm.io/xorm/dialects"
)

var (
	DB     *xorm.Engine
	Logger *logging.Logger
)

func New(driveName, dsn string) {
	DB, _ = NewDbConnection(driveName, dsn)
	DB.ShowSQL(true)
	DB.Dialect().SetQuotePolicy(dialects.QuotePolicyReserved)
	err := DB.Sync2(
		&Node{},
		&NodeState{},
	)
	if err != nil {
		zap.L().Error("同步数据库表结构", zap.Error(err))
		os.Exit(1)
	}
}

func NewDbConnection(driveName, dsn string) (db *xorm.Engine, err error) {
	db, err = xorm.NewEngine(driveName, dsn)
	if err != nil {
		zap.L().Error("创建数据库连接", zap.Error(err))
		os.Exit(-1)
	}
	db.SetMaxOpenConns(300)
	return
}