func init() {
dbname := "default" // 数据库别名
dbtype := beego.AppConfig.String("dbtype")
dbcfg, err := beego.AppConfig.GetSection(dbtype)
if err != nil {
panic(err)
}
// 统一采用 UTC 时间
orm.DefaultTimeLoc = time.UTC
orm.Debug = true
switch dbtype {
case "mysql":
var username string = dbcfg["username"]
if username, err = utils.AesDecrypt(username); err != nil {
panic(err)
}
var password string = dbcfg["password"]
if password, err = utils.AesDecrypt(password); err != nil {
panic(err)
}
url := dbcfg["url"]
maxidle, _ := strconv.Atoi(dbcfg["maxidle"])
maxconn, _ := strconv.Atoi(dbcfg["maxconn"])
orm.RegisterDriver("mysql", orm.DR_MySQL)
orm.RegisterDataBase(dbname, "mysql",
username+":"+password+"@"+url,
maxidle, maxconn)
case "sqlite":
url := dbcfg["url"]
orm.RegisterDriver("sqlite3", orm.DR_Sqlite)
orm.RegisterDataBase(dbname, "sqlite3", url)
}
orm.RegisterModel(new(User), new(UserBind), new(UserStats))
orm.RegisterModel(new(Question), new(Reply), new(QuestionMark), new(AnswerMark))
orm.RegisterModel(new(Category))
orm.RegisterModel(new(Blog))
orm.RegisterModel(new(Comment))
force := false // drop table 后再建表
verbose, _ := beego.AppConfig.Bool("SqlLogOn") // 打印执行过程
// 遇到错误立即返回
err = orm.RunSyncdb(dbname, force, verbose)
if err != nil {
fmt.Println(err)
}
}
func init() {
// read config from file
EmailUser = beego.AppConfig.String("EmailUser")
EmailPasswd = beego.AppConfig.String("EmailPasswd")
if pwd, err := utils.AesDecrypt(EmailPasswd); err != nil {
panic(err)
} else {
EmailPasswd = pwd
}
EmailHost = beego.AppConfig.String("EmailHost")
Log = logs.NewLogger(1000)
Log.Async()
Log.SetLogger("console", "")
// 较简单的校验码
store := cache.NewMemoryCache()
Cpt = captcha.NewWithFilter("/captcha/", store)
}