pms国网上送代码提交

pqs-device/pms-device/pms-device-api/src/main/java/com/njcn/device/pms/pojo/po/Monitor.java

@@ -231,10 +231,7 @@ public class Monitor extends BaseEntity {
      * 0.未上送 1.已上送 2.取消上送 3.待重新上送
      */
     private Integer isUploadHead;
-    /**
+
-     * 行业分类
-     */
-    private String tradeCode;
 
     /**
      * 牵引站id

pqs-system/system-boot/src/main/java/com/njcn/system/controller/DictTreeController.java

@@ -121,7 +121,24 @@ public class DictTreeController extends BaseController {
         return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, result, methodDescribe);
     }
 
+    @OperateInfo(info = LogEnum.SYSTEM_COMMON)
+    @GetMapping("/queryAll")
+    @ApiOperation("查询所有树形字典")
+    public HttpResult<List<SysDicTreePO>> queryAll() {
+        String methodDescribe = getMethodDescribe("queryAll");
+        List<SysDicTreePO> result = sysDicTreePOService.queryAll();
+        return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, result, methodDescribe);
+    }
 
 
+    @OperateInfo(info = LogEnum.SYSTEM_COMMON)
+    @GetMapping("/queryTree")
+    @ApiOperation("树形字典树结构展示")
+    public HttpResult<List<SysDicTreePO>> queryTree() {
+        String methodDescribe = getMethodDescribe("queryTree");
+        List<SysDicTreePO> result = sysDicTreePOService.queryTree();
+        return HttpResultUtil.assembleCommonResponseResult(CommonResponseEnum.SUCCESS, result, methodDescribe);
+    }
+
 }
 

pqs-system/system-boot/src/main/java/com/njcn/system/service/SysDicTreePOService.java

@@ -32,4 +32,13 @@ public interface SysDicTreePOService extends IService<SysDicTreePO>{
          * @return
          */
         SysDicTreePO queryById(String id);
+
+        /**
+         * 查询所有树形字典
+         * @author cdf
+         * @date 2023/12/18
+         */
+        List<SysDicTreePO> queryAll();
+
+        List<SysDicTreePO> queryTree();
 }

pqs-system/system-boot/src/main/java/com/njcn/system/service/impl/SysDicTreePOServiceImpl.java

@@ -2,6 +2,7 @@ package com.njcn.system.service.impl;
 
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.njcn.common.pojo.enums.common.DataStateEnum;
 import com.njcn.system.pojo.param.DictTreeParam;
 import com.njcn.system.pojo.vo.DictTreeVO;
 import org.springframework.beans.BeanUtils;
@@ -45,7 +46,7 @@ public class SysDicTreePOServiceImpl extends ServiceImpl<SysDicTreePOMapper, Sys
         }else{
             sysDicTreePO.setPids("0");
         }
-        sysDicTreePO.setStatus("0");
+        sysDicTreePO.setStatus(0);
         boolean save = this.save(sysDicTreePO);
 
         return save;
@@ -100,4 +101,25 @@ public class SysDicTreePOServiceImpl extends ServiceImpl<SysDicTreePOMapper, Sys
         return this.lambdaQuery().eq(SysDicTreePO::getId,id).eq(SysDicTreePO::getStatus,0).one();
     }
 
+    @Override
+    public List<SysDicTreePO> queryAll() {
+        LambdaQueryWrapper<SysDicTreePO> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(SysDicTreePO::getStatus, 0);
+        return this.list(lambdaQueryWrapper);
+    }
+
+    @Override
+    public List<SysDicTreePO> queryTree() {
+        LambdaQueryWrapper<SysDicTreePO> lambdaQueryWrapper = new LambdaQueryWrapper<>();
+        lambdaQueryWrapper.eq(SysDicTreePO::getStatus, 0);
+       List<SysDicTreePO> sysDicTreePOList = this.list(lambdaQueryWrapper);
+       List<SysDicTreePO> parent = sysDicTreePOList.stream().filter(item->item.getPid().equals("0") && !item.getCode().equals("Device_Unit")).peek(item->item.setChildren(getChildren(item,sysDicTreePOList))).collect(Collectors.toList());
+       return parent.stream().peek(item->item.setChildren(getChildren(item,sysDicTreePOList))).collect(Collectors.toList());
+    }
+
+
+    private List<SysDicTreePO> getChildren(SysDicTreePO sysDicTreePO,List<SysDicTreePO> all){
+       return all.stream().filter(item->item.getPid().equals(sysDicTreePO.getId())).peek(item->item.setChildren(getChildren(item,all))).collect(Collectors.toList());
+    }
+
 }