From 03b9be24e5a9bc5cf746beac1ac43330183c2c63 Mon Sep 17 00:00:00 2001
From: chen.llin <1442464845@qq.comm>
Date: 星期四, 15 一月 2026 13:32:37 +0800
Subject: [PATCH] agv定时任务后分配站点
---
src/main/java/com/zy/asrs/task/handler/AgvHandler.java | 159 +++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 154 insertions(+), 5 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..60afb65 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,37 @@
}
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);
+ }
+
+ // 妫�鏌ョ洰鏍囩珯鐐规槸鍚︽湁鏈畬鎴愮殑AGV浠诲姟
+ // 濡傛灉绔欑偣鏈夋湭瀹屾垚鐨勪换鍔★紝鍒欒烦杩囨湰娆″彂閫侊紝绛夊緟涓嬫
+ if (staNo != null && !staNo.isEmpty()) {
+ List<Task> unfinishedTasks = taskService.selectList(new EntityWrapper<Task>()
+ .eq("sta_no", staNo)
+ .eq("task_type", "agv")
+ .ne("id", task.getId()) // 鎺掗櫎褰撳墠浠诲姟鏈韩
+ .last("AND wrk_sts NOT IN (5, 15)") // 鎺掗櫎宸插畬鎴愮姸鎬�
+ );
+
+ if (!unfinishedTasks.isEmpty()) {
+ log.info("绔欑偣{}鏈墈}涓湭瀹屾垚鐨凙GV浠诲姟锛岃烦杩囨湰娆″彂閫侊紝绛夊緟浠诲姟瀹屾垚銆傚綋鍓嶄换鍔D锛歿}",
+ staNo, unfinishedTasks.size(), task.getId());
+ continue; // 璺宠繃鏈鍙戦�侊紝绛夊緟涓嬫
+ }
+ }
+
// 鍛煎彨agv
String response = "";
boolean success = false;
@@ -197,6 +237,115 @@
}
/**
+ * 涓轰换鍔″垎閰嶇珯鐐癸紙瀹氭椂浠诲姟涓皟鐢級
+ * @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));
+
+ // 閫夋嫨绔欑偣
+ BasDevp basDevp;
+ int minInQty = devList.get(0).getInQty();
+
+ // 绛涢�夊嚭浠诲姟鏁版渶灏戠殑绔欑偣鍒楄〃
+ List<BasDevp> minTaskSites = devList.stream()
+ .filter(dev -> dev.getInQty() == minInQty)
+ .collect(Collectors.toList());
+
+ // 鏍规嵁閰嶇疆閫夋嫨鍒嗛厤绛栫暐
+ 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 index = counter.getAndIncrement() % minTaskSites.size();
+ basDevp = minTaskSites.get(index);
+ log.info("浣跨敤杞鍒嗛厤绛栫暐锛岀珯鐐圭粍锛歿}锛岃疆璇㈢储寮曪細{}锛岄�変腑绔欑偣锛歿}", groupKey, index, basDevp.getDevNo());
+ } else if (minTaskSites.size() > 1 && enableRoundRobin && "random".equals(strategy)) {
+ // 闅忔満鍒嗛厤
+ Random random = new Random();
+ int index = random.nextInt(minTaskSites.size());
+ basDevp = minTaskSites.get(index);
+ log.info("浣跨敤闅忔満鍒嗛厤绛栫暐锛岄�変腑绔欑偣锛歿}", basDevp.getDevNo());
+ } else {
+ // 榛樿锛氶�夋嫨绗竴涓紙浠诲姟鏈�灏戠殑锛�
+ basDevp = devList.get(0);
+ }
+
+ Integer endSite = basDevp.getDevNo();
+
+ // 鍏ュ簱鏆傚瓨+1
+ basDevpMapper.incrementInQty(endSite);
+
+ log.info("浠诲姟ID锛歿}宸插垎閰嶇珯鐐癸細{}", task.getId(), endSite);
+ return endSite;
+ }
+
+ /**
* 鏍规嵁绔欑偣缂栧彿鍒ゆ柇鏈哄櫒浜虹粍
* @param staNo 绔欑偣缂栧彿
* @return 鏈哄櫒浜虹粍鍚嶇О
--
Gitblit v1.9.1