From 8773c724919714845d4426a54d7b5aab8a9c0cae Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Thu, 20 Mar 2025 16:31:26 +0800
Subject: [PATCH] 优化结构
---
ard-work/src/main/java/com/ard/tianyu/component/websocket/handler/TyMessageHandler.java | 18 +++++++++---------
1 files changed, 9 insertions(+), 9 deletions(-)
diff --git a/ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java b/ard-work/src/main/java/com/ard/tianyu/component/websocket/handler/TyMessageHandler.java
similarity index 94%
rename from ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java
rename to ard-work/src/main/java/com/ard/tianyu/component/websocket/handler/TyMessageHandler.java
index 082386f..f5094b0 100644
--- a/ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java
+++ b/ard-work/src/main/java/com/ard/tianyu/component/websocket/handler/TyMessageHandler.java
@@ -1,17 +1,17 @@
-package com.ard.tianyu.websocket.handler;
+package com.ard.tianyu.component.websocket.handler;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.TypeReference;
+import com.ard.tianyu.component.websocket.WebSocketSessionManager;
+import com.ard.tianyu.domain.message.*;
import com.ard.tianyu.service.control.UavControlService;
-import com.ard.tianyu.websocket.domain.*;
-import com.ard.tianyu.websocket.utils.WebSocketUtil;
+import com.ard.tianyu.component.websocket.utils.WebSocketUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
-
import javax.annotation.Resource;
@@ -39,7 +39,7 @@
Long userId = Long.parseLong(userIdStr);
String payload = message.getPayload();
log.info("【cmd】Received message: {}", payload);
- log.info("【cmd】size: {}", WebSocketUtil.ONLINE_USER_SESSIONS.size());
+ log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());
BaseMessage<?> baseMessage = JSON.parseObject(payload, BaseMessage.class);
switch (baseMessage.getType()) {
@@ -243,8 +243,8 @@
log.info("【cmd】Connected: {}", session.getId());
String userIdStr = (String) session.getAttributes().get("userId");
Long userId = Long.parseLong(userIdStr);
- WebSocketUtil.ONLINE_USER_SESSIONS.put(userId, session);
- log.info("【cmd】size: {}", WebSocketUtil.ONLINE_USER_SESSIONS.size());
+ WebSocketSessionManager.addSession(userId, session);
+ log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());
}
@Override
@@ -252,8 +252,8 @@
log.info("【cmd】Disconnected: {}", session.getId());
String userIdStr = (String) session.getAttributes().get("userId");
Long userId = Long.parseLong(userIdStr);
- WebSocketUtil.ONLINE_USER_SESSIONS.remove(userId);
- log.info("【cmd】size: {}", WebSocketUtil.ONLINE_USER_SESSIONS.size());
+ WebSocketSessionManager.removeSession(userId);
+ log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());
// 停止用户的所有流
uavStreamManager.stopAllPullStreamByUser(userId);
if (userId != null) {
--
Gitblit v1.9.3