46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
/*
|
|
* @Author: haodaquan
|
|
* @Date: 2017-06-20 09:44:44
|
|
* @Last Modified by: haodaquan
|
|
* @Last Modified time: 2017-06-21 12:21:37
|
|
*/
|
|
|
|
package models
|
|
|
|
import (
|
|
"net/url"
|
|
|
|
"github.com/astaxie/beego"
|
|
"github.com/astaxie/beego/orm"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/gpmgo/gopm/modules/log"
|
|
)
|
|
|
|
func Init() {
|
|
dbhost := beego.AppConfig.String("db.host")
|
|
dbport := beego.AppConfig.String("db.port")
|
|
dbuser := beego.AppConfig.String("db.user")
|
|
dbpassword := beego.AppConfig.String("db.password")
|
|
dbname := beego.AppConfig.String("db.name")
|
|
timezone := beego.AppConfig.String("db.timezone")
|
|
if dbport == "" {
|
|
dbport = "3306"
|
|
}
|
|
dsn := dbuser + ":" + dbpassword + "@tcp(" + dbhost + ":" + dbport + ")/" + dbname + "?charset=utf8"
|
|
log.Fatal(dsn)
|
|
|
|
if timezone != "" {
|
|
dsn = dsn + "&loc=" + url.QueryEscape(timezone)
|
|
}
|
|
orm.RegisterDataBase("default", "mysql", dsn)
|
|
orm.RegisterModel(new(User), new(Task), new(TaskGroup), new(TaskLog), new(TaskServer))
|
|
|
|
if beego.AppConfig.String("runmode") == "dev" {
|
|
orm.Debug = true
|
|
}
|
|
}
|
|
|
|
func TableName(name string) string {
|
|
return beego.AppConfig.String("db.prefix") + name
|
|
}
|