优化tcp连接

liusuyi

2025-10-17 f9ee3fa619d478e5e592dfaa99a5b993feb6c67a

 ard-work/src/main/java/com/ard/work/component/websocket/handler/TyMessageHandler.java

@@ -23,6 +23,8 @@
    @Resource
    private UavControlService uavControlService;
    @Resource
    private UavStreamManager uavStreamManager;
    @Resource
    private UavStateManager uavStateManager;
    @Resource
    private UavTelemetryManager uavTelemetryManager;
@@ -44,8 +46,8 @@
            BaseMessage<?> baseMessage = JSON.parseObject(payload, BaseMessage.class);
            MessageType messageType = MessageType.fromString(baseMessage.getType());
            switch (messageType) {
                case STREAM: // 直播地址
                    
                case STREAM:
                    handleStreamMessage(payload, userId);
                    break;
                case STATE:
                    handleStateMessage(payload, userId);
@@ -65,6 +67,34 @@
            }
        } catch (Exception e) {
            log.error("Error handling message: {}", message, e);
        }
    }

    /**
     * 处理流消息
     * <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();

            // 根据Subscribe决定操作类型：true表示开始拉流，false表示停止拉流
            if (uavStreamMessage.getSubscribe()) {
                uavStreamManager.subscribe(boxSn, userId);
            } else {
                uavStreamManager.unsubscribe(boxSn, userId);
            }
        }
    }

@@ -233,11 +263,9 @@
        WebSocketSessionManager.removeSession(userId);
        log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount());

        if (userId != null) {
            // 释放该用户所有 UAV 控制权
            uavControlService.releaseAllControls(userId);
        }

        // 释放该用户所有 UAV 控制权
        uavControlService.releaseAllControls(userId);
        uavStreamManager.stopAllPullStreamByUser(userId);
        uavStateManager.removeSubscribe(userId);
        uavTelemetryManager.removeSubscribe(userId);
        uavSignalManager.removeSubscribe(userId);