优化消息同步

Author: NICEXAI

syncmsg.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"github.com/NICEXAI/WeChatCustomerServiceSDK/syncmsg"
 	"github.com/NICEXAI/WeChatCustomerServiceSDK/util"
 )
 
@@ -19,13 +20,22 @@ type SyncMsgOptions struct {
 	Limit uint `json:"limit"`								// 期望请求的数据量，默认值和最大值都为1000, 注意：可能会出现返回条数少于limit的情况，需结合返回的has_more字段判断是否继续请求。
 }
 
+// SyncMsgSchema 获取消息查询响应内容
+type syncMsgSchema struct {
+	ErrCode int32 `json:"errcode"`							// 返回码
+	ErrMsg string `json:"errmsg"`							// 错误码描述
+	NextCursor string `json:"next_cursor"`					// 下次调用带上该值则从该key值往后拉，用于增量拉取
+	HasMore uint32 `json:"has_more"`						// 是否还有更多数据。0-否；1-是。不能通过判断msg_list是否空来停止拉取，可能会出现has_more为1，而msg_list为空的情况
+	MsgList []map[string]interface{} `json:"msg_list"`		// 消息列表
+}
+
 // SyncMsgSchema 获取消息查询响应内容
 type SyncMsgSchema struct {
 	ErrCode int32 `json:"errcode"`							// 返回码
 	ErrMsg string `json:"errmsg"`							// 错误码描述
 	NextCursor string `json:"next_cursor"`					// 下次调用带上该值则从该key值往后拉，用于增量拉取
 	HasMore uint32 `json:"has_more"`						// 是否还有更多数据。0-否；1-是。不能通过判断msg_list是否空来停止拉取，可能会出现has_more为1，而msg_list为空的情况
-	MsgList [][]byte `json:"msg_list"`						// 消息列表
+	MsgList []syncmsg.Message `json:"msg_list"`				// 消息列表
 }
 
 // SyncMsg 获取消息
@@ -34,9 +44,56 @@ func (r *Client) SyncMsg(options SyncMsgOptions) (info SyncMsgSchema, err error)
 	if err != nil {
 		return info, err
 	}
-	_ = json.Unmarshal(data, &info)
-	if info.ErrCode != 0 {
-		return info, errors.New(info.ErrMsg)
+	originInfo := syncMsgSchema{}
+	if err = json.Unmarshal(data, &originInfo); err != nil {
+		return info, err
+	}
+	if originInfo.ErrCode != 0 {
+		return info, errors.New(originInfo.ErrMsg)
+	}
+	msgList := make([]syncmsg.Message, 0)
+	if len(originInfo.MsgList) > 0 {
+		for _, msg := range originInfo.MsgList {
+			newMsg := syncmsg.Message{}
+			if val, ok := msg["msgid"].(string); ok {
+				newMsg.MsgID = val
+			}
+			if val, ok := msg["open_kfid"].(string); ok {
+				newMsg.OpenKFID = val
+			}
+			if val, ok := msg["external_userid"].(string); ok {
+				newMsg.ExternalUserID = val
+			}
+			if val, ok := msg["send_time"].(float64); ok {
+				newMsg.SendTime = uint64(val)
+			}
+			if val, ok := msg["origin"].(float64); ok {
+				newMsg.Origin = uint32(val)
+			}
+
+			if val, ok := msg["msgtype"].(string); ok {
+				newMsg.MsgType = val
+			}
+			if newMsg.MsgType == "event" {
+				if event, ok :=  msg["event"].(map[string]interface{}); ok {
+					if eType, ok := event["event_type"].(string); ok {
+						newMsg.EventType = eType
+					}
+				}
+			}
+			originData, err := json.Marshal(msg)
+			if err != nil {
+				return info, err
+			}
+			newMsg.OriginData = originData
+			msgList = append(msgList, newMsg)
+		}
 	}
-	return info, nil
+	return SyncMsgSchema{
+		ErrCode:    originInfo.ErrCode,
+		ErrMsg:     originInfo.ErrMsg,
+		NextCursor: originInfo.NextCursor,
+		HasMore:    originInfo.HasMore,
+		MsgList:    msgList,
+	}, nil
 }

syncmsg/callback.go

@@ -1,9 +1,9 @@
 package syncmsg
 
 type Event struct {
-	ToUserName string 					// 微信客服组件ID
-	CreateTime int						// 消息创建时间，unix时间戳
-	MsgType	string						// 消息的类型，此时固定为 event
-	Event string						// 事件的类型，此时固定为 kf_msg_or_event
-	Token string						// 调用拉取消息接口时，需要传此token，用于校验请求的合法性
+	ToUserName string `json:"to_user_name"`					// 微信客服组件ID
+	CreateTime int `json:"create_time"`						// 消息创建时间，unix时间戳
+	MsgType	string `json:"msgtype"`						// 消息的类型，此时固定为 event
+	Event string `json:"event"`								// 事件的类型，此时固定为 kf_msg_or_event
+	Token string `json:"token"`								// 调用拉取消息接口时，需要传此token，用于校验请求的合法性
 }

syncmsg/syncmsg.go

@@ -11,58 +11,58 @@ type Message struct {
 	Origin uint32 `json:"origin"`						// 消息来源。3-客户回复的消息 4-系统推送的消 息
 	MsgType string `json:"msgtype"`						// 消息类型
 	EventType string `json:"event_type"`				// 事件类型
-	originData []byte 									// 原始数据内容
+	OriginData []byte `json:"origin_data"`				// 原始数据内容
 }
 
 // GetOriginMessage 获取原始消息
 func (r Message) GetOriginMessage() (info []byte) {
-	return r.originData
+	return r.OriginData
 }
 
 // GetTextMessage 获取文本消息
 func (r Message) GetTextMessage() (info Text, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetImageMessage 获取图片消息
 func (r Message) GetImageMessage() (info Image, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetVoiceMessage 获取语音消息
 func (r Message) GetVoiceMessage() (info Voice, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetVideoMessage 获取视频消息
 func (r Message) GetVideoMessage() (info Video, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetFileMessage 获取文件消息
 func (r Message) GetFileMessage() (info File, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetLocationMessage 获取文件消息
 func (r Message) GetLocationMessage() (info Location, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetEnterSessionEvent 用户进入会话事件
 func (r Message) GetEnterSessionEvent() (info EnterSessionEvent, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }
 
 // GetMsgSendFailEvent 消息发送失败事件
 func (r Message) GetMsgSendFailEvent() (info MsgSendFailEvent, err error) {
-	err = json.Unmarshal(r.originData, &info)
+	err = json.Unmarshal(r.OriginData, &info)
 	return info, err
 }