first commit

srv/wxcore/wxmsghandler.go

@@ -0,0 +1,101 @@
+package wxcore
+
+import (
+	"fmt"
+	"xiawan/wx/srv/srvconfig"
+	"xiawan/wx/srv/wxface"
+)
+
+// WXMsgHandler 微信响应管理器
+type WXMsgHandler struct {
+	wxRouterMap      map[uint32]wxface.IWXRouter //存放每个MsgId 所对应的处理方法的map属性
+	wxWorkerPoolSize uint32                      //业务工作Worker池的数量
+	wxTaskQueue      []chan wxface.IWXResponse   //Worker负责取任务的消息队列
+}
+
+// NewWXMsgHandler 新建微信消息处理器
+func NewWXMsgHandler() *WXMsgHandler {
+	return &WXMsgHandler{
+		wxRouterMap: make(map[uint32]wxface.IWXRouter),
+		// 一个worker对应一个queue
+		wxWorkerPoolSize: srvconfig.GlobalSetting.WorkerPoolSize * 2,
+		wxTaskQueue:      make([]chan wxface.IWXResponse, srvconfig.GlobalSetting.WorkerPoolSize*2),
+	}
+}
+
+// AddRouter 增加微信消息路由
+func (wxmh *WXMsgHandler) AddRouter(respID uint32, wxRouter wxface.IWXRouter) {
+	//1 判断当前msg绑定的API处理方法是否已经存在
+	if _, ok := wxmh.wxRouterMap[respID]; ok {
+		return
+	}
+	//2 添加msg与api的绑定关系
+	wxmh.wxRouterMap[respID] = wxRouter
+}
+
+// GetRouterByRespID 根据响应ID获取对应的路由
+func (wxmh *WXMsgHandler) GetRouterByRespID(urlID uint32) wxface.IWXRouter {
+	handler, ok := wxmh.wxRouterMap[urlID]
+	if !ok {
+		return nil
+	}
+	return handler
+}
+
+// doMsgHandler 马上以非阻塞方式处理消息
+func (wxmh *WXMsgHandler) doMsgHandler(response wxface.IWXResponse) {
+	//xiaoyue处理消息
+	// fmt.Println("xiaoyue------doMsgHandler")
+	// defer TryE(response.GetWXConncet().GetWXAccount().GetUserInfo().GetUserName())
+	handler, ok := wxmh.wxRouterMap[response.GetPackHeader().URLID]
+	// fmt.Println("[URLID]", response.GetPackHeader().URLID)
+	if !ok {
+		return
+	}
+
+	//执行对应处理方法
+	handler.PreHandle(response)
+	handler.Handle(response)
+	handler.PostHandle(response)
+}
+
+// startOneWorker 启动一个Worker工作流程
+func (wxmh *WXMsgHandler) startOneWorker(workerID int, taskQueue chan wxface.IWXResponse) {
+	// fmt.Println("xiaoyue------startOneWorker")
+	defer func() {
+		if r := recover(); r != nil {
+			fmt.Printf("startOneWorker recovered from panic: %v\n", r)
+			// 可以在这里记录日志或执行其他紧急恢复处理措施
+		}
+	}()
+	//不断的等待队列中的消息
+	for {
+		select {
+		//有消息则取出队列的Request，并执行绑定的业务方法
+		case request := <-taskQueue:
+			wxmh.doMsgHandler(request)
+		}
+	}
+}
+
+// StartWorkerPool 启动worker工作池
+func (wxmh *WXMsgHandler) StartWorkerPool() {
+	defer TryE("")
+	// fmt.Println("xiaoyue------StartWorkerPool")
+	//遍历需要启动worker的数量，依此启动
+	for i := 0; i < int(wxmh.wxWorkerPoolSize); i++ {
+		//一个worker被启动
+		//给当前worker对应的任务队列开辟空间
+		wxmh.wxTaskQueue[i] = make(chan wxface.IWXResponse, srvconfig.GlobalSetting.MaxWorkerTaskLen*2)
+		//启动当前Worker，阻塞的等待对应的任务队列是否有消息传递进来
+		go wxmh.startOneWorker(i, wxmh.wxTaskQueue[i])
+	}
+}
+
+// SendWXRespToTaskQueue 将消息交给TaskQueue,由worker进行处理
+func (wxmh *WXMsgHandler) SendWXRespToTaskQueue(response wxface.IWXResponse) {
+	//得到需要处理此条连接的workerID
+	workerID := response.GetWXConncet().GetWXConnID() % wxmh.wxWorkerPoolSize
+	//将请求消息发送给任务队列
+	wxmh.wxTaskQueue[workerID] <- response
+}