From 6eb5bf40297791d74fd815cc4ef39965056e1888 Mon Sep 17 00:00:00 2001
From: huangzj <826100833@qq.com>
Date: Fri, 16 Jun 2023 10:12:05 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../main/java/com/njcn/csdevice/constant/DataParam.java    | 1 +
 .../com/njcn/csdevice/enums/AlgorithmResponseEnum.java     | 3 ++-
 .../java/com/njcn/csdevice/mapper/AppProjectMapper.java    | 2 +-
 .../com/njcn/csdevice/mapper/mapping/AppProjectMapper.xml  | 3 +++
 .../njcn/csdevice/service/impl/AppProjectServiceImpl.java  | 7 ++++++-
 .../service/impl/AppTopologyDiagramServiceImpl.java        | 6 ++++++
 .../csdevice/service/impl/CsEngineeringServiceImpl.java    | 4 ++--
 7 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/constant/DataParam.java b/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/constant/DataParam.java
index 914db39..1c6d858 100644
--- a/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/constant/DataParam.java
+++ b/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/constant/DataParam.java
@@ -14,5 +14,6 @@ import java.util.stream.Stream;
 public interface DataParam {
     List<String> phases = Stream.of("A","B","C","T").collect(Collectors.toList());
     List<String> statMethods = Stream.of("avg","max","min","cp95").collect(Collectors.toList());
+    long FILE_SIZE = 10 * 1024 * 1024;
 
 }
