From 226b062ef301b61260efac126e1db373ace85b87 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Fri, 16 Jan 2026 09:01:43 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/dev' into dev
---
ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java | 60 ++++++++++++++++++++++++++++++++++++++++++++++++------------
1 files changed, 48 insertions(+), 12 deletions(-)
diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java b/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
index 93e7bc4..f719a7f 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java
@@ -42,15 +42,18 @@
@Resource
private UavAllManager uavAllManager;
@Resource
+ private NoFlyDistanceManager noFlyDistanceManager;
+ @Resource
private TokenService tokenService;
private final Map<Long, Long> lastDisconnectTime = new ConcurrentHashMap<>();
+
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) {
try {
String userIdStr = (String) session.getAttributes().get("userId");
Long userId = Long.parseLong(userIdStr);
String payload = message.getPayload();
- log.info("【cmd】Received【{}】message: {} size: {}", userId, payload,
+ log.info("【ws】【{}】Received message: {}, size: {}", userId, payload,
WebSocketSessionManager.getOnlineUserCount());
BaseMessage<?> baseMessage = JSON.parseObject(payload, BaseMessage.class);
MessageType messageType = MessageType.fromString(baseMessage.getType());
@@ -72,6 +75,9 @@
break;
case CONTROLLER:
handleControlMessage(payload, userId);
+ break;
+ case NO_FLY_DISTANCE:
+ handleNoFlyDistanceMessage(payload, userId);
break;
case ALL:
handleAllMessage(payload, userId);
@@ -271,9 +277,39 @@
}
}
+ /**
+ * 处理禁飞区距离
+ *
+ * @param payload 消息负载，包含订阅信息
+ * @param userId 用户ID，用于标识用户
+ */
+ private void handleNoFlyDistanceMessage(String payload, Long userId) {
+ // 解析接收到的JSON消息，将其转换为BaseMessage对象
+ BaseMessage<NoFlyDistanceMessage> baseMessage = JSON.parseObject(payload,
+ new TypeReference<BaseMessage<NoFlyDistanceMessage>>() {
+ });
+ // 获取消息的有效载荷部分
+ NoFlyDistanceMessage noFlyDistanceMessage = baseMessage.getPayload();
+
+ // 检查有效载荷是否存在
+ if (noFlyDistanceMessage != null) {
+ // 获取无人机盒子的序列号
+ String boxSn = noFlyDistanceMessage.getBoxSn();
+ // 根据消息中的订阅标志，决定是订阅还是退订无人机状态
+ Boolean subscribe = noFlyDistanceMessage.getSubscribe();
+ if (subscribe) {
+ // 执行订阅操作
+ noFlyDistanceManager.subscribe(boxSn, userId, 1);
+ } else {
+ // 执行退订操作
+ noFlyDistanceManager.unsubscribe(boxSn, userId);
+ }
+ }
+ }
+
+
@Override
public void afterConnectionEstablished(WebSocketSession session) {
- log.info("【cmd】Connected: {}", session.getId());
String userIdStr = (String) session.getAttributes().get("userId");
if (userIdStr == null || userIdStr.isBlank()) {
return;
@@ -281,19 +317,19 @@
Long userId = Long.parseLong(userIdStr);
lastDisconnectTime.remove(userId); // 清除断开标记
WebSocketSessionManager.addSession(userId, session);
- log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());
+ log.info("【ws】【{}】Connected,size: {}",userId, WebSocketSessionManager.getOnlineUserCount());
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
- log.info("【cmd】Disconnected: {}", session.getId());
+
String userIdStr = (String) session.getAttributes().get("userId");
if (userIdStr == null || userIdStr.isBlank()) {
return;
}
Long userId = Long.parseLong(userIdStr);
WebSocketSessionManager.removeSession(userId);
- log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());
+ log.info("【ws】【{}】Disconnected,size: {}",userId, WebSocketSessionManager.getOnlineUserCount());
long version = System.currentTimeMillis();
lastDisconnectTime.put(userId, version);
// 释放该用户所有 UAV 控制权
@@ -301,7 +337,7 @@
// 延迟10秒删除token（若期间没重连）
CompletableFuture.runAsync(() -> {
try {
- Thread.sleep(3_000); // 延迟3秒
+ Thread.sleep(5_000); // 延迟5秒
} catch (InterruptedException ignored) {
}
// 如果版本号不一致，说明用户曾重连 → 不执行删除
@@ -309,14 +345,14 @@
return;
}
if (!WebSocketSessionManager.containsUser(userId)) {
+ uavControlService.releaseAllControls(userId);
+ uavStreamManager.stopAllPullStreamByUser(userId);
+ uavStateManager.removeSubscribe(userId);
+ uavTelemetryManager.removeSubscribe(userId);
+ uavSignalManager.removeSubscribe(userId);
+ uavControlManager.removeSubscribe(userId);
LoginUser loginUser = tokenService.getLoginUserByUserId(userId);
if (loginUser != null) {
- uavControlService.releaseAllControls(userId);
- uavStreamManager.stopAllPullStreamByUser(userId);
- uavStateManager.removeSubscribe(userId);
- uavTelemetryManager.removeSubscribe(userId);
- uavSignalManager.removeSubscribe(userId);
- uavControlManager.removeSubscribe(userId);
tokenService.delLoginUser(loginUser.getToken(), userId);
log.info("【cmd】用户 [{}] 超时未重连，已删除 token", userId);
}
--
Gitblit v1.9.3