dongfeng-pay/jhagent/utils/mysql.go

45 lines
1.3 KiB
Go
Raw Normal View History

2019-12-19 14:47:58 +08:00
/***************************************************
** @Desc : This file for 配置数据库连接
** @Time : 2018-12-22 13:55:26
** @Author : Joker
** @File : init_database.go
** @Last Modified by : Joker
** @Last Modified time:2018-12-22 13:55:26
** @Software: GoLand
****************************************************/
package utils
import (
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
_ "github.com/go-sql-driver/mysql"
)
//初始化数据连接
func InitDatabase() bool {
//读取配置文件,设置数据库参数
dbType := beego.AppConfig.String("db_type")
dbAlias := beego.AppConfig.String(dbType + "::db_alias")
dbName := beego.AppConfig.String(dbType + "::db_name")
dbUser := beego.AppConfig.String(dbType + "::db_user")
dbPwd := beego.AppConfig.String(dbType + "::db_pwd")
dbHost := beego.AppConfig.String(dbType + "::db_host")
dbPort := beego.AppConfig.String(dbType + "::db_port")
var err error
switch dbType {
case "sqlite3":
err = orm.RegisterDataBase(dbAlias, dbType, dbName)
case "mysql":
dbCharset := beego.AppConfig.String(dbType + "::db_charset")
err = orm.RegisterDriver(dbType, orm.DRMySQL)
err = orm.RegisterDataBase(dbAlias, dbType, dbUser+":"+dbPwd+"@tcp("+dbHost+":"+
dbPort+")/"+dbName+"?charset="+dbCharset, 30)
}
if err != nil {
return false
}
return true
}