修改websockt消息处理

2025-09-19 18:00:54 +08:00
parent c2f5721ddd
commit 0b1e312c84
2 changed files with 50 additions and 14 deletions
--- a/services/device_service.go
+++ b/services/device_service.go
@@ -110,9 +110,12 @@ func (d *DeviceService) GetAll(userId int64) []models.Device {
 	return device
 }

-func (d *DeviceService) Login(token string) {
+func (d *DeviceService) Login(token string) models.Device {
 	d.db.Model(&models.Device{}).Where("token = ?", token).Updates(&models.Device{
 		LoginStatus: 1,
 		UpdatedAt:   time.Now(),
 	})
+	var device models.Device
+	d.db.Model(&models.Device{}).Where("token = ?", token).First(&device)
+	return device
 }