New file |
| | |
| | | package com.whyc.webSocket; |
| | | |
| | | import com.whyc.config.WebSocketConfig; |
| | | import com.whyc.dto.Response; |
| | | import com.whyc.factory.ThreadPoolExecutorFactory; |
| | | import com.whyc.pojo.Battinf; |
| | | import com.whyc.service.*; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.stereotype.Component; |
| | | |
| | | import javax.websocket.*; |
| | | import javax.websocket.server.ServerEndpoint; |
| | | import java.io.IOException; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.concurrent.CountDownLatch; |
| | | import java.util.concurrent.ThreadPoolExecutor; |
| | | import java.util.concurrent.TimeUnit; |
| | | import java.util.stream.Collectors; |
| | | |
| | | /** |
| | | * 管理员首页 |
| | | */ |
| | | @Component |
| | | @ServerEndpoint(value = "/homeAdmin", encoders = WebSocketEncoder.class, configurator = WebSocketConfig.class) |
| | | public class HomeAdminSocket { |
| | | private Session session; |
| | | |
| | | private Thread thread; |
| | | |
| | | private static BattInfService battInfService; |
| | | |
| | | private static BadBattMonService badBattMonService; |
| | | |
| | | private static Fbs9100StateService fbs9100StateService; |
| | | |
| | | private static PowerInfService powerInfService; |
| | | |
| | | private static BatttestdataInfService batttestdataInfService; |
| | | |
| | | @Autowired |
| | | public void setBattInfService(BattInfService battInfService) { |
| | | HomeAdminSocket.battInfService = battInfService; |
| | | } |
| | | |
| | | @Autowired |
| | | public void setBadBattMonService(BadBattMonService badBattMonService) { |
| | | HomeAdminSocket.badBattMonService = badBattMonService; |
| | | } |
| | | |
| | | @Autowired |
| | | public void setFbs9100StateService(Fbs9100StateService fbs9100StateService) { |
| | | HomeAdminSocket.fbs9100StateService = fbs9100StateService; |
| | | } |
| | | |
| | | @Autowired |
| | | public void setPowerInfService(PowerInfService powerInfService) { |
| | | HomeAdminSocket.powerInfService = powerInfService; |
| | | } |
| | | |
| | | @Autowired |
| | | public void setBatttestdataInfService(BatttestdataInfService batttestdataInfService) { |
| | | HomeAdminSocket.batttestdataInfService = batttestdataInfService; |
| | | } |
| | | |
| | | @OnOpen |
| | | public void onOpen(Session session, EndpointConfig config) { |
| | | this.session = session; |
| | | /*HttpSession httpSession = (HttpSession) config.getUserProperties().get("httpSession"); |
| | | UserInf user = (UserInf) httpSession.getAttribute("user"); |
| | | final int userId = user.getUId().intValue();*/ |
| | | final int userId = 1001; |
| | | Thread thread = new Thread() { |
| | | @Override |
| | | public void run() { |
| | | try { |
| | | Map<String, Object> res = new HashMap<>(); |
| | | while (!currentThread().isInterrupted()) { |
| | | ThreadPoolExecutor poolExecutor = ThreadPoolExecutorFactory.getPoolExecutor(); |
| | | CountDownLatch latch = new CountDownLatch(7); |
| | | poolExecutor.execute(() -> { |
| | | //蓄电池优劣分析 |
| | | Response resQualityAnalysis = badBattMonService.getQualityAnalysis(userId); |
| | | res.put("resQualityAnalysis", resQualityAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //基础资源信息 |
| | | Response resBattInfoAnalysis = battInfService.getBattInfoAnalysis(userId); |
| | | res.put("resBattInfoAnalysis", resBattInfoAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //站点信息 |
| | | Response resStateAnalysis = fbs9100StateService.getSateAnalysis(userId); |
| | | res.put("resStateAnalysis", resStateAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //不同品牌同一时间品质和同品牌同时间品质 |
| | | Response resProductQuaAnalysis = badBattMonService.getProductQuaAnalysis(userId); |
| | | res.put("resProductQuaAnalysis", resProductQuaAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //电池信息 |
| | | Response res_battGroupInfo = new Response(); |
| | | try { |
| | | List<Battinf> battinfList = battInfService.getBattGroupList(userId); |
| | | Map<String, List<Battinf>> battProducerMap = battinfList.stream().collect(Collectors.groupingBy(Battinf::getBattProducer)); |
| | | Map<Float, List<Battinf>> battMonVolStdMap = battinfList.stream().collect(Collectors.groupingBy(Battinf::getMonVolStd)); |
| | | Map<String, Object> battGroupProducerMap = new HashMap<>(); |
| | | Map<String, Object> battGroupMonVolMap = new HashMap<>(); |
| | | Map<String, Object> battGroupInfoMap = new HashMap<>(); |
| | | int battSize = battinfList.size(); |
| | | for (String producer : battProducerMap.keySet()) { |
| | | battGroupProducerMap.put(producer, battProducerMap.get(producer).size()); |
| | | } |
| | | for (Float monVolStd : battMonVolStdMap.keySet()) { |
| | | battGroupMonVolMap.put(Float.toString(monVolStd), battMonVolStdMap.get(monVolStd).size()); |
| | | } |
| | | battGroupInfoMap.put("producer", battGroupProducerMap); |
| | | battGroupInfoMap.put("monVol", battGroupMonVolMap); |
| | | battGroupInfoMap.put("totalNum", battSize); |
| | | res_battGroupInfo.setII(1, true, battGroupInfoMap, null); |
| | | } catch (Exception e) { |
| | | res_battGroupInfo.set(1, false, "发生异常:" + e.getCause()); |
| | | } |
| | | res.put("battGroupInfo", res_battGroupInfo); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //电源信息 |
| | | Response<Map> resPwrdevInfAnalysis = powerInfService.getPwrdevInfAnalysis(userId); |
| | | res.put("resPwrdevInfAnalysis", resPwrdevInfAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | poolExecutor.execute(() -> { |
| | | //测试信息 |
| | | Response<Map> resTestdataInfAnalysis = batttestdataInfService.getTestdataInfAnalysis(userId); |
| | | res.put("resTestdataInfAnalysis", resTestdataInfAnalysis); |
| | | latch.countDown(); |
| | | }); |
| | | latch.await(10, TimeUnit.MINUTES); |
| | | session.getBasicRemote().sendObject(new Response().set(1, res)); |
| | | sleep(4000); |
| | | } |
| | | } catch (Exception e) { |
| | | this.interrupt(); |
| | | } |
| | | } |
| | | }; |
| | | thread.start(); |
| | | this.thread = thread; |
| | | } |
| | | |
| | | @OnClose |
| | | public void onClose(CloseReason closeReason) throws IOException { |
| | | //System.err.println("closeReason = " + closeReason); |
| | | if (session.isOpen()) { |
| | | session.close(); |
| | | } |
| | | } |
| | | |
| | | @OnError |
| | | public void onError(Throwable error) throws IOException { |
| | | //error.printStackTrace(); |
| | | thread.isInterrupted(); |
| | | if (session.isOpen()) { |
| | | session.close(); |
| | | } |
| | | } |
| | | |
| | | } |