From 3971255dadf007563d83d887e8c2ef465242fc87 Mon Sep 17 00:00:00 2001
From: zwl <1051256694@qq.com>
Date: 星期二, 28 四月 2026 22:03:31 +0800
Subject: [PATCH] 双伸出库改成订单出库

---
 src/main/java/com/zy/asrs/service/impl/WrkMastServiceImpl.java |   86 +++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 86 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/zy/asrs/service/impl/WrkMastServiceImpl.java b/src/main/java/com/zy/asrs/service/impl/WrkMastServiceImpl.java
index 4a867e7..114a7cc 100644
--- a/src/main/java/com/zy/asrs/service/impl/WrkMastServiceImpl.java
+++ b/src/main/java/com/zy/asrs/service/impl/WrkMastServiceImpl.java
@@ -3,16 +3,19 @@
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.baomidou.mybatisplus.service.impl.ServiceImpl;
 import com.core.common.Cools;
+import com.zy.asrs.entity.LocMast;
 import com.zy.asrs.entity.WrkMast;
 import com.zy.asrs.entity.result.FindLocNoAttributeVo;
 import com.zy.asrs.mapper.WrkMastLogMapper;
 import com.zy.asrs.mapper.WrkMastMapper;
 import com.zy.asrs.service.InboundCameraCaptureService;
+import com.zy.asrs.service.LocMastService;
 import com.zy.asrs.service.WrkMastService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.Date;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
@@ -23,6 +26,21 @@
 
     @Autowired
     private WrkMastLogMapper wrkMastLogMapper;
+
+    @Autowired
+    private LocMastService locMastService;
+
+    @Override
+    public boolean insert(WrkMast entity) {
+        fillAreaId(entity);
+        return super.insert(entity);
+    }
+
+    @Override
+    public boolean updateById(WrkMast entity) {
+        fillAreaId(entity);
+        return super.updateById(entity);
+    }
 
     @Override
     public int getWorkingMast(Integer devpNo) {
@@ -98,6 +116,33 @@
         return h;
     }
 
+    @Override
+    public Date firstInboundCreateTimeByTrainNo(String trainNo) {
+        if (Cools.isEmpty(trainNo)) {
+            return null;
+        }
+        Date fromMast = baseMapper.selectFirstInboundCreateTimeByTrainNo(trainNo);
+        Date fromLog = wrkMastLogMapper.selectFirstInboundCreateTimeByTrainNo(trainNo);
+        if (fromMast == null) {
+            return fromLog;
+        }
+        if (fromLog == null) {
+            return fromMast;
+        }
+        return fromMast.before(fromLog) ? fromMast : fromLog;
+    }
+
+    @Override
+    public int finishedInboundPalletCountByTrainNo(String trainNo) {
+        if (Cools.isEmpty(trainNo)) {
+            return 0;
+        }
+        Set<Integer> wrkNos = new HashSet<>();
+        addPositiveWrkNos(baseMapper.listFinishedInboundWrkNosByTrainNo(trainNo), wrkNos);
+        addPositiveWrkNos(wrkMastLogMapper.listFinishedInboundWrkNosByTrainNo(trainNo), wrkNos);
+        return wrkNos.size();
+    }
+
     private static void addPositivePlt(List<Integer> list, Set<Integer> target) {
         if (list == null) {
             return;
@@ -108,4 +153,45 @@
             }
         }
     }
+
+    private static void addPositiveWrkNos(List<Integer> list, Set<Integer> target) {
+        if (list == null) {
+            return;
+        }
+        for (Integer wrkNo : list) {
+            if (wrkNo != null && wrkNo > 0) {
+                target.add(wrkNo);
+            }
+        }
+    }
+
+    private void fillAreaId(WrkMast wrkMast) {
+        if (wrkMast == null) {
+            return;
+        }
+        LocMast locMast = resolveAreaLocMast(wrkMast);
+        if (locMast != null) {
+            wrkMast.setAreaId(locMast.getAreaId());
+        }
+    }
+
+    private LocMast resolveAreaLocMast(WrkMast wrkMast) {
+        boolean outbound = wrkMast.getIoType() != null && wrkMast.getIoType() >= 100;
+        LocMast primary = outbound
+                ? selectLocMast(wrkMast.getSourceLocNo())
+                : selectLocMast(wrkMast.getLocNo());
+        if (primary != null) {
+            return primary;
+        }
+        return outbound
+                ? selectLocMast(wrkMast.getLocNo())
+                : selectLocMast(wrkMast.getSourceLocNo());
+    }
+
+    private LocMast selectLocMast(String locNo) {
+        if (Cools.isEmpty(locNo)) {
+            return null;
+        }
+        return locMastService.selectById(locNo);
+    }
 }

--
Gitblit v1.9.1