mirror of
https://github.com/kongyuebin1/dongfeng-pay.git
synced 2024-11-26 12:19:25 +08:00
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
/***************************************************
|
|
** @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
|
|
}
|