From fb57e7836338304b16c09fec4fb5c68fb688c39e Mon Sep 17 00:00:00 2001
From: zwl <1051256694@qq.com>
Date: 星期二, 28 四月 2026 14:03:24 +0800
Subject: [PATCH] 1.新增分区查询 2.任务档新增分区字段

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

diff --git a/src/main/java/com/zy/asrs/service/impl/WrkMastLogServiceImpl.java b/src/main/java/com/zy/asrs/service/impl/WrkMastLogServiceImpl.java
index 7c0ef09..975c01c 100644
--- a/src/main/java/com/zy/asrs/service/impl/WrkMastLogServiceImpl.java
+++ b/src/main/java/com/zy/asrs/service/impl/WrkMastLogServiceImpl.java
@@ -2,11 +2,15 @@
 
 import com.baomidou.mybatisplus.plugins.Page;
 import com.baomidou.mybatisplus.service.impl.ServiceImpl;
+import com.core.common.Cools;
 import com.core.common.R;
 import com.zy.asrs.entity.InventoryFlowDto;
+import com.zy.asrs.entity.LocMast;
 import com.zy.asrs.entity.WrkMastLog;
 import com.zy.asrs.mapper.WrkMastLogMapper;
+import com.zy.asrs.service.LocMastService;
 import com.zy.asrs.service.WrkMastLogService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.Date;
@@ -14,6 +18,15 @@
 
 @Service("wrkMastLogService")
 public class WrkMastLogServiceImpl extends ServiceImpl<WrkMastLogMapper, WrkMastLog> implements WrkMastLogService {
+
+    @Autowired
+    private LocMastService locMastService;
+
+    @Override
+    public boolean insert(WrkMastLog entity) {
+        fillAreaId(entity);
+        return super.insert(entity);
+    }
 
     @Override
     public boolean save(Integer workNo) {
@@ -45,4 +58,33 @@
         return R.ok(page);
     }
 
+    private void fillAreaId(WrkMastLog wrkMastLog) {
+        if (wrkMastLog == null) {
+            return;
+        }
+        LocMast locMast = resolveAreaLocMast(wrkMastLog);
+        if (locMast != null) {
+            wrkMastLog.setAreaId(locMast.getAreaId());
+        }
+    }
+
+    private LocMast resolveAreaLocMast(WrkMastLog wrkMastLog) {
+        boolean outbound = wrkMastLog.getIoType() != null && wrkMastLog.getIoType() >= 100;
+        LocMast primary = outbound
+                ? selectLocMast(wrkMastLog.getSourceLocNo())
+                : selectLocMast(wrkMastLog.getLocNo());
+        if (primary != null) {
+            return primary;
+        }
+        return outbound
+                ? selectLocMast(wrkMastLog.getLocNo())
+                : selectLocMast(wrkMastLog.getSourceLocNo());
+    }
+
+    private LocMast selectLocMast(String locNo) {
+        if (Cools.isEmpty(locNo)) {
+            return null;
+        }
+        return locMastService.selectById(locNo);
+    }
 }

--
Gitblit v1.9.1