From c55aca77e84607238c3354168a8642176c575ca1 Mon Sep 17 00:00:00 2001
From: gaoluyang <2820782392@qq.com>
Date: 星期二, 10 十一月 2026 11:42:16 +0800
Subject: [PATCH] feat(bpm): 待办任务站内信实时推送
---
yudao-framework/yudao-spring-boot-starter-websocket/src/main/java/cn/iocoder/yudao/framework/websocket/core/security/LoginUserHandshakeInterceptor.java | 7 +++++++
1 files changed, 7 insertions(+), 0 deletions(-)
diff --git a/yudao-framework/yudao-spring-boot-starter-websocket/src/main/java/cn/iocoder/yudao/framework/websocket/core/security/LoginUserHandshakeInterceptor.java b/yudao-framework/yudao-spring-boot-starter-websocket/src/main/java/cn/iocoder/yudao/framework/websocket/core/security/LoginUserHandshakeInterceptor.java
index d4fb515..30663bc 100644
--- a/yudao-framework/yudao-spring-boot-starter-websocket/src/main/java/cn/iocoder/yudao/framework/websocket/core/security/LoginUserHandshakeInterceptor.java
+++ b/yudao-framework/yudao-spring-boot-starter-websocket/src/main/java/cn/iocoder/yudao/framework/websocket/core/security/LoginUserHandshakeInterceptor.java
@@ -1,5 +1,6 @@
package cn.iocoder.yudao.framework.websocket.core.security;
+import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
import cn.iocoder.yudao.framework.security.core.LoginUser;
import cn.iocoder.yudao.framework.security.core.filter.TokenAuthenticationFilter;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
@@ -30,6 +31,12 @@
if (loginUser == null) {
return false;
}
+ // WebSocket 鍦板潃锛堝 /infra/ws锛変笉甯� /admin-api銆�/app-api 鍓嶇紑锛�
+ // 鍥犳 mock 鐧诲綍鏃� WebFrameworkUtils#getLoginUserType 鍙栦笉鍒扮敤鎴风被鍨嬨��
+ // 杩欓噷鍏滃簳涓虹鐞嗗憳绫诲瀷锛岄伩鍏� WebSocketSessionManagerImpl 鐢� null 浣滀负 key 璁板綍浼氳瘽鏃舵姏 NPE 鏂紑杩炴帴銆�
+ if (loginUser.getUserType() == null) {
+ loginUser.setUserType(UserTypeEnum.ADMIN.getValue());
+ }
WebSocketFrameworkUtils.setLoginUser(loginUser, attributes);
return true;
}
--
Gitblit v1.9.3