111 lines
2.4 KiB
Go
111 lines
2.4 KiB
Go
|
package controller
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/1340691923/xwl_bi/engine/logs"
|
||
|
|
"github.com/1340691923/xwl_bi/platform-basic-libs/jwt"
|
||
|
|
"github.com/gofiber/websocket/v2"
|
||
|
|
"go.uber.org/zap"
|
||
|
|
"io"
|
||
|
|
"net"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var ConnUUidMap sync.Map
|
||
|
|
|
||
|
|
type ManagerConnMap struct {
|
||
|
|
Conns *sync.Map
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewManagerConnMap() *ManagerConnMap {
|
||
|
|
return &ManagerConnMap{
|
||
|
|
Conns: new(sync.Map),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (this *ManagerConnMap) AddManagerConn(manager string, conn *websocket.Conn) {
|
||
|
|
this.Conns.Store(manager, conn)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (this *ManagerConnMap) DeleteConn(manager string) {
|
||
|
|
this.Conns.Delete(manager)
|
||
|
|
}
|
||
|
|
|
||
|
|
func removeConns(deleteConn *websocket.Conn) {
|
||
|
|
|
||
|
|
ConnUUidMap.Range(func(distinctId, managerMap interface{}) bool {
|
||
|
|
managerMap.(*ManagerConnMap).Conns.Range(func(key, conn interface{}) bool {
|
||
|
|
|
||
|
|
if conn == deleteConn {
|
||
|
|
managerMap.(*ManagerConnMap).DeleteConn(key.(string))
|
||
|
|
ConnUUidMap.Store(distinctId, managerMap)
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
return true
|
||
|
|
})
|
||
|
|
deleteConn.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
//长链接 用于测试设备的数据接收
|
||
|
|
func DebugDataWs(c *websocket.Conn) {
|
||
|
|
|
||
|
|
type ReqData struct {
|
||
|
|
UUid string `json:"uuid"`
|
||
|
|
Token string `json:"token"`
|
||
|
|
}
|
||
|
|
|
||
|
|
var reqData ReqData
|
||
|
|
|
||
|
|
for {
|
||
|
|
reqData.UUid = ""
|
||
|
|
_, msg, err := c.ReadMessage()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
logs.Logger.Error("ws ReadMessage", zap.Error(err))
|
||
|
|
removeConns(c)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
err = json.Unmarshal(msg, &reqData)
|
||
|
|
if err != nil {
|
||
|
|
logs.Logger.Error("json Unmarshal", zap.Error(err))
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
if reqData.UUid == "" {
|
||
|
|
logs.Logger.Sugar().Errorf("reqData.UUid = %v", reqData.UUid)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
cliams, err := jwt.ParseToken(reqData.Token)
|
||
|
|
if err != nil {
|
||
|
|
logs.Logger.Error("jwt.ParseToken", zap.Error(err))
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
if cliams.UserID == 0 {
|
||
|
|
logs.Logger.Error("cliams.UserID = 0")
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
if data, found := ConnUUidMap.Load(reqData.UUid); !found {
|
||
|
|
conn := NewManagerConnMap()
|
||
|
|
conn.AddManagerConn(reqData.UUid, c)
|
||
|
|
ConnUUidMap.Store(reqData.UUid, conn)
|
||
|
|
} else {
|
||
|
|
data.(*ManagerConnMap).AddManagerConn(reqData.UUid, c)
|
||
|
|
ConnUUidMap.Store(reqData.UUid, data)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = c.WriteJSON(map[string]interface{}{"code": 0})
|
||
|
|
if err != nil {
|
||
|
|
if err == io.EOF {
|
||
|
|
logs.Logger.Error("客户端已经断开WsSocket!", zap.Error(err))
|
||
|
|
} else if err.(*net.OpError).Err.Error() == "use of closed network connection" {
|
||
|
|
logs.Logger.Error("服务端已经断开WsSocket!", zap.Error(err))
|
||
|
|
}
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|