From 714bb1cce4817b7e4e696f35069bf31b992c7af6 Mon Sep 17 00:00:00 2001
From: zhou-hongcheng <1064771933@qq.com>
Date: Tue, 27 Jan 2026 10:24:01 +0800
Subject: [PATCH] =?UTF-8?q?=E7=9B=98=E7=82=B9=E7=AE=A1=E7=90=86=E7=BC=96?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../persistence/InvestigationManageImpl.java  | 38 ++++++++++++++++---
 1 file changed, 33 insertions(+), 5 deletions(-)

diff --git a/mhd_wms/src/main/java/com/mhd/wms/domain/investigationManage/repository/persistence/InvestigationManageImpl.java b/mhd_wms/src/main/java/com/mhd/wms/domain/investigationManage/repository/persistence/InvestigationManageImpl.java
index d2e5f66e5..f0a3565de 100644
--- a/mhd_wms/src/main/java/com/mhd/wms/domain/investigationManage/repository/persistence/InvestigationManageImpl.java
+++ b/mhd_wms/src/main/java/com/mhd/wms/domain/investigationManage/repository/persistence/InvestigationManageImpl.java
@@ -134,11 +134,39 @@ public class InvestigationManageImpl extends ServiceImpl<InvestigationManageMapp
     public Boolean update(InvestigationManageDO investigationManageDO) {
         InvestigationManage investigationManage = JSONUtil.toBean(JSONUtil.toJsonStr(investigationManageDO), InvestigationManage.class);
         List<InvestigetionManageDetail> investigetionManageDetailList = investigationManage.getInvestigetionManageDetailList();
-        Long[] ids = new Long[investigetionManageDetailList.size()];
-        for (int i = 0; i < investigetionManageDetailList.size(); i++) {
-            ids[i] = investigetionManageDetailList.get(i).getInvestigationManageDetailId();
-        }
-        investigetionManageDetailService.delete(ids);
+//        Long[] ids = new Long[investigetionManageDetailList.size()];
+//        for (int i = 0; i < investigetionManageDetailList.size(); i++) {
+//            ids[i] = investigetionManageDetailList.get(i).getInvestigationManageDetailId();
+//        }
+//        investigetionManageDetailService.delete(ids);
+        List<InvestigetionManageDetail> updateList = investigetionManageDetailList.stream().filter(detail ->
+                detail.getInvestigationManageDetailId() != null).collect(Collectors.toList());
+        investigetionManageDetailService.updateBatchById(updateList);
+
+        List<InvestigetionManageDetail> saveList = investigetionManageDetailList.stream()
+                .filter(detail -> detail.getInvestigationManageDetailId() == null)
+                .collect(Collectors.toList());
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        UserPo userPo = loginUser.getUserPo();
+        saveList.forEach(e -> {
+            e.setWarehouseCode(e.getWarehouseCode());
+            e.setWarehouseId(e.getWarehouseId());
+            e.setWarehouseName(e.getWarehouseName());
+            e.setOrganizationId(userPo.getOrganizationId());
+            e.setOrganizationName(userPo.getOrganizationName());
+            e.setTopOrganizationId(userPo.getTopOrganizationId());
+            e.setCreateByName(loginUser.getUsername());
+            e.setCreateBy(loginUser.getUserid());
+            e.setCreateTime(new Date());
+            e.setUpdateByName(loginUser.getUsername());
+            e.setUpdateBy(loginUser.getUserid());
+            e.setUpdateTime(new Date());
+            e.setInvestigationNumber(investigationManage.getInvestigationNumber());
+            e.setInvestigationFrequency(1);
+            InvestigetionManageDetailDO investigetionManageDetailDO = new InvestigetionManageDetailDO();
+            BeanUtils.copyProperties(e, investigetionManageDetailDO);
+            investigetionManageDetailService.insert(investigetionManageDetailDO);
+        });
         return this.updateById(investigationManage);
     }