This commit is contained in:
suyl
2021-06-28 18:46:37 +08:00
parent 0540488e2f
commit 274b9c7cba
2 changed files with 6 additions and 18 deletions

View File

@@ -141,5 +141,9 @@ func IsMainProductEnv() bool {
} }
func IsProductEnv() bool { func IsProductEnv() bool {
return beego.BConfig.RunMode == "rsm"
}
func IsPrintEnv() bool {
return beego.BConfig.RunMode == "print" return beego.BConfig.RunMode == "print"
} }

20
main.go
View File

@@ -101,24 +101,8 @@ func checkCmdFlags() bool {
func main() { func main() {
if !checkCmdFlags() { if !checkCmdFlags() {
Init() Init()
// if err := tasks.RefreshWeixinToken(); err != nil {
// globals.SugarLogger.Errorf("RefreshWeixinToken failed with error:%s", err)
// return
// }
// if err := tasks.RefreshWeixin2Token(); err != nil {
// globals.SugarLogger.Errorf("RefreshWeixin2Token failed with error:%s", err)
// return
// }
// if err := tasks.RefreshWeixin3Token(); err != nil {
// globals.SugarLogger.Errorf("RefreshWeixin3Token failed with error:%s", err)
// return
// }
// if err := tasks.RefreshPushToken(); err != nil {
// globals.SugarLogger.Errorf("RefreshPushToken failed with error:%s", err)
// return
// }
// 延时的原因是等回调准备好 // 延时的原因是等回调准备好
if globals.IsProductEnv() || beego.BConfig.RunMode == "beta" { if globals.IsMainProductEnv() || beego.BConfig.RunMode == "beta" {
utils.AfterFuncWithRecover(2*time.Second, func() { utils.AfterFuncWithRecover(2*time.Second, func() {
if err := tasks.RefreshDingDingToken(); err != nil { if err := tasks.RefreshDingDingToken(); err != nil {
globals.SugarLogger.Errorf("RefreshDingDingToken failed with error:%s", err) globals.SugarLogger.Errorf("RefreshDingDingToken failed with error:%s", err)
@@ -144,7 +128,7 @@ func main() {
http.ListenAndServe("0.0.0.0:6060", nil) http.ListenAndServe("0.0.0.0:6060", nil)
} }
}() }()
if globals.IsProductEnv() { if globals.IsPrintEnv() {
event.Init() event.Init()
} }
beego.Run() beego.Run()