From 2638679df3c71814ee64b38cc2ff32279d482748 Mon Sep 17 00:00:00 2001
From: hzj <826100833@qq.com>
Date: Mon, 13 May 2024 15:43:20 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../user/impl/UserReportPOServiceImpl.java    | 26 ++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/user/impl/UserReportPOServiceImpl.java b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/user/impl/UserReportPOServiceImpl.java
index de92e6b66..e9534fe89 100644
--- a/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/user/impl/UserReportPOServiceImpl.java
+++ b/pqs-supervision/supervision-boot/src/main/java/com/njcn/supervision/service/user/impl/UserReportPOServiceImpl.java
@@ -198,7 +198,31 @@ public class UserReportPOServiceImpl extends ServiceImpl<UserReportPOMapper, Use
             1、每个附件需要返回文件名称以及可以预览的url
             todo...by黄正剑
          */
-        userReportVO.setUserReportProjectPO(new UserReportProjectPO());
+        if (
+                CollectionUtil.newArrayList(
+                        UserNatureEnum.BUILD_POWER_GRID.getCode(),
+                        UserNatureEnum.EXTEND_POWER_GRID.getCode()
+                ).contains(userReportPO.getUserType())) {
+            //电网工程类用户额外数据
+            UserReportProjectPO byId = userReportProjectPOService.getById(id);
+            userReportVO.setUserReportProjectPO(byId);
+
+        } else if (
+                CollectionUtil.newArrayList(
+                        UserNatureEnum.BUILD_NON_LINEAR_LOAD.getCode(),
+                        UserNatureEnum.EXTEND_NON_LINEAR_LOAD.getCode(),
+                        UserNatureEnum.BUILD_NEW_ENERGY_POWER_STATION.getCode(),
+                        UserNatureEnum.EXTEND_NEW_ENERGY_POWER_STATION.getCode()
+                ).contains(userReportPO.getUserType())) {
+            //非线性负荷用户 & 新能源发电站用户
+            UserReportSubstationPO byId= userReportSubstationPOService.getById(id);
+            userReportSubstationPOService.saveOrUpdate(byId);
+            userReportVO.setUserReportSubstationPO(byId);
+        } else if (UserNatureEnum.SENSITIVE_USER.getCode().equals(userReportPO.getUserType())) {
+            // 敏感及重要用户
+            UserReportSensitivePO byId = userReportSensitivePOService.getById(id);
+            userReportVO.setUserReportSensitivePO(byId);
+        }
         return userReportVO;
     }