From 51b9da93049b64b062e047bd56045993639c1216 Mon Sep 17 00:00:00 2001
From: c0re100 <corehusky@gmail.com>
Date: Wed, 3 Aug 2022 09:52:06 +0800
Subject: [PATCH] Improve forwardMessages catcher

---
 client/client.go | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/client/client.go b/client/client.go
index 809247a..16b589f 100644
--- a/client/client.go
+++ b/client/client.go
@@ -19,6 +19,7 @@ type Client struct {
 	listenerStore   *listenerStore
 	catchersStore   *sync.Map
 	successMsgStore *sync.Map
+	forwardMsgStore *sync.Map
 	updatesTimeout  time.Duration
 	catchTimeout    time.Duration
 }
@@ -74,6 +75,7 @@ func NewClient(authorizationStateHandler AuthorizationStateHandler, options ...O
 		listenerStore:   newListenerStore(),
 		catchersStore:   &sync.Map{},
 		successMsgStore: &sync.Map{},
+		forwardMsgStore: &sync.Map{},
 	}
 
 	client.extraGenerator = UuidV4Generator()
@@ -110,9 +112,13 @@ func (client *Client) processResponse(response *Response) {
 	}
 
 	if typ.GetType() == (&UpdateMessageSendSucceeded{}).GetType() {
-		value, ok := client.successMsgStore.Load(typ.(*UpdateMessageSendSucceeded).OldMessageId)
-		if ok {
-			value.(chan *Response) <- response
+		sendVal, sOk := client.successMsgStore.Load(typ.(*UpdateMessageSendSucceeded).OldMessageId)
+		if sOk {
+			sendVal.(chan *Response) <- response
+		}
+		forwardVal, fOk := client.forwardMsgStore.Load(typ.(*UpdateMessageSendSucceeded).OldMessageId)
+		if fOk {
+			forwardVal.(chan *Response) <- response
 		}
 	}
 
@@ -220,16 +226,16 @@ func (client *Client) Send(req Request) (*Response, error) {
 			}
 
 			for _, m := range ms.Messages {
-				successCatcher := make(chan *Response, 1)
-				client.successMsgStore.Store(m.Id, successCatcher)
+				forwardCatcher := make(chan *Response, 1)
+				client.forwardMsgStore.Store(m.Id, forwardCatcher)
 
 				defer (func() {
-					client.successMsgStore.Delete(m.Id)
-					close(successCatcher)
+					client.forwardMsgStore.Delete(m.Id)
+					close(forwardCatcher)
 				})()
 
 				select {
-				case modResponse := <-successCatcher:
+				case modResponse := <-forwardCatcher:
 					m2, err2 := UnmarshalUpdateMessageSendSucceeded(modResponse.Data)
 					if err2 != nil {
 						return response, nil