From 3b40ef10054199c386b0d58d03424182a33b5043 Mon Sep 17 00:00:00 2001
From: 16639036659 <577530412@qq.com>
Date: 星期四, 13 七月 2023 08:57:51 +0800
Subject: [PATCH] 添加风险单元名称

---
 src/main/java/com/ruoyi/doublePrevention/service/baseService/impl/PreventOverhaulLogServiceImpl.java |   26 +++++++++++++++++---------
 1 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/ruoyi/doublePrevention/service/baseService/impl/PreventOverhaulLogServiceImpl.java b/src/main/java/com/ruoyi/doublePrevention/service/baseService/impl/PreventOverhaulLogServiceImpl.java
index cdf3c5f..b0574dd 100644
--- a/src/main/java/com/ruoyi/doublePrevention/service/baseService/impl/PreventOverhaulLogServiceImpl.java
+++ b/src/main/java/com/ruoyi/doublePrevention/service/baseService/impl/PreventOverhaulLogServiceImpl.java
@@ -19,6 +19,8 @@
 import com.ruoyi.doublePrevention.service.baseService.PreventReportConfigService;
 import com.ruoyi.doublePrevention.vo.ResultVO;
 import com.ruoyi.project.system.user.domain.User;
+import com.ruoyi.project.tr.riskList.domain.RiskList;
+import com.ruoyi.project.tr.riskList.service.IRiskListService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.util.ObjectUtils;
@@ -36,6 +38,8 @@
     private PreventOverhaulLogRepository overhaulLogRepository;
     @Autowired
     private PreventReportConfigService preventReportConfigService;
+    @Autowired
+    private IRiskListService iRiskListService;
 
 
     @Override
@@ -56,7 +60,18 @@
         List<PreventOverhaulLog> preventOverhaulLogs = overhaulLogRepository.selectOverhaulLogPage(overhaulLogReqBO);
         List<PreventOverhaulLog> result = page.getResult();
 
-        resultVO.setData(result);
+        List<PreventOverhaulLogRespDTO> preventOverhaulLogRespDTOS = BeanCopyUtils.copyBeanList(result, PreventOverhaulLogRespDTO.class);
+
+        List<RiskList> riskUnitLists = iRiskListService.getRiskUnitLists();
+        for (PreventOverhaulLogRespDTO overhaulLog : preventOverhaulLogRespDTOS) {
+            for (RiskList riskUnit : riskUnitLists) {
+                if (overhaulLog.getRiskUnitId() == riskUnit.getRiskListId()){
+                    overhaulLog.setRiskUnitName(riskUnit.getRiskListName());
+                }
+            }
+        }
+
+        resultVO.setData(preventOverhaulLogRespDTOS);
         resultVO.setCount((int) page.getTotal());
         resultVO.setPageNum(page.getPageNum());
         resultVO.setPageSize(page.getPageSize());
@@ -71,9 +86,7 @@
         ResultVO resultVO = new ResultVO<>();
         resultVO.setCode(ErrorCodes.REQUEST_PARAM_ERROR.getCode());
 
-//        if (insertReqBO.getHazardCode().isEmpty()){
-//            throw new RuntimeException("风险分析对象编码为空");
-//        }
+
         if (ObjectUtils.isEmpty(insertReqBO.getRiskUnitId())){
             throw new RuntimeException("安全风险分析单元为空");
         }
@@ -98,7 +111,6 @@
         preventOverhaulLog.setGmtCreate(date);
         preventOverhaulLog.setLastEditUserName(sysUser.getUserName());
         preventOverhaulLog.setGmtModitify(date);
-        preventOverhaulLog.setRiskUnitId(null);// todo 暂时未设置
 
         //todo 对所属企业id和uuid没有进行设置
         PreventReportConfig reportConfigById = preventReportConfigService.getReportConfigById(SyncEnum.REPORT_CONFIG_RISK_ANA_UNIT.getCode());
@@ -149,10 +161,6 @@
         ResultVO resultVO = new ResultVO<>();
         resultVO.setCode(ErrorCodes.REQUEST_PARAM_ERROR.getCode());
 
-        if (updateReqBO.getHazardCode().isEmpty()){
-            resultVO.setMsg("风险分析对象编码");
-            return resultVO;
-        }
         if (ObjectUtils.isEmpty(updateReqBO.getRiskUnitId())){
             resultVO.setMsg("所属安全风险分析单元");
             return resultVO;

--
Gitblit v1.9.2