From 22ab0bb87c89f349465ed36e8802c16a915ef5ee Mon Sep 17 00:00:00 2001
From: lxw <810412026@qq.com>
Date: 星期二, 07 三月 2023 15:17:43 +0800
Subject: [PATCH] 监控层,运维层首页+跳转页

---
 src/main/java/com/whyc/webSocket/HomeMonitorSocket.java |   65 ++++++++++++++++++++++++++++++--
 1 files changed, 60 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/whyc/webSocket/HomeMonitorSocket.java b/src/main/java/com/whyc/webSocket/HomeMonitorSocket.java
index 06c09df..6ab2b34 100644
--- a/src/main/java/com/whyc/webSocket/HomeMonitorSocket.java
+++ b/src/main/java/com/whyc/webSocket/HomeMonitorSocket.java
@@ -1,14 +1,22 @@
 package com.whyc.webSocket;
 
 import com.whyc.config.WebSocketConfig;
-import com.whyc.service.StationInfService;
+import com.whyc.dto.Response;
+import com.whyc.pojo.BattdischargePlan;
+import com.whyc.service.BattalarmDataService;
+import com.whyc.service.BattdischargePlanService;
+import com.whyc.service.DevalarmDataService;
+import com.whyc.service.PwrdevAlarmService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
 import javax.websocket.*;
 import javax.websocket.server.ServerEndpoint;
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * 鐩戞帶灞傞椤�
@@ -26,13 +34,35 @@
 
     private volatile Map<Long, Boolean> threadFlagMap = new HashMap<>();
 
-    private static StationInfService service;
+    private static BattdischargePlanService planService;
+
+    private static BattalarmDataService bAlmService;
+
+    private static DevalarmDataService dAlmService;
+
+    private static PwrdevAlarmService pAlmService;
+
 
     private Session session;
 
     @Autowired
-    public void setStationInfService(StationInfService service) {
-        HomeMonitorSocket.service = service;
+    public void setBattdischargePlanService(BattdischargePlanService planService) {
+        HomeMonitorSocket.planService = planService;
+    }
+
+    @Autowired
+    public void setBattalarmDataService(BattalarmDataService bAlmService) {
+        HomeMonitorSocket.bAlmService = bAlmService;
+    }
+
+    @Autowired
+    public void setDevalarmDataService(DevalarmDataService dAlmService) {
+        HomeMonitorSocket.dAlmService = dAlmService;
+    }
+
+    @Autowired
+    public void setPwrdevAlarmService(PwrdevAlarmService pAlmService) {
+        HomeMonitorSocket.pAlmService = pAlmService;
     }
 
     @OnOpen
@@ -52,7 +82,32 @@
                         if (session.isOpen()) {
                             //鎺ㄩ�佷俊鎭�
                             synchronized (session) {
-                                session.getBasicRemote().sendObject(service.getMonitorAll(userId));
+                                Map<String, Object> map = new HashMap<>();
+                                Response planRes = new Response<>();
+                                List<BattdischargePlan> planList = new ArrayList<>();
+                                //浠婃棩鏀剧數浠诲姟缁熻
+                                try {
+                                    Map<String, Integer> planMap = new HashMap<>();
+                                    planList = planService.getPlanStaticToday(userId);
+                                    Map<Integer, List<BattdischargePlan>> disMap = planList.stream().collect(Collectors.groupingBy(BattdischargePlan::getDischargeState));
+                                    for (Integer dischargeState : disMap.keySet()) {
+                                        planMap.put(String.valueOf(dischargeState), disMap.get(dischargeState).size());
+                                    }
+                                    planMap.put("sumNum", planList.size());
+                                    planRes.setII(1, planList.size() > 0, map, "浠婃棩鏀剧數浠诲姟缁熻");
+                                } catch (Exception e) {
+                                    planRes.set(1, false, "鍑虹幇寮傚父" + e.getMessage());
+                                }
+                                map.put("planRes", planRes);
+                                map.put("planList", planList);
+                                //浠婃棩瀹炴椂鍛婅
+                                Response bAlmRes = bAlmService.getBalmToday(userId);
+                                map.put("bAlmRes", bAlmRes);
+                                Response bDebRes = dAlmService.getDalmToday(userId);
+                                map.put("bDebRes", bDebRes);
+                                Response bPwrRes = pAlmService.getPalmToday(userId);
+                                map.put("bPwrRes", bPwrRes);
+                                session.getBasicRemote().sendObject(map);
                             }
                             threadFlagMap.put(thread.getId(), false);
                         }

--
Gitblit v1.9.1