lcj/xinnengyuan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

质安管理

Browse Source
This commit is contained in:
zzz
2025-12-16 18:46:37 +08:00
parent ebb8f09a75
commit dd6402d8b6
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/websocket/websocket/service/BigScreenWebSocketServer.java
View File

@ -59,18 +59,16 @@ public class BigScreenWebSocketServer {
public void onOpen(Session session) { public void onOpen(Session session) {
// 从连接URL的查询参数中获取订阅ID（客户端连接格式：ws://xxx/websocket/bigScreen?subscriptionId=123） // 从连接URL的查询参数中获取订阅ID（客户端连接格式：ws://xxx/websocket/bigScreen?subscriptionId=123）
Map<String, List<String>> params = session.getRequestParameterMap(); Map<String, List<String>> params = session.getRequestParameterMap();
Long timeType;
List<String> subscriptionIds = params.get("subscriptionId"); List<String> subscriptionIds = params.get("subscriptionId");
if (subscriptionIds != null && !subscriptionIds.isEmpty()) { if (subscriptionIds != null && !subscriptionIds.isEmpty()) {
this.currentSubscriptionId = subscriptionIds.get(0); // 取第一个订阅ID this.currentSubscriptionId = subscriptionIds.get(0); // 取第一个订阅ID
timeType = Long.parseLong(params.get("timeType").getFirst());
// 建立映射关系 // 建立映射关系
SUBSCRIPTION_SESSIONS.put(currentSubscriptionId, session); SUBSCRIPTION_SESSIONS.put(currentSubscriptionId, session);
SESSION_TO_SUBSCRIPTION.put(session.getId(), currentSubscriptionId); SESSION_TO_SUBSCRIPTION.put(session.getId(), currentSubscriptionId);
log.info("📌 客户端订阅成功！订阅ID：{}，会话ID：{}，当前订阅数：{}", log.info("📌 客户端订阅成功！订阅ID：{}，会话ID：{}，当前订阅数：{}",
currentSubscriptionId, session.getId(), SUBSCRIPTION_SESSIONS.size()); currentSubscriptionId, session.getId(), SUBSCRIPTION_SESSIONS.size());
} else { } else {
timeType = null;
log.warn("📌 客户端连接未携带订阅ID！会话ID：{}", session.getId()); log.warn("📌 客户端连接未携带订阅ID！会话ID：{}", session.getId());
} }
@ -98,6 +96,8 @@ public class BigScreenWebSocketServer {
case 1: case 1:
break; break;
case 2: case 2:
Long timeType;
timeType = Long.parseLong(params.get("timeType").getFirst());
//判断参数 //判断参数
if (timeType == null || (timeType != 1L && timeType != 2L && timeType != 3L)){ if (timeType == null || (timeType != 1L && timeType != 2L && timeType != 3L)){
throw new RuntimeException("时间类型参数错误"); throw new RuntimeException("时间类型参数错误");