From 46168fbb7c925b0ec04def176095e967720e684a Mon Sep 17 00:00:00 2001
From: chen.llin <1442464845@qq.comm>
Date: 星期五, 16 一月 2026 15:22:07 +0800
Subject: [PATCH] agv查询修复

---
 src/main/java/com/zy/asrs/task/handler/AgvHandler.java |  242 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 236 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/zy/asrs/task/handler/AgvHandler.java b/src/main/java/com/zy/asrs/task/handler/AgvHandler.java
index cbada90..4349f98 100644
--- a/src/main/java/com/zy/asrs/task/handler/AgvHandler.java
+++ b/src/main/java/com/zy/asrs/task/handler/AgvHandler.java
@@ -6,6 +6,8 @@
 import com.zy.asrs.entity.Task;
 import com.zy.asrs.entity.TaskLog;
 import com.zy.asrs.entity.WrkMast;
+import com.zy.asrs.entity.BasDevp;
+import com.zy.asrs.mapper.BasDevpMapper;
 import com.zy.asrs.mapper.BasStationMapper;
 import com.zy.asrs.mapper.WrkMastMapper;
 import com.zy.asrs.service.ApiLogService;
@@ -20,11 +22,9 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import javax.annotation.Resource;
-import java.util.ArrayList;
-import java.util.Collections;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.stream.Collectors;
 
 /**
@@ -52,7 +52,16 @@
     private BasStationMapper basStationMapper;
 
     @Resource
+    private BasDevpMapper basDevpMapper;
+
+    @Resource
     private AgvProperties agvProperties;
+
+    /**
+     * 绔欑偣杞璁℃暟鍣紝鐢ㄤ簬骞冲潎鍒嗛厤绔欑偣
+     * Key: 绔欑偣缁勬爣璇嗭紙濡� "east" 鎴� "west"锛夛紝Value: 褰撳墠杞绱㈠紩
+     */
+    private final Map<String, AtomicInteger> siteRoundRobinCounters = new ConcurrentHashMap<>();
 
     /**
      * 鍛煎彨agv鎼繍
@@ -64,6 +73,56 @@
         }
 
         for (Task task : taskList) {
+            // 濡傛灉浠诲姟娌℃湁鍒嗛厤绔欑偣锛屽厛鍒嗛厤绔欑偣
+            String staNo = task.getStaNo();
+            if (staNo == null || staNo.isEmpty()) {
+                Integer allocatedSite = allocateSiteForTask(task);
+                if (allocatedSite == null) {
+                    log.warn("浠诲姟ID锛歿}鏃犳硶鍒嗛厤绔欑偣锛岃烦杩囨湰娆″彂閫�", task.getId());
+                    continue; // 鏃犳硶鍒嗛厤绔欑偣锛岃烦杩囨湰娆″彂閫�
+                }
+                staNo = String.valueOf(allocatedSite);
+                task.setStaNo(staNo);
+                taskService.updateById(task);
+                log.info("浠诲姟ID锛歿}宸插垎閰嶇珯鐐癸細{}", task.getId(), staNo);
+            }
+            
+            // 妫�鏌ョ洰鏍囩珯鐐规槸鍚︽湁姝e湪鎼繍鐨勫悓绫诲瀷AGV浠诲姟锛堝嚭搴撳拰鍏ュ簱浜掍笉骞叉壈锛�
+            // 鍙湁鐘舵��8锛堝凡鍛煎彨AGV锛屾鍦ㄦ惉杩愶級鐨勪换鍔℃墠浼氶樆濉烇紝鐘舵��7锛堝緟鍛煎彨锛夌殑浠诲姟涓嶉樆濉�
+            // 杩欐牱鍙互閬垮厤鎵�鏈変换鍔¢兘鍗″湪鍛煎彨鐘舵�侊紝鎸塱d鏈�灏忕殑浼樺厛鍛煎彨
+            if (staNo != null && !staNo.isEmpty() && task.getIoType() != null) {
+                // 鏍规嵁褰撳墠浠诲姟绫诲瀷锛屽彧妫�鏌ュ悓绫诲瀷鐨勬鍦ㄦ惉杩愪换鍔★紙鐘舵��8锛�
+                // 鍏ュ簱浠诲姟锛坕oType < 100锛夛細鍙鏌ュ叆搴撶被鍨嬬殑姝e湪鎼繍浠诲姟
+                // 鍑哄簱浠诲姟锛坕oType >= 100锛夛細鍙鏌ュ嚭搴撶被鍨嬬殑姝e湪鎼繍浠诲姟
+                List<Integer> ioTypes;
+                String taskType;
+                if (task.getIoType() < 100) {
+                    // 鍏ュ簱浠诲姟锛氬彧妫�鏌ュ叆搴撶被鍨嬶紙1, 10, 53, 57锛�
+                    ioTypes = Arrays.asList(1, 10, 53, 57);
+                    taskType = "鍏ュ簱";
+                } else {
+                    // 鍑哄簱浠诲姟锛氬彧妫�鏌ュ嚭搴撶被鍨嬶紙101, 110, 103, 107锛�
+                    ioTypes = Arrays.asList(101, 110, 103, 107);
+                    taskType = "鍑哄簱";
+                }
+                
+                // 鍙鏌ョ姸鎬佷负8锛堝凡鍛煎彨AGV锛屾鍦ㄦ惉杩愶級鐨勫悓绫诲瀷浠诲姟
+                List<Task> transportingTasks = taskService.selectList(
+                    new EntityWrapper<Task>()
+                        .eq("sta_no", staNo)
+                        .eq("task_type", "agv")
+                        .eq("wrk_sts", 8L) // 鍙鏌ユ鍦ㄦ惉杩愮姸鎬佺殑浠诲姟
+                        .in("io_type", ioTypes)
+                        .ne("id", task.getId()) // 鎺掗櫎褰撳墠浠诲姟鏈韩
+                );
+                
+                if (!transportingTasks.isEmpty()) {
+                    log.info("绔欑偣{}鏈墈}涓鍦ㄦ惉杩愮殑{}AGV浠诲姟锛岃烦杩囨湰娆″彂閫侊紝绛夊緟鎼繍瀹屾垚銆傚綋鍓嶄换鍔D锛歿}", 
+                            staNo, transportingTasks.size(), taskType, task.getId());
+                    continue; // 璺宠繃鏈鍙戦�侊紝绛夊緟涓嬫
+                }
+            }
+            
             // 鍛煎彨agv
             String response = "";
             boolean success = false;
@@ -197,6 +256,176 @@
     }
 
     /**
+     * 涓轰换鍔″垎閰嶇珯鐐癸紙瀹氭椂浠诲姟涓皟鐢級
+     * @param task 浠诲姟瀵硅薄
+     * @return 鍒嗛厤鐨勭珯鐐圭紪鍙凤紝濡傛灉鏃犳硶鍒嗛厤鍒欒繑鍥瀗ull
+     */
+    private Integer allocateSiteForTask(Task task) {
+        // 鏍规嵁浠诲姟鐨刬nvWh锛堟満鍣ㄤ汉缁勶級鍒ゆ柇鏄笢渚ц繕鏄タ渚�
+        String robotGroup = task.getInvWh();
+        List<String> targetStations;
+        String groupKey;
+        
+        if (robotGroup != null && robotGroup.equals(agvProperties.getRobotGroupEast())) {
+            // 涓滀晶绔欑偣
+            targetStations = agvProperties.getEastStations();
+            groupKey = "east";
+        } else if (robotGroup != null && robotGroup.equals(agvProperties.getRobotGroupWest())) {
+            // 瑗夸晶绔欑偣
+            targetStations = agvProperties.getWestStations();
+            groupKey = "west";
+        } else {
+            // 榛樿浣跨敤涓滀晶
+            targetStations = agvProperties.getEastStations();
+            groupKey = "east";
+            log.warn("浠诲姟ID锛歿}鐨勬満鍣ㄤ汉缁剓}鏈瘑鍒紝浣跨敤榛樿涓滀晶绔欑偣", task.getId(), robotGroup);
+        }
+        
+        if (targetStations.isEmpty()) {
+            log.warn("浠诲姟ID锛歿}娌℃湁鍙敤鐨勭洰鏍囩珯鐐归厤缃�", task.getId());
+            return null;
+        }
+        
+        // 灏嗙珯鐐瑰瓧绗︿覆鍒楄〃杞崲涓烘暣鏁板垪琛�
+        List<Integer> siteIntList = targetStations.stream()
+                .map(Integer::parseInt)
+                .collect(Collectors.toList());
+        
+        // 鍒ゆ柇鑳藉叆绔欑偣锛坕n_enable="Y"琛ㄧず鑳藉叆锛�
+        List<Integer> sites = basDevpMapper.selectList(
+                new EntityWrapper<BasDevp>()
+                        .eq("in_enable", "Y")
+                        .in("dev_no", siteIntList)
+        ).stream().map(BasDevp::getDevNo).collect(Collectors.toList());
+        
+        if (sites.isEmpty()) {
+            log.warn("浠诲姟ID锛歿}娌℃湁鑳藉叆绔欑偣", task.getId());
+            return null;
+        }
+        
+        // 鑾峰彇娌℃湁鍑哄簱浠诲姟鐨勭珯鐐�
+        List<Integer> canInSites = basDevpMapper.getCanInSites(sites);
+        if (canInSites.isEmpty()) {
+            log.warn("浠诲姟ID锛歿}娌℃湁鍙叆绔欑偣锛堣绛夊緟鍑哄簱瀹屾垚锛�", task.getId());
+            return null;
+        }
+        
+        // 瀵绘壘鍏ュ簱浠诲姟鏈�灏戠殑绔欑偣锛堜笖蹇呴』in_enable="Y"鑳藉叆 鍜� canining="Y"鍙叆锛�
+        List<BasDevp> devList = basDevpMapper.selectList(new EntityWrapper<BasDevp>()
+                .in("dev_no", canInSites)
+                .eq("in_enable", "Y")
+                .eq("canining", "Y")
+        );
+        
+        if (devList.isEmpty()) {
+            log.warn("浠诲姟ID锛歿}娌℃湁鍙叆绔欑偣锛坕n_enable='Y'涓攃anining='Y'锛�", task.getId());
+            return null;
+        }
+        
+        // 鍏堟寜瑙勫垯鎺掑簭锛堝叆搴撲换鍔℃暟鎺掑簭锛�
+        devList.sort(Comparator.comparing(BasDevp::getInQty));
+        
+        // 鏍规嵁浠诲姟绫诲瀷纭畾瑕佹鏌ョ殑io_type鍒楄〃
+        Integer taskIoType = task.getIoType();
+        List<Integer> checkIoTypes = null;
+        String taskTypeName = "";
+        if (taskIoType != null) {
+            if (taskIoType < 100) {
+                // 鍏ュ簱浠诲姟锛氬彧妫�鏌ュ叆搴撶被鍨嬶紙1, 10, 53, 57锛�
+                checkIoTypes = Arrays.asList(1, 10, 53, 57);
+                taskTypeName = "鍏ュ簱";
+            } else {
+                // 鍑哄簱浠诲姟锛氬彧妫�鏌ュ嚭搴撶被鍨嬶紙101, 110, 103, 107锛�
+                checkIoTypes = Arrays.asList(101, 110, 103, 107);
+                taskTypeName = "鍑哄簱";
+            }
+        }
+        
+        // 绛涢�夊嚭浠诲姟鏁版渶灏戠殑绔欑偣鍒楄〃锛堟寜瑙勫垯鎺掑簭鍚庣殑鍊欓�夌珯鐐癸級
+        int minInQty = devList.get(0).getInQty();
+        List<BasDevp> minTaskSites = devList.stream()
+                .filter(dev -> dev.getInQty() == minInQty)
+                .collect(Collectors.toList());
+        
+        // 鏍规嵁閰嶇疆閫夋嫨鍒嗛厤绛栫暐锛岀‘瀹氫紭鍏堝垎閰嶇殑绔欑偣椤哄簭
+        List<BasDevp> orderedSites = new ArrayList<>();
+        String strategy = agvProperties.getSiteAllocation().getStrategy();
+        boolean enableRoundRobin = agvProperties.getSiteAllocation().isEnableRoundRobin();
+        
+        if (minTaskSites.size() > 1 && enableRoundRobin && "round-robin".equals(strategy)) {
+            // 杞鍒嗛厤锛氬厛鎸夎疆璇㈢瓥鐣ユ帓搴�
+            AtomicInteger counter = siteRoundRobinCounters.computeIfAbsent(groupKey, k -> new AtomicInteger(0));
+            int startIndex = counter.get() % minTaskSites.size();
+            // 灏嗚疆璇㈤�変腑鐨勭珯鐐规斁鍦ㄦ渶鍓嶉潰
+            orderedSites.addAll(minTaskSites.subList(startIndex, minTaskSites.size()));
+            orderedSites.addAll(minTaskSites.subList(0, startIndex));
+            // 娣诲姞鍏朵粬绔欑偣锛堜换鍔℃暟鏇村鐨勶級
+            orderedSites.addAll(devList.stream()
+                    .filter(dev -> dev.getInQty() > minInQty)
+                    .collect(Collectors.toList()));
+            log.debug("浣跨敤杞鍒嗛厤绛栫暐锛岀珯鐐圭粍锛歿}锛岃疆璇㈣捣濮嬬储寮曪細{}", groupKey, startIndex);
+        } else if (minTaskSites.size() > 1 && enableRoundRobin && "random".equals(strategy)) {
+            // 闅忔満鍒嗛厤锛氬厛闅忔満鎺掑簭浠诲姟鏁版渶灏戠殑绔欑偣
+            List<BasDevp> shuffledMinSites = new ArrayList<>(minTaskSites);
+            Collections.shuffle(shuffledMinSites);
+            orderedSites.addAll(shuffledMinSites);
+            // 娣诲姞鍏朵粬绔欑偣锛堜换鍔℃暟鏇村鐨勶級
+            orderedSites.addAll(devList.stream()
+                    .filter(dev -> dev.getInQty() > minInQty)
+                    .collect(Collectors.toList()));
+            log.debug("浣跨敤闅忔満鍒嗛厤绛栫暐");
+        } else {
+            // 榛樿锛氭寜鍏ュ簱浠诲姟鏁版帓搴忥紙宸茬粡鎺掑簭濂戒簡锛�
+            orderedSites = devList;
+        }
+        
+        // 渚濇妫�鏌ユ瘡涓珯鐐规槸鍚﹀湪鎼繍锛屾壘鍒扮涓�涓┖闂茬珯鐐瑰氨鍒嗛厤
+        BasDevp selectedSite = null;
+        for (BasDevp dev : orderedSites) {
+            String staNo = String.valueOf(dev.getDevNo());
+            
+            // 濡傛灉浠诲姟绫诲瀷涓嶄负绌猴紝妫�鏌ヨ绔欑偣鏄惁鏈夋鍦ㄦ惉杩愮殑鍚岀被鍨嬩换鍔�
+            boolean isTransporting = false;
+            if (checkIoTypes != null && !checkIoTypes.isEmpty()) {
+                List<Task> transportingTasks = taskService.selectList(
+                    new EntityWrapper<Task>()
+                        .eq("sta_no", staNo)
+                        .eq("task_type", "agv")
+                        .eq("wrk_sts", 8L) // 鍙鏌ユ鍦ㄦ惉杩愮姸鎬佺殑浠诲姟
+                        .in("io_type", checkIoTypes)
+                );
+                isTransporting = !transportingTasks.isEmpty();
+                
+                if (isTransporting) {
+                    log.debug("绔欑偣{}鏈墈}涓鍦ㄦ惉杩愮殑{}AGV浠诲姟锛屾鏌ヤ笅涓�涓珯鐐�", 
+                        staNo, transportingTasks.size(), taskTypeName);
+                    continue; // 璇ョ珯鐐规鍦ㄦ惉杩愶紝妫�鏌ヤ笅涓�涓珯鐐�
+                }
+            }
+            
+            // 鎵惧埌绗竴涓┖闂茬珯鐐癸紝鍒嗛厤
+            selectedSite = dev;
+            log.info("浠诲姟ID锛歿}鎸夎鍒欏簲鍒嗛厤鍒扮珯鐐箋}锛岃绔欑偣绌洪棽锛屽垎閰嶆垚鍔�", task.getId(), staNo);
+            break;
+        }
+        
+        // 濡傛灉鎵�鏈夌珯鐐归兘鍦ㄦ惉杩愶紝鍒欎笉鍒嗛厤绔欑偣
+        if (selectedSite == null) {
+            log.warn("浠诲姟ID锛歿}鐨勬墍鏈夊�欓�夌珯鐐归兘鏈夋鍦ㄦ惉杩愮殑{}浠诲姟锛屾殏涓嶅垎閰嶇珯鐐癸紝绛夊緟绌洪棽", 
+                task.getId(), taskIoType != null && taskIoType < 100 ? "鍏ュ簱" : "鍑哄簱");
+            return null;
+        }
+        
+        Integer endSite = selectedSite.getDevNo();
+        
+        // 鍏ュ簱鏆傚瓨+1
+        basDevpMapper.incrementInQty(endSite);
+        
+        log.info("浠诲姟ID锛歿}宸插垎閰嶇珯鐐癸細{}", task.getId(), endSite);
+        return endSite;
+    }
+
+    /**
      * 鏍规嵁绔欑偣缂栧彿鍒ゆ柇鏈哄櫒浜虹粍
      * @param staNo 绔欑偣缂栧彿
      * @return 鏈哄櫒浜虹粍鍚嶇О
@@ -227,10 +456,11 @@
     @Transactional(rollbackFor = Exception.class)
     public void moveTaskToHistory(List<Task> taskList) {
 
-        // 鍐欏叆鍘嗗彶琛�
+        // 鍐欏叆鍘嗗彶琛紝淇濇寔ID涓�鑷�
         for (Task task : taskList) {
             TaskLog log = new TaskLog();
             BeanUtils.copyProperties(task, log);
+            // 淇濇寔ID涓�鑷达紝涓嶈缃负null
             taskLogService.insert(log);
         }
 

--
Gitblit v1.9.1