From 2acf49278129e5870d8b775519a2bc1461294fb8 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Wed, 11 Mar 2026 08:55:46 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/dev' into dev

---
 ard-work/src/main/java/com/ard/work/service/control/UavControlService.java |   81 ++++++++++++++++++++++++++++++----------
 1 files changed, 60 insertions(+), 21 deletions(-)

diff --git a/ard-work/src/main/java/com/ard/work/service/control/UavControlService.java b/ard-work/src/main/java/com/ard/work/service/control/UavControlService.java
index 3014fe2..df7efa9 100644
--- a/ard-work/src/main/java/com/ard/work/service/control/UavControlService.java
+++ b/ard-work/src/main/java/com/ard/work/service/control/UavControlService.java
@@ -10,6 +10,7 @@
 import javax.annotation.Resource;
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 
 /**
@@ -25,17 +26,26 @@
     @Resource
     private ISysUserService sysUserService;
 
-    /** 当前飞行控制权限用户 */
+    /**
+     * 当前飞行控制权限用户
+     */
     private final ConcurrentHashMap<String, Long> uavFlyControlMap = new ConcurrentHashMap<>();
 
-    /** 当前载荷控制权限用户 */
+    /**
+     * 当前载荷控制权限用户
+     */
     private final ConcurrentHashMap<String, Long> uavPayloadControlMap = new ConcurrentHashMap<>();
 
-    /** 飞行控制请求 */
-    private final ConcurrentHashMap<String, Long> flyControlRequests = new ConcurrentHashMap<>();
+    /**
+     * 飞行控制请求:boxSn -> 多个请求者
+     */
+    private final ConcurrentHashMap<String, Set<Long>> flyControlRequests = new ConcurrentHashMap<>();
 
-    /** 载荷控制请求 */
-    private final ConcurrentHashMap<String, Long> payloadControlRequests = new ConcurrentHashMap<>();
+    /**
+     * 载荷控制请求
+     */
+    private final ConcurrentHashMap<String, Set<Long>> payloadControlRequests = new ConcurrentHashMap<>();
+
 
     /**
      * 请求获取 UAV 控制权限
@@ -49,9 +59,13 @@
     /**
      * 处理单个控制类型的请求
      */
+    /**
+     * 处理单个控制类型的请求(支持多请求者)
+     */
     private boolean requestSingleControl(String boxSn, UavControlType controlType, Long requester) {
+
         ConcurrentHashMap<String, Long> controlMap = getControlMap(controlType);
-        ConcurrentHashMap<String, Long> requestMap = getRequestMap(controlType);
+        ConcurrentHashMap<String, Set<Long>> requestMap = getRequestMap(controlType);
 
         Long controller = controlMap.get(boxSn);
 
@@ -76,49 +90,73 @@
 
             // 通知原控制者被抢占
             uavControlManager.sendControlTakeover(controller, requester, boxSn, controlType.getType());
-
-            // 通知新控制者获得控制
-            uavControlManager.sendControlApproval(requester, boxSn, controlType.getType(), requester);
             return true;
         }
 
-        // 优先级相同或更低 → 申请流程
-        requestMap.put(boxSn, requester);
+        // ===== 以下是关键改动点 =====
+
+        // 优先级相同或更低 → 加入请求集合(不覆盖)
+        Set<Long> requesters = requestMap.computeIfAbsent(boxSn, k -> ConcurrentHashMap.newKeySet());
+        requesters.add(requester);
+
+        // 通知当前控制者:有新的请求者
         uavControlManager.sendControlRequest(controller, boxSn, controlType.getType(), requester);
+
         return false;
     }
+
 
     /**
      * 处理 UAV 控制请求的响应
      */
-    public boolean handleControlResponse(String boxSn, boolean approve, String type) {
+    public boolean handleControlResponse(String boxSn, boolean approve, String type, Long requester) {
         UavControlType controlType = UavControlType.fromString(type);
         Long controller = SecurityUtils.getUserId();
-        return handleSingleControlResponse(boxSn, approve, controlType, controller);
+        return handleSingleControlResponse(boxSn, approve, controlType, controller, requester);
     }
+
 
     /**
      * 处理单个控制类型请求响应
      */
-    private boolean handleSingleControlResponse(String boxSn, boolean approve, UavControlType controlType,
-                                                Long controller) {
-        ConcurrentHashMap<String, Long> requestMap = getRequestMap(controlType);
-        Long requester = requestMap.get(boxSn);
+    private boolean handleSingleControlResponse(String boxSn,
+                                                boolean approve,
+                                                UavControlType controlType,
+                                                Long controller,
+                                                Long requester) {
 
-        if (requester == null || requester.equals(controller)) {
+        ConcurrentHashMap<String, Set<Long>> requestMap = getRequestMap(controlType);
+        Set<Long> requesters = requestMap.get(boxSn);
+
+        if (requesters == null || !requesters.contains(requester)) {
             throw new ControlException("无控制请求");
         }
 
         if (approve) {
             getControlMap(controlType).put(boxSn, requester);
+
+            // 通知被批准者
             uavControlManager.sendControlApproval(requester, boxSn, controlType.getType(), controller);
+
+            // 🚨 其他请求者自动拒绝(强烈建议)
+            for (Long other : requesters) {
+                if (!other.equals(requester)) {
+                    requesters.remove(other);
+                    uavControlManager.sendControlRejection(other, boxSn, controlType.getType(), controller);
+                }
+            }
         } else {
             uavControlManager.sendControlRejection(requester, boxSn, controlType.getType(), controller);
         }
 
-        requestMap.remove(boxSn);
+        requesters.remove(requester);
+        if (requesters.isEmpty()) {
+            requestMap.remove(boxSn);
+        }
+
         return true;
     }
+
 
     /**
      * 释放 UAV 控制权限
@@ -184,7 +222,7 @@
     /**
      * 根据控制类型获取对应请求 Map
      */
-    private ConcurrentHashMap<String, Long> getRequestMap(UavControlType controlType) {
+    private ConcurrentHashMap<String, Set<Long>> getRequestMap(UavControlType controlType) {
         switch (controlType) {
             case FLY:
                 return flyControlRequests;
@@ -195,6 +233,7 @@
         }
     }
 
+
     /**
      * 获取用户优先级
      */

--
Gitblit v1.9.3