- 解决wxtoken的刷新问题

business/jxutils/eventhub/eventhub.go

@@ -33,7 +33,6 @@ const (
 )
 
 type IEventProducer interface {
-	GetEvent(eventTypeList []string, criteria interface{}) (event *EventInfo, err error)
 	IsCriteriaMatch(eventInfo *EventInfo, criteria interface{}) bool
 }
 
@@ -212,36 +211,29 @@ func (e *EventHub) PostNewEvent(eventCategory string, event *EventInfo) {
 }
 
 func (e *EventHub) GetEvent(eventCategory string, eventTypeList []string, criteria interface{}, waitTime time.Duration) (event *EventInfo, err error) {
-	eventProducer := e.getEventProducer(eventCategory)
-	if eventProducer == nil {
-		return nil, fmt.Errorf("eventCategory:%s没有注册", eventCategory)
-	}
-	event, err = eventProducer.GetEvent(eventTypeList, criteria)
-	if err == nil && event == nil {
-		notifyChan := e.registerConsumer(eventCategory, eventTypeList, criteria)
-		pollingDuration := defPollingDuration
-		if waitTime != 0 {
-			pollingDuration = waitTime
-			if globals.IsProductEnv() {
-				if pollingDuration > maxPollingDuration {
-					pollingDuration = maxPollingDuration
-				} else if pollingDuration < minPollingDuration {
-					pollingDuration = minPollingDuration
-				}
+	notifyChan := e.registerConsumer(eventCategory, eventTypeList, criteria)
+	pollingDuration := defPollingDuration
+	if waitTime != 0 {
+		pollingDuration = waitTime
+		if globals.IsProductEnv() {
+			if pollingDuration > maxPollingDuration {
+				pollingDuration = maxPollingDuration
+			} else if pollingDuration < minPollingDuration {
+				pollingDuration = minPollingDuration
 			}
 		}
-		timer := time.NewTimer(pollingDuration)
-		select {
-		case tmpEvent, ok := <-notifyChan:
-			timer.Stop()
-			if ok {
-				event = tmpEvent
-			}
-		case <-timer.C:
-			e.unregisterConsumer(notifyChan)
-		}
-		close(notifyChan)
 	}
+	timer := time.NewTimer(pollingDuration)
+	select {
+	case tmpEvent, ok := <-notifyChan:
+		timer.Stop()
+		if ok {
+			event = tmpEvent
+		}
+	case <-timer.C:
+		e.unregisterConsumer(notifyChan)
+	}
+	close(notifyChan)
 	return event, err
 }
 

business/jxutils/eventhub/syseventhub/syseventhub.go

@@ -20,6 +20,11 @@ type Hub struct {
 type Criteria struct {
 }
 
+type WXTokenInfo struct {
+	IsNew bool   `json:"isNew"`
+	Token string `json:"token"`
+}
+
 var (
 	SysEventHub *Hub
 )
@@ -36,31 +41,6 @@ func New() (hub *Hub) {
 	return hub
 }
 
-func (h *Hub) GetEvent(eventTypeList []string, criteria interface{}) (event *eventhub.EventInfo, err error) {
-	for _, eventType := range eventTypeList {
-		switch eventType {
-		case EventTypeWXToken:
-			return h.getWXToken(criteria)
-		}
-	}
-	return event, err
-}
-
-func (h *Hub) getWXToken(criteria interface{}) (event *eventhub.EventInfo, err error) {
-	criteriaToken := ""
-	if criteria != nil {
-		criteriaToken = criteria.(string)
-	}
-	token := api.WeixinAPI.CBGetToken()
-	if token != criteriaToken {
-		return &eventhub.EventInfo{
-			Type: EventTypeWXToken,
-			Data: token,
-		}, nil
-	}
-	return nil, nil
-}
-
 func (h *Hub) IsCriteriaMatch(eventInfo *eventhub.EventInfo, criteria interface{}) bool {
 	return true
 }
@@ -72,10 +52,21 @@ func (h *Hub) OnNewWXToken(token string) {
 	})
 }
 
-func (h *Hub) GetWXToken(oldToken string, waitTime time.Duration) (token string) {
-	eventInfo, err := h.eventHub.GetEvent(EventCategory, []string{EventTypeWXToken}, oldToken, waitTime)
-	if err == nil && eventInfo != nil {
-		token = eventInfo.Data.(string)
+func (h *Hub) GetWXToken(oldToken string, waitTime time.Duration) (tokenInfo *WXTokenInfo) {
+	token := api.WeixinAPI.CBGetToken()
+	if token != oldToken {
+		tokenInfo = &WXTokenInfo{
+			IsNew: false,
+			Token: token,
+		}
+	} else {
+		eventInfo, err := h.eventHub.GetEvent(EventCategory, []string{EventTypeWXToken}, nil, waitTime)
+		if err == nil && eventInfo != nil {
+			tokenInfo = &WXTokenInfo{
+				IsNew: true,
+				Token: eventInfo.Data.(string),
+			}
+		}
 	}
-	return token
+	return tokenInfo
 }