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

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

@@ -70,12 +70,18 @@ public class IndexController {
       SysUser sysUser = ShiroUtils.getSysUser();
 
       List<SysRole> sysRoles = iSysRoleService.selectRolesByUserId(sysUser.getUserId());
-      for (SysRole sysRole : sysRoles) {
-        //
-        BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
-        baseMonitorunitInfo.setRoleId(sysRole.getRoleId());
+      if(!ShiroUtils.getLoginName().equals("admin")){
+        for (SysRole sysRole : sysRoles) {
+          //
+          BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
+          baseMonitorunitInfo.setRoleId(sysRole.getRoleId());
+          List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
+                  baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
+          baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
+        }
+      }else{
         List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
-                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
+                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(new BaseMonitorunitInfo());
         baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
       }
 
@@ -139,51 +145,67 @@ public class IndexController {
       List<List<String>> data = new ArrayList<>();
 
       List<String> header =
-          new ArrayList<String>() {
-            {
-              this.add("名称");
-              this.add("监控数量");
-              this.add("告警数量");
-            }
-          };
+              new ArrayList<String>() {
+                {
+                  this.add("名称");
+                  this.add("监控数量");
+                  this.add("告警数量");
+                }
+              };
 
       List<BaseMonitorunitType> baseMonitorunitTypes =
-          baseMonitorunitTypeService.selectBaseMonitorunitTypeList(new BaseMonitorunitType());
+              baseMonitorunitTypeService.selectBaseMonitorunitTypeList(new BaseMonitorunitType());
 
       List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtoList = new ArrayList<BaseMonitorunitInfoDto>();
       SysUser sysUser = ShiroUtils.getSysUser();
 
       List<SysRole> sysRoles = iSysRoleService.selectRolesByUserId(sysUser.getUserId());
-      for (SysRole sysRole : sysRoles) {
-        //
-        BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
-        baseMonitorunitInfo.setRoleId(sysRole.getRoleId());
+
+      if(!ShiroUtils.getLoginName().equals("admin")){
+        for (SysRole sysRole : sysRoles) {
+          //
+          BaseMonitorunitInfo baseMonitorunitInfo = new BaseMonitorunitInfo();
+          baseMonitorunitInfo.setRoleId(sysRole.getRoleId());
+          List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
+                  baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
+          baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
+        }
+      }else {
         List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtos =
-                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(baseMonitorunitInfo);
+                baseMonitorunitInfoService.selectBaseMonitorunitInfoDtoList(new BaseMonitorunitInfo());
         baseMonitorunitInfoDtoList.addAll(baseMonitorunitInfoDtos);
       }
-      baseMonitorunitTypes.forEach(
-          x -> {
-            List<BaseMonitorunitInfoDto> collect =
-                    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))
-                    .collect(Collectors.toList());
-            data.add(
-                new ArrayList<String>() {
-                  {
-                    this.add(x.getMonitorunittypeName());
-                    this.add(collect.size() + "");
-                    this.add(collect2.size() + "");
-                  }
-                });
-          });
+
+      for (BaseMonitorunitType baseMonitorunitType : baseMonitorunitTypes) {
+        List<BaseMonitorunitInfoDto> baseMonitorunitInfoDtoList1 =
+                baseMonitorunitInfoDtoList.stream()
+                        .filter(
+                                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(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) {