From 99b201ea344b2b21efebb6bf8b72397714288783 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Mon, 10 Mar 2025 13:15:19 +0800
Subject: [PATCH] 增加websocket状态管理用来推送
---
ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java | 98 ++++++++++++++++++++++++++++++++++++++-----------
1 files changed, 76 insertions(+), 22 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/websocket/handler/TyMessageHandler.java
index b74d892..7da65b5 100644
--- a/ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java
+++ b/ard-work/src/main/java/com/ard/tianyu/websocket/handler/TyMessageHandler.java
@@ -3,9 +3,9 @@
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.TypeReference;
import com.ard.tianyu.service.control.UavControlService;
-import com.ard.tianyu.service.stream.UavStreamManager;
import com.ard.tianyu.websocket.domain.BaseMessage;
import com.ard.tianyu.websocket.domain.UavStreamMessage;
+import com.ard.tianyu.websocket.domain.UavStateMessage;
import com.ard.tianyu.websocket.utils.WebSocketUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
@@ -13,7 +13,6 @@
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
-
import javax.annotation.Resource;
@@ -25,57 +24,112 @@
private UavStreamManager uavStreamManager;
@Resource
private UavControlService uavControlService;
+ @Resource
+ private UavStateManager uavStateManager;
+
@Override
public void handleTextMessage(WebSocketSession session, TextMessage message) {
- // 处理接收到的消息
- String userIdStr = (String) session.getAttributes().get("userId");
+ String userIdStr = (String) session.getAttributes().get("userId");
Long userId = Long.parseLong(userIdStr);
String payload = message.getPayload();
log.info("【cmd】Received message: {}", payload);
log.info("【cmd】size: {}", WebSocketUtil.ONLINE_USER_SESSIONS.size());
+
BaseMessage<?> baseMessage = JSON.parseObject(payload, BaseMessage.class);
switch (baseMessage.getType()) {
case "stream":
- BaseMessage<UavStreamMessage> streamMessage = JSON.parseObject(payload,
- new TypeReference<BaseMessage<UavStreamMessage>>() {
- });
- UavStreamMessage uavStreamMessage = streamMessage.getPayload();
- if (uavStreamMessage != null) {
- String boxSn = uavStreamMessage.getBoxSn();
- if (uavStreamMessage.getPullType().equals(1)) {
- uavStreamManager.startPullStream(boxSn, userId);
- } else {
- uavStreamManager.stopPullStream(boxSn, userId);
- }
- }
+ handleStreamMessage(payload, userId);
+ break;
+ case "state":
+ handleStateMessage(payload, userId);
break;
}
-
}
+
+ /**
+ * 处理流消息
+ * <p>
+ * 该方法解析给定的负载数据，以获取无人机流消息，并根据消息类型决定是否开始或停止流
+ *
+ * @param payload 消息的负载数据，通常为JSON字符串
+ * @param userId 接收消息的用户ID
+ */
+ private void handleStreamMessage(String payload, Long userId) {
+ // 解析JSON负载数据为BaseMessage对象，包含UavStreamMessage类型的有效载荷
+ BaseMessage<UavStreamMessage> streamMessage = JSON.parseObject(payload,
+ new TypeReference<BaseMessage<UavStreamMessage>>() {
+ });
+ UavStreamMessage uavStreamMessage = streamMessage.getPayload();
+
+ // 检查解析后的UavStreamMessage对象是否非空
+ if (uavStreamMessage != null) {
+ String boxSn = uavStreamMessage.getBoxSn();
+
+ // 根据pullType决定操作类型：1表示开始流，其他值表示停止流
+ if (uavStreamMessage.getPullType().equals(1)) {
+ uavStreamManager.startPullStream(boxSn, userId);
+ } else {
+ uavStreamManager.stopPullStream(boxSn, userId);
+ }
+ }
+ }
+
+ /**
+ * 处理状态消息
+ * 根据接收到的消息内容，决定无人机状态信息的订阅或退订
+ *
+ * @param payload 消息负载，包含订阅信息
+ * @param userId 用户ID，用于标识用户
+ */
+ private void handleStateMessage(String payload, Long userId) {
+ // 解析接收到的JSON消息，将其转换为BaseMessage对象
+ BaseMessage<UavStateMessage> stateMessage = JSON.parseObject(payload,
+ new TypeReference<BaseMessage<UavStateMessage>>() {
+ });
+ // 获取消息的有效载荷部分
+ UavStateMessage uavStateMessage = stateMessage.getPayload();
+
+ // 检查有效载荷是否存在
+ if (uavStateMessage != null) {
+ // 获取无人机盒子的序列号
+ String boxSn = uavStateMessage.getBoxSn();
+ // 根据消息中的订阅标志，决定是订阅还是退订无人机状态
+ Boolean subscribe = uavStateMessage.getSubscribe();
+ if (subscribe) {
+ // 执行订阅操作
+ uavStateManager.subscribeState(boxSn, userId);
+ } else {
+ // 执行退订操作
+ uavStateManager.unsubscribeState(boxSn, userId);
+ }
+ }
+ }
+
@Override
public void afterConnectionEstablished(WebSocketSession session) {
log.info("【cmd】Connected: {}", session.getId());
- String userIdStr = (String) session.getAttributes().get("userId");
+ 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());
-
}
@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
log.info("【cmd】Disconnected: {}", session.getId());
- String userIdStr = (String) session.getAttributes().get("userId");
+ 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());
- //停止用户的所有流
+ // 停止用户的所有流
uavStreamManager.stopAllPullStreamByUser(userId);
if (userId != null) {
// 释放该用户所有 UAV 控制权
uavControlService.releaseAllControls(userId);
}
+
+ uavStateManager.removeSubscribeState(userId);
}
-}
\ No newline at end of file
+}
--
Gitblit v1.9.3