修改基建工程

ruoyi-zdsz/src/main/java/com/ruoyi/zdsz/service/impl/ZEngineeringInfrastructureServiceImpl.java

@@ -198,10 +198,38 @@ public class ZEngineeringInfrastructureServiceImpl implements IZEngineeringInfra
         boolean flag = baseMapper.updateById(update) > 0;
         if (flag) {
             ZEngineeringInfrastructureVo vo = baseMapper.selectVoById(update.getId());
-            ZEngineeringNodeBo zEngineeringNodeBo = bo.getzEngineeringNodeBo();
-            zEngineeringNodeBo.setCivliId(vo.getId());
-            zEngineeringNodeBo.setCreateTime(vo.getCreateTime());
-            izEngineeringNodeService.update(zEngineeringNodeBo);
+
+//            if (flag) {
+//                ZEngineeringNodeBo zEngineeringNodeBo = bo.getzEngineeringNodeBo();
+//                zEngineeringNodeBo.setCivliId(vo.getId());
+//                zEngineeringNodeBo.setCreateTime(vo.getCreateTime());
+//                izEngineeringNodeService.update(zEngineeringNodeBo);
+//                return flag;
+//            }
+
+            List<ZEngineeringNodeBo> nodeBoList = bo.getzEngineeringNodeBoList();
+            if(nodeBoList != null && nodeBoList.size() != 0){
+                for (ZEngineeringNodeBo nodeBo:nodeBoList) {
+                    ZEngineeringNodeBo zEngineeringNodeBo = nodeBo;
+                    zEngineeringNodeBo.setCivliId(vo.getId());
+                    zEngineeringNodeBo.setCreateTime(vo.getCreateTime());
+                    if(zEngineeringNodeBo.getId() != null){
+                        izEngineeringNodeService.update(zEngineeringNodeBo);
+                    }else {
+                        izEngineeringNodeService.insert(zEngineeringNodeBo);
+                    }
+                }
+            }else {
+//                ZEngineeringInfrastructureVo vo = baseMapper.selectVoById(update.getId());
+                ZEngineeringNodeBo zEngineeringNodeBo = bo.getzEngineeringNodeBo();
+                zEngineeringNodeBo.setCivliId(vo.getId());
+                zEngineeringNodeBo.setCreateTime(vo.getCreateTime());
+                if(zEngineeringNodeBo.getId() != null){
+                    izEngineeringNodeService.update(zEngineeringNodeBo);
+                }else {
+                    izEngineeringNodeService.insert(zEngineeringNodeBo);
+                }
+            }
         }
         return flag;
     }