修改设备状态数量接口

ruoyi-demo/src/main/java/com/ruoyi/demo/controller/EquipmentController.java

@@ -170,8 +170,8 @@ public class EquipmentController {
      */
     @GetMapping("/getEquipmentCount")
     public R getEquipmentCount(){
-        int count = equipmentService.getEquipmentCount();
-        return R.ok(count);
+        HashMap equipmentCount = equipmentService.getEquipmentCount();
+        return R.ok(equipmentCount);
     }
 
     /**

ruoyi-demo/src/main/java/com/ruoyi/demo/service/EquipmentService.java

@@ -33,7 +33,7 @@ public interface EquipmentService {
 
     void updateEquipmentPlan(PlanItemVo planItemVo);
 
-    int getEquipmentCount();
+    HashMap getEquipmentCount();
 
     HashMap getEquipmentList(EquipmentVo equipmentVo);
 

ruoyi-demo/src/main/java/com/ruoyi/demo/service/impl/EquipmentServiceImpl.java

@@ -293,8 +293,19 @@ public class EquipmentServiceImpl implements EquipmentService {
     }
 
     @Override
-    public int getEquipmentCount() {
-        return Math.toIntExact(equipmentMapper.selectCount(null));
+    public HashMap getEquipmentCount() {
+        QueryWrapper<Equipment> queryWrapper = new QueryWrapper<>();
+        queryWrapper.select("state","count(*) as onState");
+        queryWrapper.groupBy("state");
+        List<Equipment> equipment = equipmentMapper.selectList(queryWrapper);
+        HashMap<String, Integer> objectObjectHashMap = new HashMap<>();
+        int total = 0;
+        for (Equipment equipment1 : equipment) {
+            objectObjectHashMap.put(String.valueOf(equipment1.getState()),equipment1.getOnState());
+            total+=equipment1.getOnState();
+        }
+        objectObjectHashMap.put("total",total);
+        return objectObjectHashMap;
     }
 
     @Override