mirror of
https://github.com/kongyuebin1/dongfeng-pay.git
synced 2024-12-26 03:52:24 +08:00
79 lines
2.3 KiB
Go
79 lines
2.3 KiB
Go
package models
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/beego/beego/v2/client/orm"
|
||
"github.com/beego/beego/v2/core/logs"
|
||
"github.com/beego/beego/v2/server/web"
|
||
_ "github.com/go-sql-driver/mysql"
|
||
"legend/models/fast"
|
||
"os"
|
||
)
|
||
|
||
/**
|
||
** 链接数据库，注册已经存在的数据表，进行orm映射操作
|
||
*/
|
||
func init() {
|
||
//initFastPay()
|
||
initLegend()
|
||
}
|
||
|
||
/**
|
||
** 初始化快付支付系统的mysql数据库
|
||
*/
|
||
func initFastPay() {
|
||
dbType, _ := web.AppConfig.String("dbtype")
|
||
mysqlHost, _ := web.AppConfig.String("fast::host")
|
||
mysqlPort, _ := web.AppConfig.String("fast::port")
|
||
mysqlUserName, _ := web.AppConfig.String("fast::username")
|
||
mysqlPassword, _ := web.AppConfig.String("fast::password")
|
||
mysqlDbName, _ := web.AppConfig.String("fast::dbname")
|
||
|
||
logs.Info("host:%s, port:%s, usreName:%s, password:%s, dbname:%s, dbType:%s", mysqlHost, mysqlPort,
|
||
mysqlUserName, mysqlPassword, mysqlDbName, dbType)
|
||
|
||
pStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&loc=Local",
|
||
mysqlUserName, mysqlPassword, mysqlHost, mysqlPort, mysqlDbName)
|
||
|
||
if err := orm.RegisterDataBase("default", dbType, pStr); err != nil {
|
||
logs.Error("init fast fail：%s", err)
|
||
os.Exit(1)
|
||
}
|
||
orm.SetMaxIdleConns("default", 30)
|
||
orm.SetMaxIdleConns("default", 30)
|
||
|
||
orm.RegisterModel(new(fast.RpUserInfo))
|
||
orm.RegisterModel(new(fast.RpUserPayConfig))
|
||
orm.RegisterModel(new(fast.RpUserBankAccount))
|
||
orm.RegisterModel(new(fast.RpAccount))
|
||
|
||
logs.Info("init fast success ......")
|
||
}
|
||
|
||
/**
|
||
** 初始化传奇支付系统的mysql数据库
|
||
*/
|
||
func initLegend() {
|
||
dbType, _ := web.AppConfig.String("dbtype")
|
||
mysqlHost, _ := web.AppConfig.String("legend::host")
|
||
mysqlPort, _ := web.AppConfig.String("legend::port")
|
||
mysqlUserName, _ := web.AppConfig.String("legend::username")
|
||
mysqlPassword, _ := web.AppConfig.String("legend::password")
|
||
mysqlDbName, _ := web.AppConfig.String("legend::dbname")
|
||
|
||
logs.Info("host:%s, port:%s, usreName:%s, password:%s, dbname:%s", mysqlHost, mysqlPort,
|
||
mysqlUserName, mysqlPassword, mysqlDbName)
|
||
|
||
pStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&loc=Local",
|
||
mysqlUserName, mysqlPassword, mysqlHost, mysqlPort, mysqlDbName)
|
||
|
||
if err := orm.RegisterDataBase("default", dbType, pStr); err != nil {
|
||
logs.Error("init legend fail：%s", err)
|
||
os.Exit(1)
|
||
}
|
||
|
||
logs.Info("init legend success ......")
|
||
|
||
orm.RegisterModel()
|
||
}
|