From 31d58d2751418210af1228d4ed738a170fd18aeb Mon Sep 17 00:00:00 2001
From: zyj <18107291228@163.com>
Date: Mon, 22 Sep 2025 18:12:24 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=80=9A=E7=94=A8=E7=9A=84?=
 =?UTF-8?q?=E6=B6=88=E6=81=AF=E5=A4=84=E7=90=86=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/websocks.go            | 13 ++++++++++---
 services/device_service.go |  6 ++++++
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/api/websocks.go b/api/websocks.go
index f488477..9ec0cb6 100644
--- a/api/websocks.go
+++ b/api/websocks.go
@@ -20,6 +20,8 @@ import (
 type Websocks struct {
 }
 
+var DeviceList map[int64]models.Device
+
 type Message struct {
 	Type      string `form:"type" json:"type" uri:"type" xml:"type"`
 	Data      any    `form:"data" json:"data" uri:"data" xml:"data"`
@@ -73,6 +75,7 @@ func (w *Websocks) Init(ctx *gin.Context) {
 	defer conn.Close()
 	deviceService := services.InitDeviceService()
 	deviceLoginInfo := deviceService.Login(tokenStr)
+	DeviceList[deviceLoginInfo.ID] = deviceLoginInfo
 	// 将保存连接状态
 	ClientsMu.Lock()
 	Clients[TokenStatus.UserId] = map[int64]*websocket.Conn{deviceLoginInfo.ID: conn}
@@ -84,6 +87,7 @@ func (w *Websocks) Init(ctx *gin.Context) {
 func (*Websocks) MessageHandle(conn *websocket.Conn, deviceLoginInfo models.Device) {
 	for {
 		_, message, err := conn.ReadMessage()
+
 		if err != nil {
 			log.Printf("读取错误: %v (客户端: %s)", err, conn.RemoteAddr())
 			log.Println("客户端断开", deviceLoginInfo.Token)
@@ -98,19 +102,22 @@ func (*Websocks) MessageHandle(conn *websocket.Conn, deviceLoginInfo models.Devi
 		if err := json.Unmarshal(message, &data); err == nil && data.Type != "" {
 			// 说明是结构体(JSON)
 			log.Printf("收到结构体: %+v\n", data)
+			var RecipientDevice models.Device
+			if data.Recipient != 0 {
+				RecipientDevice = DeviceList[data.Recipient]
+			}
 			switch data.Type {
 			case "message":
 				if data.Recipient != 0 {
 					// 将数据推送给对应设备
-					userConn := Clients[deviceLoginInfo.UserId][deviceLoginInfo.ID]
+					userConn := Clients[RecipientDevice.UserId][RecipientDevice.ID]
 					userConn.WriteJSON(&Message{
 						Type:      "message",
 						Data:      data.Data,
-						Sender:    deviceLoginInfo.UserId,
+						Sender:    deviceLoginInfo.ID,
 						Recipient: data.Recipient,
 					})
 				}
-				break
 			}
 		} else {
 			log.Printf("收到普通消息: %s\n", string(message))
diff --git a/services/device_service.go b/services/device_service.go
index 92b02cb..a080e5d 100644
--- a/services/device_service.go
+++ b/services/device_service.go
@@ -58,6 +58,12 @@ func (d *DeviceService) GetInfoByToken(token string) (*models.Device, error) {
 	return &device, nil
 }
 
+func (d *DeviceService) GetInfo(id int64) *models.Device {
+	var device models.Device
+	d.db.Model(&models.Device{}).Where("id = ?", id).First(&device)
+	return &device
+}
+
 func (d *DeviceService) GetUserLoginStatus(userId int64) (int, error) {
 	var count int64
 	err := d.db.Model(&models.Device{}).Where("user_id = ?", userId).Count(&count).Error