From 4ac9edd4c0030d2daafd712cd14ef9080b2f1e43 Mon Sep 17 00:00:00 2001
From: liusuyi <1951119284@qq.com>
Date: Tue, 10 Feb 2026 11:52:40 +0800
Subject: [PATCH] 优化

---
 ard-work/src/main/java/com/ard/work/component/websocket/handler/UavAllManager.java |   20 ++++++++++++++++++--
 1 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavAllManager.java b/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavAllManager.java
index 72c2fe7..8a1c7c2 100644
--- a/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavAllManager.java
+++ b/ard-work/src/main/java/com/ard/work/component/websocket/handler/UavAllManager.java
@@ -3,12 +3,16 @@
 import com.ard.common.constant.CacheConstants;
 import com.ard.common.core.domain.entity.SysUser;
 import com.ard.common.core.redis.RedisCache;
+import com.ard.work.config.RouteStateMachineService;
 import com.ard.work.domain.cache.UavSignalInfoCache;
 import com.ard.work.domain.cache.UavStateCache;
 import com.ard.work.domain.cache.UavTelemetryDataCache;
+import com.ard.work.domain.cache.UavWaylineCache;
 import com.ard.work.domain.dto.signal.SignalInfo;
 import com.ard.work.domain.dto.state.UavState;
 import com.ard.work.domain.dto.telemetry.TelemetryData;
+import com.ard.work.domain.dto.wayline.WaylineInfo;
+import com.ard.work.domain.enums.RouteState;
 import com.ard.work.service.control.UavControlService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -27,26 +31,38 @@
     private UavControlService uavControlService;
     @Autowired
     private RedisCache redisCache;
+    @Autowired
+    private RouteStateMachineService stateMachineService;
 
     @Override
     protected Map<String, Object> getData(String boxSn) {
         log.debug("获取无人机所有数据");
         Map<String, Object> map = new HashMap<>();
         map.put("boxSn", boxSn);
+
+        WaylineInfo waylineInfo = UavWaylineCache.get(boxSn);
+        if (waylineInfo != null) {
+            RouteState currentState = stateMachineService.getCurrentState(boxSn);
+            waylineInfo.setRouteState(currentState);
+        }
+        map.put("wayline", waylineInfo);
+
         UavState uavState = UavStateCache.get(boxSn);
         map.put("state", uavState);
+
         TelemetryData uavTelemetryData = UavTelemetryDataCache.get(boxSn);
         map.put("telemetry", uavTelemetryData);
+
         SignalInfo uavSignalInfo = UavSignalInfoCache.get(boxSn);
         map.put("signal", uavSignalInfo);
         // 获取当前控制飞行的用户ID
         Long flyUserId = uavControlService.getCurrentController(boxSn, "fly");
-        SysUser flyUser = redisCache.getCacheObject(CacheConstants.SYS_USER_KEY + flyUserId);
+        SysUser flyUser = redisCache.getCacheMapValue(CacheConstants.SYS_USER, flyUserId.toString());
         // 将飞行当前控制者用户ID添加到映射中
         map.put("flyController", flyUser);
         // 获取当前控制载荷的用户ID
         Long payloadUserId = uavControlService.getCurrentController(boxSn, "payload");
-        SysUser payloadUser = redisCache.getCacheObject(CacheConstants.SYS_USER_KEY + payloadUserId);
+        SysUser payloadUser = redisCache.getCacheMapValue(CacheConstants.SYS_USER, payloadUserId.toString());
         // 将载荷当前控制者用户ID添加到映射中
         map.put("payloadController", payloadUser);
         return map;

--
Gitblit v1.9.3