优化

liusuyi

2025-12-18 3814365a9c3380161886dc0b6cee6774783b0eb9

 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,16 +31,28 @@
    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