From cdb8297d0f08ebff314ae726c5b21def9887ff75 Mon Sep 17 00:00:00 2001
From: whyclxw <810412026@qq.com>
Date: 星期二, 29 十月 2024 10:25:18 +0800
Subject: [PATCH] websocket默认101全改为从session获取

---
 src/main/java/com/whyc/webSocket/DevInfSocket.java |    7 ++++---
 1 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/whyc/webSocket/DevInfSocket.java b/src/main/java/com/whyc/webSocket/DevInfSocket.java
index 65c9a46..0b37f68 100644
--- a/src/main/java/com/whyc/webSocket/DevInfSocket.java
+++ b/src/main/java/com/whyc/webSocket/DevInfSocket.java
@@ -3,6 +3,7 @@
 import com.whyc.config.WebSocketConfig;
 import com.whyc.dto.DevInfDto;
 import com.whyc.dto.Response;
+import com.whyc.pojo.db_user.UserInf;
 import com.whyc.service.DevInfService;
 import com.whyc.util.ActionUtil;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -48,9 +49,9 @@
     @OnMessage
     public void onMessage(Session session, String message) {
         DevInfDto devInfDto = ActionUtil.getGson("yyyy-MM-dd").fromJson(message, DevInfDto.class);
-        //UserInf user = (UserInf) this.httpSession.getAttribute("user");
-        //final int userId = user.getUid();
-        final int userId = 101;
+        UserInf user = (UserInf) this.httpSession.getAttribute("user");
+        final int userId = user.getUid();
+        //final int userId = 101;
         thread = new Thread("Thread_DinfSocket") {
             @Override
             public void run() {

--
Gitblit v1.9.1