You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
925 B
49 lines
925 B
2 months ago
|
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
|
||
|
|
}
|