修改websockt身份验证处理

services/message_service.go

@@ -0,0 +1,26 @@
+package services
+
+import (
+	"log"
+
+	"github.com/gorilla/websocket"
+)
+
+type MessageService struct{}
+
+// websockt消息处理
+func (*MessageService) WebSocketMessage(conn *websocket.Conn) {
+	for {
+		mt, message, err := conn.ReadMessage()
+		if err != nil {
+			log.Printf("读取错误: %v (客户端: %s)", err, conn.RemoteAddr())
+			break
+		}
+		log.Printf("收到来自 %s 的消息: %s", conn.RemoteAddr(), string(message))
+		err = conn.WriteMessage(mt, message)
+		if err != nil {
+			log.Println("write:", err)
+			break
+		}
+	}
+}

services/profile_service.go

@@ -0,0 +1,32 @@
+package services
+
+import (
+	"go-desk-service/libs"
+	"go-desk-service/models"
+
+	"gorm.io/gorm"
+)
+
+type ProfileService struct {
+	db *gorm.DB
+}
+
+func InitProfileService() *ProfileService {
+	db := libs.GetDB()
+	return &ProfileService{
+		db: db,
+	}
+}
+
+// 获取个人信息
+func (p *ProfileService) GetInfo(params *models.Profile) models.Profile {
+	var user models.Profile
+	p.db.Model(&models.Profile{}).Where(params).First(&user)
+	return user
+}
+
+// 更新账号信息
+func (p *ProfileService) UpdateAccount(userId int8, params *models.Profile) bool {
+	p.db.Model(&models.Profile{}).Where("user_id = ?", userId).Updates(params)
+	return true
+}