增加状态统一订阅 增加载荷和飞控用户同时获取

liusuyi

2025-03-19 d267efe74181bbc89e7ecda1032f11965984769b

 ard-work/src/main/java/com/ard/tianyu/service/control/UavControlService.java

@@ -1,12 +1,14 @@
package com.ard.tianyu.service.control;

import com.ard.common.exception.NoPermissionException;
import com.ard.common.exception.ControlException;
import com.ard.common.utils.SecurityUtils;
import com.ard.tianyu.websocket.domain.UavControlType;
import com.ard.tianyu.websocket.handler.UavControlManager;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
@@ -66,7 +68,7 @@
            requestMap.put(boxSn, requester);
            uavControlManager.sendControlRequest(controller, boxSn, controlType.getType(), requester);
        } else {
            throw new NoPermissionException("当前已拥有控制权");
            throw new ControlException("当前已拥有控制权");
        }
        return false;
    }
@@ -94,7 +96,7 @@
        Long requester = requestMap.get(boxSn);

        if (requester == null || requester.equals(controller)) {
            throw new NoPermissionException("无控制请求");
            throw new ControlException("无控制请求");
        }

        if (approve) {
@@ -128,7 +130,7 @@
        ConcurrentHashMap<String, Long> controlMap = getControlMap(controlType);

        if (!controller.equals(controlMap.get(boxSn))) {
            throw new NoPermissionException("没有该设备的" + controlType.getType().toUpperCase() + "控制权");
            throw new ControlException("没有该设备的" + controlType.getType().toUpperCase() + "控制权");
        }

        controlMap.remove(boxSn);
@@ -158,7 +160,14 @@
        UavControlType controlType = UavControlType.fromString(type);
        return getControlMap(controlType).get(boxSn);
    }

    public Map<String,Long> getCurrentController(String boxSn) {
        Long flyController = uavFlyControlMap.get(boxSn);
        Map<String,Long> map=new HashMap<>();
        map.put("flyController",flyController);
        Long payloadController = uavPayloadControlMap.get(boxSn);
        map.put("payloadController",payloadController);
        return map;
    }
    /**
     * 根据控制类型获取对应的控制 Map
     */