diff --git a/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/enums/AlgorithmResponseEnum.java b/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/enums/AlgorithmResponseEnum.java
index 2257e8b..75e1080 100644
--- a/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/enums/AlgorithmResponseEnum.java
+++ b/cs-device/cs-device-api/src/main/java/com/njcn/csdevice/enums/AlgorithmResponseEnum.java
@@ -21,7 +21,8 @@ public enum AlgorithmResponseEnum {
     DATA_ERROR("A00503","存在相同的数据"),
     LINE_DATA_ERROR ("A00504","设备下监测点数据缺失"),
     ENGINEERING_DATA_ERROR ("A00505","工程数据数据缺失"),
-    ELEEPDPQD_DATA_ERROR ("A00505","统计指标据数据缺失"),
+    ELEEPDPQD_DATA_ERROR ("A00506","统计指标据数据缺失"),
+    FILE_SIZE_ERROR ("A00507","文件不能超过10M"),
 
 
     ;
diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/AppProjectMapper.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/AppProjectMapper.java
index b81d893..e64f4cd 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/AppProjectMapper.java
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/AppProjectMapper.java
@@ -60,5 +60,5 @@ public interface AppProjectMapper extends BaseMapper<AppProjectPO> {
 //                    " order by a.create_time desc",
 //                    "</script>"}
 //                    )
-    Page<AppProjectVO> getPageVo(Page<AppProjectVO> returnpage, @Param("temp")AppProjectQueryParm appProjectQueryParm);
+    Page<AppProjectVO> getPageVo(Page<AppProjectVO> returnpage, @Param("temp")AppProjectQueryParm appProjectQueryParm, @Param("userId") String userId);
 }
\ No newline at end of file
diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/mapping/AppProjectMapper.xml b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/mapping/AppProjectMapper.xml
index 81f986c..5ae8aca 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/mapping/AppProjectMapper.xml
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/mapper/mapping/AppProjectMapper.xml
@@ -25,6 +25,9 @@
   <select id="getPageVo" resultType="com.njcn.csdevice.pojo.vo.AppProjectVO">
     select a.*,b.pid as engineering_id,(select name from cs_ledger c where c.id=b.pid) as engineering_name from  cs_project a LEFT JOIN cs_ledger b on a.id = b.id where 1=1
     and a.status ="1"
+    <if test="userId != null and userId !=''">
+      and exists (select 1 from cs_engineering_user d where d.engineering_id=b.pid and d.user_id=#{userId})
+    </if>
     <if test="temp!=null and temp.projectId != null and temp.projectId !=''">
       AND a.id  = #{temp.projectId}
     </if>
diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppProjectServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppProjectServiceImpl.java
index 7eb4171..be19adb 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppProjectServiceImpl.java
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppProjectServiceImpl.java
@@ -18,6 +18,7 @@ import com.njcn.csdevice.pojo.vo.AppTopologyDiagramVO;
 import com.njcn.csdevice.service.AppProjectService;
 import com.njcn.csdevice.service.AppTopologyDiagramService;
 import com.njcn.oss.utils.FileStorageUtil;
+import com.njcn.web.utils.RequestUtil;
 import kotlin.jvm.Throws;
 import lombok.RequiredArgsConstructor;
 import net.bytebuddy.implementation.bytecode.Throw;
@@ -142,7 +143,11 @@ class AppProjectServiceImpl extends ServiceImpl<AppProjectMapper, AppProjectPO>
     public IPage<AppProjectVO> queryProject(AppProjectQueryParm appProjectQueryParm) {
 
         Page<AppProjectVO> returnpage = new Page<> (appProjectQueryParm.getPageNum ( ), appProjectQueryParm.getPageSize ( ));
-        returnpage = appProjectMapper.getPageVo (returnpage, appProjectQueryParm);
+        String userId = RequestUtil.getUserIndex();
+        if(Objects.equals(RequestUtil.getUsername(),"root")){
+            userId = null;
+        }
+        returnpage = appProjectMapper.getPageVo (returnpage, appProjectQueryParm, userId);
         List<AppProjectVO> collect = returnpage.getRecords().stream().map(temp -> {
             AppTopologyDiagramQueryParm appTopologyDiagramQueryParm = new AppTopologyDiagramQueryParm();
             appTopologyDiagramQueryParm.setProjectId(temp.getId());
diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppTopologyDiagramServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppTopologyDiagramServiceImpl.java
index 441eaf3..248a218 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppTopologyDiagramServiceImpl.java
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/AppTopologyDiagramServiceImpl.java
@@ -6,6 +6,9 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.google.common.base.Objects;
+import com.njcn.common.pojo.exception.BusinessException;
+import com.njcn.csdevice.constant.DataParam;
+import com.njcn.csdevice.enums.AlgorithmResponseEnum;
 import com.njcn.csdevice.mapper.AppProjectMapper;
 import com.njcn.csdevice.mapper.AppTopologyDiagramMapper;
 import com.njcn.csdevice.pojo.param.AppTopologyDiagramAddParm;
@@ -52,6 +55,9 @@ public class AppTopologyDiagramServiceImpl extends ServiceImpl<AppTopologyDiagra
         AppTopologyDiagramPO appTopologyDiagramPO = new AppTopologyDiagramPO();
 
         MultipartFile file = appTopologyDiagramAddParm.getFile ( );
+        if(file.getSize()> DataParam.FILE_SIZE){
+            throw new BusinessException(AlgorithmResponseEnum.FILE_SIZE_ERROR);
+        }
         String filePath = fileStorageUtil.uploadMultipart (file, OssPath.TOPOLOGY);
         appTopologyDiagramPO.setFilePath (filePath);
         appTopologyDiagramPO.setProjectId (appTopologyDiagramAddParm.getProjectId ()==null?"":appTopologyDiagramAddParm.getProjectId ());
diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
index b7a61e9..d065489 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
@@ -111,7 +111,7 @@ public class CsEngineeringServiceImpl extends ServiceImpl<CsEngineeringMapper, C
         QueryWrapper<CsEngineeringUserPO> csEngineeringUserPOQueryWrapper = new QueryWrapper<> ();
         //todo 后期修改
         String userIndex = RequestUtil.getUserIndex();
-        if(RequestUtil.getUsername()!="root"){
+        if(!Objects.equals(RequestUtil.getUsername(),"root")){
             csEngineeringUserPOQueryWrapper.lambda().eq(CsEngineeringUserPO::getUserId,userIndex);
             List<CsEngineeringUserPO> csEngineeringUserPOS = csEngineeringUserMapper.selectList(csEngineeringUserPOQueryWrapper);
             List<String> collect = csEngineeringUserPOS.stream().map(CsEngineeringUserPO::getEngineeringId).collect(Collectors.toList());
@@ -147,7 +147,7 @@ public class CsEngineeringServiceImpl extends ServiceImpl<CsEngineeringMapper, C
         QueryWrapper<CsEngineeringUserPO> csEngineeringUserPOQueryWrapper = new QueryWrapper<> ();
         //todo 后期修改
         String userIndex = RequestUtil.getUserIndex();
-        if(RequestUtil.getUsername()!="root"){
+        if(!Objects.equals(RequestUtil.getUsername(),"root")){
             csEngineeringUserPOQueryWrapper.lambda().eq(CsEngineeringUserPO::getUserId,userIndex);
             List<CsEngineeringUserPO> csEngineeringUserPOS = csEngineeringUserMapper.selectList(csEngineeringUserPOQueryWrapper);
             List<String> collect = csEngineeringUserPOS.stream().map(CsEngineeringUserPO::getEngineeringId).collect(Collectors.toList());