zhuyujun/go-desk-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改消息推送处理

Browse Source
This commit is contained in:
zyj
2025-09-23 16:57:31 +08:00
parent 31d58d2751
commit 982be3fe9c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/websocks.go
View File

@@ -20,7 +20,7 @@ import (
type Websocks struct { type Websocks struct {
} }
var DeviceList map[int64]models.Device var DeviceList = make(map[int64]models.Device)
type Message struct { type Message struct {
Type string `form:"type" json:"type" uri:"type" xml:"type"` Type string `form:"type" json:"type" uri:"type" xml:"type"`
@@ -76,6 +76,7 @@ func (w *Websocks) Init(ctx *gin.Context) {
deviceService := services.InitDeviceService() deviceService := services.InitDeviceService()
deviceLoginInfo := deviceService.Login(tokenStr) deviceLoginInfo := deviceService.Login(tokenStr)
DeviceList[deviceLoginInfo.ID] = deviceLoginInfo DeviceList[deviceLoginInfo.ID] = deviceLoginInfo
// 将保存连接状态 // 将保存连接状态
ClientsMu.Lock() ClientsMu.Lock()
Clients[TokenStatus.UserId] = map[int64]*websocket.Conn{deviceLoginInfo.ID: conn} Clients[TokenStatus.UserId] = map[int64]*websocket.Conn{deviceLoginInfo.ID: conn}
@@ -106,6 +107,9 @@ func (*Websocks) MessageHandle(conn *websocket.Conn, deviceLoginInfo models.Devi
if data.Recipient != 0 { if data.Recipient != 0 {
RecipientDevice = DeviceList[data.Recipient] RecipientDevice = DeviceList[data.Recipient]
} }
log.Println("推送用户", RecipientDevice.UserId)
log.Println("推送设备", RecipientDevice.ID)
switch data.Type { switch data.Type {
case "message": case "message":
if data.Recipient != 0 { if data.Recipient != 0 {