Initial qiwei secondary development handoff

helper/types.go

@@ -0,0 +1,73 @@
+package main
+
+import (
+	"sync"
+
+	"qiweimanager/logger"
+)
+
+// 全局变量
+var (
+	globalLogger      *logger.Logger
+	globalLoaderFuncs *LoaderFunctions
+	loaderFuncsMutex  sync.Mutex
+
+	// ResponseMap 用于存储客户端ID和对应的响应通道
+	ResponseMap   = make(map[int32]chan ClientResponseData)
+	responseMapMu sync.Mutex
+
+	// ResponseMap 用于存储客户端ID和对应的响应通道
+	globalClientMap = make(map[uint32]string)
+)
+
+// LoaderFunctions 定义了从Loader DLL获取的所有函数指针
+type LoaderFunctions struct {
+	GetUserWxWorkVersion  uintptr
+	UseUtf8               uintptr
+	UseRecvJsUnicode      uintptr
+	InitWxWorkSocket      uintptr
+	SetDataLocationPath   uintptr
+	InjectWxWork          uintptr
+	InjectWxWorkMultiOpen uintptr
+	InjectWxWorkPid       uintptr
+	DestroyWxWork         uintptr
+	SendWxWorkData        uintptr
+}
+
+// ClientResponseData 定义客户端响应数据结构
+type ClientResponseData struct {
+	ClientId int32                  `json:"clientId"`
+	Data     map[string]interface{} `json:"data"`
+}
+
+// SetResponseChannel 为指定的客户端ID设置响应通道
+func SetResponseChannel(clientId int32, ch chan ClientResponseData) {
+	responseMapMu.Lock()
+	ResponseMap[clientId] = ch
+	responseMapMu.Unlock()
+}
+
+// GetResponseChannel 获取指定客户端ID的响应通道
+func TrySetResponseChannel(clientId int32, ch chan ClientResponseData) bool {
+	responseMapMu.Lock()
+	defer responseMapMu.Unlock()
+	if _, exists := ResponseMap[clientId]; exists {
+		return false
+	}
+	ResponseMap[clientId] = ch
+	return true
+}
+
+func GetResponseChannel(clientId int32) (chan ClientResponseData, bool) {
+	responseMapMu.Lock()
+	ch, exists := ResponseMap[clientId]
+	responseMapMu.Unlock()
+	return ch, exists
+}
+
+// RemoveResponseChannel 移除指定客户端ID的响应通道
+func RemoveResponseChannel(clientId int32) {
+	responseMapMu.Lock()
+	delete(ResponseMap, clientId)
+	responseMapMu.Unlock()
+}