From 274b9c7cbae6ee87f43f8f1a6352d5b5719c2033 Mon Sep 17 00:00:00 2001
From: suyl <770236076@qq.com>
Date: Mon, 28 Jun 2021 18:46:37 +0800
Subject: [PATCH] aa

---
 globals/globals.go |  4 ++++
 main.go            | 20 ++------------------
 2 files changed, 6 insertions(+), 18 deletions(-)

diff --git a/globals/globals.go b/globals/globals.go
index 26cd02767..33d46515c 100644
--- a/globals/globals.go
+++ b/globals/globals.go
@@ -141,5 +141,9 @@ func IsMainProductEnv() bool {
 }
 
 func IsProductEnv() bool {
+	return beego.BConfig.RunMode == "rsm"
+}
+
+func IsPrintEnv() bool {
 	return beego.BConfig.RunMode == "print"
 }
diff --git a/main.go b/main.go
index 6b722f448..29c8f5682 100644
--- a/main.go
+++ b/main.go
@@ -101,24 +101,8 @@ func checkCmdFlags() bool {
 func main() {
 	if !checkCmdFlags() {
 		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() {
 				if err := tasks.RefreshDingDingToken(); err != nil {
 					globals.SugarLogger.Errorf("RefreshDingDingToken failed with error:%s", err)
@@ -144,7 +128,7 @@ func main() {
 				http.ListenAndServe("0.0.0.0:6060", nil)
 			}
 		}()
-		if globals.IsProductEnv() {
+		if globals.IsPrintEnv() {
 			event.Init()
 		}
 		beego.Run()