push api

business/jxutils/eventhub/syseventhub/syseventhub.go

@@ -13,10 +13,10 @@ import (
 const (
 	EventCategory = "sys"
 
-	EventTypeWXToken  = "wxToken"
-	EventTypeWX2Token = "wx2Token"
-	EventTypeWX3Token = "wx3Token"
-
+	EventTypeWXToken     = "wxToken"
+	EventTypeWX2Token    = "wx2Token"
+	EventTypeWX3Token    = "wx3Token"
+	EventTypePushToken   = "pushToken"
 	EventTypeYLYToken    = "ylyToken"
 	EventTypeWeimobToken = "weimobToken"
 )
@@ -67,6 +67,13 @@ func (h *Hub) OnNewWX2Token(token string) {
 	})
 }
 
+func (h *Hub) OnNewPushToken(token string) {
+	h.eventHub.PostNewEvent(EventCategory, &eventhub.EventInfo{
+		Type: EventTypePushToken,
+		Data: token,
+	})
+}
+
 func (h *Hub) OnNewWX3Token(token string) {
 	h.eventHub.PostNewEvent(EventCategory, &eventhub.EventInfo{
 		Type: EventTypeWX3Token,
@@ -99,6 +106,8 @@ func (h *Hub) GetToken(tokenType, oldToken string, waitTime time.Duration) (toke
 		if weimobToken := api.WeimobAPI.GetToken(); weimobToken != nil {
 			token = string(utils.MustMarshal(weimobToken))
 		}
+	case EventTypePushToken:
+		token = api.PushAPI.CBGetToken()
 	case EventTypeWX2Token:
 		token = api.WeixinMiniAPI2.CBGetToken()
 	}
@@ -136,3 +145,7 @@ func (h *Hub) GetYLYToken(oldToken string, waitTime time.Duration) (tokenInfo *T
 func (h *Hub) GetWeimobToken(oldToken string, waitTime time.Duration) (tokenInfo *TokenInfo) {
 	return h.GetToken(EventTypeWeimobToken, oldToken, waitTime)
 }
+
+func (h *Hub) GetPushToken(oldToken string, waitTime time.Duration) (tokenInfo *TokenInfo) {
+	return h.GetToken(EventTypePushToken, oldToken, waitTime)
+}

business/jxutils/tasks/configrefresh.go

@@ -28,6 +28,7 @@ const (
 	dingdingTokenExpires  = 7200 * time.Second
 	weimobTokenExpires    = 7200 * time.Second
 	yilianyunTokenExpires = 30 * 24 * 3600 * time.Second
+	pushTokenExpires      = 7200 * time.Second
 
 	maxRefreshGap = 5 * 60 * time.Second
 	errRefreshGap = 10 * time.Second
@@ -165,6 +166,33 @@ func RefreshWeixin2Token() (err error) {
 	return err
 }
 
+func RefreshPushToken() (err error) {
+	if api.PushAPI != nil {
+		err = RefreshConfig("push", pushTokenExpires, func() (token string, expireTimeStr string) {
+			globals.SugarLogger.Debugf("RefreshPushToken RunMode:%s", beego.BConfig.RunMode)
+			if globals.IsMainProductEnv() {
+				if tokenInfo, err := api.PushAPI.CBRetrieveToken(); err == nil {
+					globals.SugarLogger.Debugf("RefreshPushToken tokenInfo:%s", utils.Format4Output(tokenInfo, true))
+					token = tokenInfo.Token
+				} else {
+					globals.SugarLogger.Errorf("RefreshPushToken RefreshToken failed with error:%v", err)
+				}
+			} else {
+				if tokenInfo := getPushTokenFromRemote(api.PushAPI.CBGetToken()); tokenInfo != nil {
+					expireTimeStr = utils.Time2Str(time.Now().Add(-pushTokenExpires))
+					token = tokenInfo.Token
+				}
+			}
+			return token, expireTimeStr
+		}, func(value string) {
+			globals.SugarLogger.Debugf("RefreshPushToken setter value:%s", value)
+			syseventhub.SysEventHub.OnNewPushToken(value)
+			api.PushAPI.CBSetToken(value)
+		})
+	}
+	return err
+}
+
 func RefreshWeixin3Token() (err error) {
 	// if api.WeixinMiniAPI3 != nil {
 	// 	err = RefreshConfig("wechat3", weixinTokenExpires, func() (token string, expireTimeStr string) {
@@ -342,6 +370,15 @@ func getWX2TokenFromRemote(oldToken string) (tokenInfo *syseventhub.TokenInfo) {
 	return tokenInfo
 }
 
+func getPushTokenFromRemote(oldToken string) (tokenInfo *syseventhub.TokenInfo) {
+	if !globals.IsMainProductEnv() && globals.GetWeixinTokenKey != "" && globals.GetPushTokenURL != "" {
+		tokenInfo = PollingRemotEvent(globals.GetPushTokenURL, 0, map[string]interface{}{
+			"oldToken": oldToken,
+		})
+	}
+	return tokenInfo
+}
+
 func getWX3TokenFromRemote(oldToken string) (tokenInfo *syseventhub.TokenInfo) {
 	if !globals.IsMainProductEnv() && globals.GetWeixinTokenKey != "" && globals.GetWeixinTokenURL != "" {
 		tokenInfo = PollingRemotEvent(globals.GetWeixinTokenURL, 0, map[string]interface{}{