package data

import (
	"database/sql"
	_ "github.com/go-sql-driver/mysql"
	"github.com/go-xorm/xorm"
	"wss-pool/logging/applogger"
	"wss-pool/pkg/model"
)

var Engine *xorm.EngineGroup
var DB *sql.DB

// InitMysql Init Mysql DB
func InitMysql(config model.Bourse) {
	var err error

	driver := config.Driver
	datasource := config.Datasource

	conns := []string{
		datasource,
	}
	Engine, err = xorm.NewEngineGroup(driver, conns, xorm.RandomPolicy())
	if err != nil {
		applogger.Error("orm failed to initialized: %v", err)
		panic(err)
	}

	Engine.ShowExecTime(true)
	Engine.ShowSQL(true)
	Engine.SetMaxOpenConns(500)
	err = Engine.Ping()
	if err != nil {
		panic(err)
	}
}

func InitMsqlDB(config model.Bourse) {
	// root:IcWw%%r3pt@tcp(34.126.77.164:3307)/bourse?charset=utf8
	db, err := sql.Open("mysql", config.Datasource)
	if err != nil {
		applogger.Error("Datasource open err: %v", err)
		return
	}
	DB = db
	return
}