From 88892fc38d50297c5ac5edf3fed629236bb9fd9d Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期二, 31 三月 2026 23:22:31 +0800
Subject: [PATCH] #入库同步串行请求
---
src/main/java/com/zy/core/plugin/store/StoreInTaskGenerationService.java | 192 ++++++++++++++++++++++++++++++++++--------------
1 files changed, 136 insertions(+), 56 deletions(-)
diff --git a/src/main/java/com/zy/core/plugin/store/StoreInTaskGenerationService.java b/src/main/java/com/zy/core/plugin/store/StoreInTaskGenerationService.java
index 4405611..9997796 100644
--- a/src/main/java/com/zy/core/plugin/store/StoreInTaskGenerationService.java
+++ b/src/main/java/com/zy/core/plugin/store/StoreInTaskGenerationService.java
@@ -27,9 +27,11 @@
import java.util.HashMap;
import java.util.List;
import java.util.Map;
+import java.util.concurrent.atomic.AtomicInteger;
@Service
public class StoreInTaskGenerationService {
+ private static final int APPLY_IN_TASK_TIMEOUT_SECONDS = 5;
@Autowired
private BasDevpService basDevpService;
@@ -44,81 +46,112 @@
@Autowired
private CommonService commonService;
+ private final AtomicInteger inFlightGenerateCount = new AtomicInteger(0);
+
public void generate(StoreInTaskPolicy policy) {
try {
if (!policy.isEnabled()) {
return;
}
- Object systemConfigMapObj = redisUtil.get(RedisKeyType.SYSTEM_CONFIG_MAP.key);
- if (systemConfigMapObj == null) {
+ HashMap<String, String> systemConfigMap = getSystemConfigMap();
+ if (systemConfigMap == null) {
return;
}
- HashMap<String, String> systemConfigMap = (HashMap<String, String>) systemConfigMapObj;
-
- int conveyorStationTaskLimit = 30;
- String conveyorStationTaskLimitStr = systemConfigMap.get("conveyorStationTaskLimit");
- if (conveyorStationTaskLimitStr != null) {
- conveyorStationTaskLimit = Integer.parseInt(conveyorStationTaskLimitStr);
- }
- int currentStationTaskCount = stationOperateProcessUtils.getCurrentStationTaskCount();
- if (currentStationTaskCount > conveyorStationTaskLimit) {
- News.error("杈撻�佺珯鐐逛换鍔″凡杈惧埌涓婇檺锛屼笂闄愬�硷細{}锛岀珯鐐逛换鍔℃暟锛歿}", conveyorStationTaskLimit, currentStationTaskCount);
+ if (!hasAvailableStationTaskCapacity(systemConfigMap)) {
return;
}
List<BasDevp> basDevps = basDevpService.list(new QueryWrapper<>());
for (BasDevp basDevp : basDevps) {
- StationThread stationThread = (StationThread) SlaveConnection.get(SlaveType.Devp, basDevp.getDevpNo());
- if (stationThread == null) {
- continue;
- }
-
- Map<Integer, StationProtocol> stationMap = stationThread.getStatusMap();
List<StationObjModel> barcodeStations = policy.getBarcodeStations(basDevp);
for (StationObjModel stationObjModel : barcodeStations) {
- Integer stationId = stationObjModel.getStationId();
- if (!stationMap.containsKey(stationId)) {
- continue;
- }
-
- StationProtocol stationProtocol = stationMap.get(stationId);
- if (stationProtocol == null) {
- continue;
- }
-
- StoreInTaskContext context = new StoreInTaskContext(basDevp, stationThread, stationObjModel,
- stationProtocol);
- if (!policy.matchCandidate(context)) {
- continue;
- }
- if (!policy.beforeApply(context)) {
- continue;
- }
-
- List<WrkMast> wrkMasts = wrkMastService.list(new QueryWrapper<WrkMast>()
- .eq("barcode", stationProtocol.getBarcode()));
- if (!wrkMasts.isEmpty()) {
- continue;
- }
-
- String generateLockKey = policy.getGenerateLockKey(context);
- Object lock = redisUtil.get(generateLockKey);
- if (lock != null) {
- continue;
- }
-
- policy.onRequestPermitGranted(context);
-
- InTaskApplyRequest request = policy.buildApplyRequest(context);
- News.info("鍙戣捣鍚屾WMS鍏ュ簱璇锋眰锛宐arcode={}锛宻tationId={}", request.getBarcode(),
- request.getSourceStaNo());
- InTaskApplyResult result = applySyncInTask(request);
- handleApplyResult(policy, context, request, result);
+ generate(policy, basDevp, stationObjModel, systemConfigMap);
}
}
} catch (Exception e) {
News.error("鐢熸垚鍏ュ簱浠诲姟寮傚父锛宲olicy={}", policy.getPolicyName(), e);
+ }
+ }
+
+ public void generate(StoreInTaskPolicy policy, BasDevp basDevp, StationObjModel stationObjModel) {
+ try {
+ if (!policy.isEnabled()) {
+ return;
+ }
+
+ HashMap<String, String> systemConfigMap = getSystemConfigMap();
+ if (systemConfigMap == null) {
+ return;
+ }
+ if (!hasAvailableStationTaskCapacity(systemConfigMap)) {
+ return;
+ }
+
+ generate(policy, basDevp, stationObjModel, systemConfigMap);
+ } catch (Exception e) {
+ Integer stationId = stationObjModel == null ? null : stationObjModel.getStationId();
+ News.error("鐢熸垚鍏ュ簱浠诲姟寮傚父锛宲olicy={}锛宻tationId={}", policy.getPolicyName(), stationId, e);
+ }
+ }
+
+ private void generate(StoreInTaskPolicy policy, BasDevp basDevp, StationObjModel stationObjModel,
+ HashMap<String, String> systemConfigMap) {
+ if (basDevp == null || stationObjModel == null || stationObjModel.getStationId() == null) {
+ return;
+ }
+
+ StationThread stationThread = (StationThread) SlaveConnection.get(SlaveType.Devp, basDevp.getDevpNo());
+ if (stationThread == null) {
+ return;
+ }
+
+ Integer stationId = stationObjModel.getStationId();
+ Map<Integer, StationProtocol> stationMap = stationThread.getStatusMap();
+ if (!stationMap.containsKey(stationId)) {
+ return;
+ }
+
+ StationProtocol stationProtocol = stationMap.get(stationId);
+ if (stationProtocol == null) {
+ return;
+ }
+
+ StoreInTaskContext context = new StoreInTaskContext(basDevp, stationThread, stationObjModel,
+ stationProtocol);
+ if (!policy.matchCandidate(context)) {
+ return;
+ }
+ if (!policy.beforeApply(context)) {
+ return;
+ }
+
+ List<WrkMast> wrkMasts = wrkMastService.list(new QueryWrapper<WrkMast>()
+ .eq("barcode", stationProtocol.getBarcode()));
+ if (!wrkMasts.isEmpty()) {
+ return;
+ }
+
+ String generateLockKey = policy.getGenerateLockKey(context);
+ Object lock = redisUtil.get(generateLockKey);
+ if (lock != null) {
+ return;
+ }
+
+ if (!tryReserveGenerateCapacity(systemConfigMap)) {
+ return;
+ }
+
+ try {
+ policy.onRequestPermitGranted(context);
+
+ InTaskApplyRequest request = policy.buildApplyRequest(context);
+ News.info("鍙戣捣鍚屾WMS鍏ュ簱璇锋眰锛宐arcode={}锛宻tationId={}锛宼imeout={}s",
+ request.getBarcode(), request.getSourceStaNo(), APPLY_IN_TASK_TIMEOUT_SECONDS);
+ InTaskApplyResult result = applySyncInTask(request);
+ handleApplyResult(policy, context, request, result);
+ } finally {
+ releaseGenerateCapacity();
}
}
@@ -199,4 +232,51 @@
return result;
}
+ private HashMap<String, String> getSystemConfigMap() {
+ Object systemConfigMapObj = redisUtil.get(RedisKeyType.SYSTEM_CONFIG_MAP.key);
+ if (systemConfigMapObj == null) {
+ return null;
+ }
+ return (HashMap<String, String>) systemConfigMapObj;
+ }
+
+ private boolean hasAvailableStationTaskCapacity(HashMap<String, String> systemConfigMap) {
+ int conveyorStationTaskLimit = getConveyorStationTaskLimit(systemConfigMap);
+ int currentStationTaskCount = stationOperateProcessUtils.getCurrentStationTaskCount();
+ if (currentStationTaskCount > conveyorStationTaskLimit) {
+ News.error("杈撻�佺珯鐐逛换鍔″凡杈惧埌涓婇檺锛屼笂闄愬�硷細{}锛岀珯鐐逛换鍔℃暟锛歿}", conveyorStationTaskLimit, currentStationTaskCount);
+ return false;
+ }
+ return true;
+ }
+
+ private boolean tryReserveGenerateCapacity(HashMap<String, String> systemConfigMap) {
+ int conveyorStationTaskLimit = getConveyorStationTaskLimit(systemConfigMap);
+ while (true) {
+ int reservedCount = inFlightGenerateCount.get();
+ int currentStationTaskCount = stationOperateProcessUtils.getCurrentStationTaskCount();
+ if (currentStationTaskCount + reservedCount >= conveyorStationTaskLimit) {
+ News.error("杈撻�佺珯鐐逛换鍔″凡杈惧埌涓婇檺锛屼笂闄愬�硷細{}锛岀珯鐐逛换鍔℃暟锛歿}锛岀敓鎴愪腑浠诲姟鏁帮細{}",
+ conveyorStationTaskLimit, currentStationTaskCount, reservedCount);
+ return false;
+ }
+ if (inFlightGenerateCount.compareAndSet(reservedCount, reservedCount + 1)) {
+ return true;
+ }
+ }
+ }
+
+ private void releaseGenerateCapacity() {
+ inFlightGenerateCount.updateAndGet(current -> current > 0 ? current - 1 : 0);
+ }
+
+ private int getConveyorStationTaskLimit(HashMap<String, String> systemConfigMap) {
+ int conveyorStationTaskLimit = 30;
+ String conveyorStationTaskLimitStr = systemConfigMap.get("conveyorStationTaskLimit");
+ if (conveyorStationTaskLimitStr != null) {
+ conveyorStationTaskLimit = Integer.parseInt(conveyorStationTaskLimitStr);
+ }
+ return conveyorStationTaskLimit;
+ }
+
}
--
Gitblit v1.9.1