From 108f99d25f9e0f1fddba012e4f1761f9d79f597e Mon Sep 17 00:00:00 2001
From: sxu <602087911@qq.com>
Date: Thu, 24 Jul 2025 17:50:26 +0800
Subject: [PATCH] =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E8=BF=9B=E5=BA=A6=E5=88=86?=
 =?UTF-8?q?=E6=9E=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../securitycontrol/screen/domain/ProjectNew.java    |  2 ++
 .../screen/service/impl/ProjectNewServiceImpl.java   | 12 +++++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/domain/ProjectNew.java b/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/domain/ProjectNew.java
index ff91a77..7e2f76c 100644
--- a/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/domain/ProjectNew.java
+++ b/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/domain/ProjectNew.java
@@ -80,6 +80,8 @@ public class ProjectNew extends BaseEntity {
     @ApiModelProperty(value = "风险等级")
     private String riskLevel;
 
+    private int riskSize;
+
     private String suggestion;
 
     private String taskName;
diff --git a/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/service/impl/ProjectNewServiceImpl.java b/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/service/impl/ProjectNewServiceImpl.java
index 26e7c0c..b020339 100644
--- a/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/service/impl/ProjectNewServiceImpl.java
+++ b/securitycontrol-model/securitycontrol-screen/src/main/java/com/securitycontrol/screen/service/impl/ProjectNewServiceImpl.java
@@ -10,6 +10,7 @@ import org.springframework.stereotype.Service;
 import com.securitycontrol.screen.mapper.ProjectNewMapper;
 import com.securitycontrol.screen.domain.ProjectNew;
 import com.securitycontrol.screen.service.IProjectNewService;
+import org.springframework.util.CollectionUtils;
 
 /**
  * 工程信息Service业务层处理
@@ -41,7 +42,16 @@ public class ProjectNewServiceImpl implements IProjectNewService {
      */
     @Override
     public List<ProjectNew> selectProjectNewList(ProjectNew projectNew) {
-        return projectNewMapper.selectProjectNewList(projectNew);
+        List<ProjectNew> projects = projectNewMapper.selectProjectNewList(projectNew);
+        for (ProjectNew project : projects) {
+            List<ProjectRisk> projectRisks = projectNewMapper.selectProjectRiskDetailList(project);
+            if (CollectionUtils.isEmpty(projectRisks)) {
+                project.setRiskSize(0);
+            } else {
+                project.setRiskSize(projectRisks.size());
+            }
+        }
+        return projects;
     }
 
     @Override