change - 首页监控单元数量统计修改

ruoyi-admin/src/main/java/com/ruoyi/web/controller/iot/IndexController.java

@@ -70,6 +70,7 @@ public class IndexController {
       SysUser sysUser = ShiroUtils.getSysUser();
 
       List<SysRole> sysRoles = iSysRoleService.selectRolesByUserId(sysUser.getUserId());
+      if(!ShiroUtils.getLoginName().equals("admin")){
         for (SysRole sysRole : sysRoles) {
           //
           BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
@@ -78,6 +79,11 @@ public class IndexController {
                   baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
           baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
         }
+      }else{
+        List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
+                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(new BaseMonitorunitInfo());
+        baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
+      }
 
       baseMonitorunitTypes.forEach(
           x -> {
@@ -154,6 +160,8 @@ public class IndexController {
       SysUser sysUser = ShiroUtils.getSysUser();
 
       List<SysRole> sysRoles = iSysRoleService.selectRolesByUserId(sysUser.getUserId());
+
+      if(!ShiroUtils.getLoginName().equals("admin")){
         for (SysRole sysRole : sysRoles) {
           //
           BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
@@ -162,28 +170,42 @@ public class IndexController {
                   baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
           baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
         }
-      baseMonitorunitTypes.forEach(
-          x -> {
-            List<BaseMonitorunitInfoDto> collect =
+      }else {
+        List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
+                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(new BaseMonitorunitInfo());
+        baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
+      }
+
+      for (BaseMonitorunitType baseMonitorunitType : baseMonitorunitTypes) {
+        List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtoList1 =
                 baseMonitorunitInfoDtoList.stream()
                         .filter(
-                        y ->
-                            !y.getParentId().isEmpty()
-                                && y.getMonitorunitType().equals(x.getMonitorunittypeId()))
-                    .collect(Collectors.toList());
-            List<BaseMonitorunitInfoDto> collect2 =
-                collect.stream()
-                    .filter(z -> z.getMonitorunitStatus().equals(0))
+                                y ->!y.getParentId().isEmpty()&& y.getMonitorunitType().equals(baseMonitorunitType.getMonitorunittypeId()))
                         .collect(Collectors.toList());
+        if(baseMonitorunitInfoDtoList1.size() > 0){
+        /*List<BaseMonitorunitInfoDto> collect2 =
+                baseMonitorunitInfoDtoList1.stream().filter(z -> z.getMonitorunitStatus().equals(0L)).collect(Collectors.toList());*/
+
+          //List<Long> longList = baseMonitorunitInfoDtoList1.stream().filter(z -> z.getMonitorunitStatus().equals(0L)).map(BaseMonitorunitInfoDto::getMonitorunitStatus).collect(Collectors.toList());
           data.add(
                   new ArrayList<String>() {
                     {
-                    this.add(x.getMonitorunittypeName());
-                    this.add(collect.size() + "");
-                    this.add(collect2.size() + "");
+                      this.add(baseMonitorunitType.getMonitorunittypeName());
+                      this.add(baseMonitorunitInfoDtoList1.size() + "");
+                      this.add(0 + "");
                     }
                   });
+        }else{
+          data.add(
+                  new ArrayList<String>() {
+                    {
+                      this.add(baseMonitorunitType.getMonitorunittypeName());
+                      this.add(0 + "");
+                      this.add(0 + "");
+                    }
                   });
+        }
+      }
 
       return JSONArray.toJSONString(new ScrollTableDto(header, data));
     } catch (Exception ex) {