From 8b3345c25152d3f02813ddebdb380755c776af79 Mon Sep 17 00:00:00 2001
From: pengjie <17373303529@163.com>
Date: Wed, 7 Jun 2023 16:29:49 +0800
Subject: [PATCH] =?UTF-8?q?=E8=80=83=E6=A0=B8=E8=AF=84=E5=88=86=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/EngineeringSingleServiceImpl.java    | 35 +++++++++++++++----
 1 file changed, 28 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/EngineeringSingleServiceImpl.java b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/EngineeringSingleServiceImpl.java
index 9c5c0aa..c4ec403 100644
--- a/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/EngineeringSingleServiceImpl.java
+++ b/src/main/java/com/zhgd/xmgl/modules/basicdata/service/impl/EngineeringSingleServiceImpl.java
@@ -10,6 +10,8 @@ import com.zhgd.xmgl.modules.basicdata.mapper.EngineeringSingleMapper;
 import com.zhgd.xmgl.modules.basicdata.service.*;
 import com.zhgd.xmgl.modules.basicdata.vo.EngineeringVo;
 import com.zhgd.xmgl.modules.safety.entity.EnterpriseScore;
+import com.zhgd.xmgl.modules.safety.entity.EnterpriseScoreDetail;
+import com.zhgd.xmgl.modules.safety.service.IEnterpriseScoreDetailService;
 import com.zhgd.xmgl.modules.safety.service.IEnterpriseScoreService;
 import com.zhgd.xmgl.util.ParamEnum;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -43,6 +45,9 @@ public class EngineeringSingleServiceImpl extends ServiceImpl<EngineeringSingleM
     @Autowired
     private IEnterpriseScoreService enterpriseScoreService;
 
+    @Autowired
+    private IEnterpriseScoreDetailService enterpriseScoreDetailService;
+
     @Autowired
     private IProjectService projectService;
 
@@ -66,13 +71,18 @@ public class EngineeringSingleServiceImpl extends ServiceImpl<EngineeringSingleM
                 engineeringMainPerson.setEngineeringSn(engineeringVo.getEngineeringSn());
             }
             engineeringMainPersonService.saveBatch(engineeringMain.getEngineeringMainPersonList());
-            EnterpriseScore enterpriseScore = new EnterpriseScore();
-            enterpriseScore.setEnterpriseSn(engineeringMain.getEnterpriseSn());
-            enterpriseScore.setEngineeringSn(engineeringVo.getEngineeringSn());
-            enterpriseScore.setEnterpriseType(engineeringMain.getType());
-            enterpriseScore.setCreateBy(projectService.list().get(0).getGovernmentSn());
-            enterpriseScore.setCreateTime(new Date());
-            scoreList.add(enterpriseScore);
+            EnterpriseScore scoreServiceOne = enterpriseScoreService.getOne(Wrappers.<EnterpriseScore>lambdaQuery()
+                    .eq(EnterpriseScore::getEnterpriseSn, engineeringMain.getEnterpriseSn())
+                    .eq(EnterpriseScore::getEngineeringSn, engineeringVo.getEngineeringSn()));
+            if (scoreServiceOne == null) {
+                EnterpriseScore enterpriseScore = new EnterpriseScore();
+                enterpriseScore.setEnterpriseSn(engineeringMain.getEnterpriseSn());
+                enterpriseScore.setEngineeringSn(engineeringVo.getEngineeringSn());
+                enterpriseScore.setEnterpriseType(engineeringMain.getType());
+                enterpriseScore.setCreateBy(projectService.list().get(0).getGovernmentSn());
+                enterpriseScore.setCreateTime(new Date());
+                scoreList.add(enterpriseScore);
+            }
         }
         enterpriseScoreService.saveBatch(scoreList);
         return true;
@@ -87,9 +97,20 @@ public class EngineeringSingleServiceImpl extends ServiceImpl<EngineeringSingleM
         List<EngineeringMain> engineeringMains = engineeringMainService.list(Wrappers.<EngineeringMain>lambdaQuery()
                 .eq(EngineeringMain::getEngineeringSn, engineeringSn));
         List<Long> mainIds = engineeringMains.stream().map(m -> m.getId()).collect(Collectors.toList());
+        List<String> enterprises = engineeringMains.stream().map(m -> m.getEnterpriseSn()).collect(Collectors.toList());
         engineeringMainPersonService.remove(Wrappers.<EngineeringMainPerson>lambdaQuery()
                 .in(EngineeringMainPerson::getMainId, mainIds));
         engineeringMainService.removeByIds(mainIds);
+        List<EnterpriseScore> scoreServiceList = enterpriseScoreService.list(Wrappers.<EnterpriseScore>lambdaQuery()
+                .in(EnterpriseScore::getEnterpriseSn, enterprises)
+                .eq(EnterpriseScore::getEngineeringSn, engineeringSn));
+        for (EnterpriseScore enterpriseScore : scoreServiceList) {
+            int num = enterpriseScoreDetailService.count(Wrappers.<EnterpriseScoreDetail>lambdaQuery()
+                    .eq(EnterpriseScoreDetail::getScoreId, enterpriseScore.getId()));
+            if (num == 0) {
+                enterpriseScoreService.removeById(enterpriseScore.getId());
+            }
+        }
         return true;
     }