审批页面修改

lxw

2023-08-15 160e150009b51a39fa95d9462c3798ba28d51a09

 src/main/java/com/whyc/webSocket/HomeAdminSocket.java

@@ -5,17 +5,15 @@
import com.whyc.factory.ThreadPoolExecutorFactory;
import com.whyc.pojo.Battinf;
import com.whyc.pojo.PowerInf;
import com.whyc.service.BadBattMonService;
import com.whyc.service.BattInfService;
import com.whyc.service.BatttestdataInfService;
import com.whyc.service.PowerInfService;
import com.whyc.pojo.StationInf;
import com.whyc.pojo.UserInf;
import com.whyc.service.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpSession;
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -25,7 +23,7 @@
import java.util.stream.Collectors;

/**
 * 总首页
 * 总首页（领导层）
 */
@Component
@ServerEndpoint(value = "/homeAdmin", encoders = WebSocketEncoder.class, configurator = WebSocketConfig.class)
@@ -34,7 +32,17 @@

    private Thread thread;

    private volatile boolean runFlag = true;

    private volatile Map<String, Thread> threadMap = new HashMap<>();

    private volatile Map<Long, Boolean> threadFlagMap = new HashMap<>();

    private static final int executeTime = 5000;

    private static BattInfService battInfService;

    private static StationInfService stationInfService;

    private static BadBattMonService badBattMonService;

@@ -42,6 +50,8 @@
    private static PowerInfService powerInfService;

    private static BatttestdataInfService batttestdataInfService;

    private HttpSession httpSession;

    @Autowired
    public void setBattInfService(BattInfService battInfService) {
@@ -63,13 +73,113 @@
        HomeAdminSocket.batttestdataInfService = batttestdataInfService;
    }

    @Autowired
    public void setStationInfService(StationInfService stationInfService) {
        HomeAdminSocket.stationInfService = stationInfService;
    }

    @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;
        this.httpSession = (HttpSession) config.getUserProperties().get("httpSession");
    }

    //实时页面推送
    public Response getAnalysis(int userId) {
        Map<String, Object> res = new HashMap<>();
        /*//蓄电池优劣分析
        Response resQualityAnalysis = badBattMonService.getQualityAnalysis(userId);
        res.put("resQualityAnalysis", resQualityAnalysis);*/

        //蓄电池组优劣分析
       /* Response resGroupAnalysis = badBattMonService.getGroupAnalysis(userId);
        res.put("resGroupAnalysis", resGroupAnalysis);*/
        Response resGroupAnalysis = batttestdataInfService.getGroupAnalysis(userId);
        res.put("resGroupAnalysis", resGroupAnalysis);


        //基础资源信息
        Response resBattInfoAnalysis = battInfService.getBattInfoAnalysis(userId);
        res.put("resBattInfoAnalysis", resBattInfoAnalysis);

        //站点信息
        Response resStationAnalysis = new Response();
        try {
            //List<Battinf> battinfList = battInfService.getSateAnalysis(userId);
            List<StationInf> sinfList = stationInfService.getSateAnalysis(userId);
            Map<Integer, List<StationInf>> nodeStationMap = sinfList.stream().collect(Collectors.groupingBy(StationInf::getNodeStation));
            Map<String, List<StationInf>> stationTypeMap = sinfList.stream().collect(Collectors.groupingBy(StationInf::getStationType));
            Map<Integer, Object> nodeMap = new HashMap<>();
            Map<String, Object> stationMap = new HashMap<>();
            Map<String, Object> battGroupInfoMap = new HashMap<>();
            int battSize = sinfList.size();
            for (Integer node : nodeStationMap.keySet()) {
                nodeMap.put(node, nodeStationMap.get(node).size());
            }
            for (String station : stationTypeMap.keySet()) {
                stationMap.put(station, stationTypeMap.get(station).size());
            }
            battGroupInfoMap.put("node", nodeMap);
            battGroupInfoMap.put("stationType", stationMap);
            battGroupInfoMap.put("totalNum", battSize);
            resStationAnalysis.setII(1, true, battGroupInfoMap, null);
        } catch (Exception e) {
            resStationAnalysis.set(1, false, "发生异常:" + e.getCause());
        }
        res.put("resStationAnalysis", resStationAnalysis);
        //不同品牌同一时间品质和同品牌同时间品质
        Response resProductQuaAnalysis = badBattMonService.getProductQuaAnalysis(userId);
        res.put("resProductQuaAnalysis", resProductQuaAnalysis);

        //电池信息
        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);
        //电源信息
        Response resPwrdevInfAnalysis = new Response();
        try {
            List<PowerInf> powerInfList = powerInfService.getPwrdevInfAnalysis(userId);
            Map<String, List<PowerInf>> powerProducerMap = powerInfList.stream().collect(Collectors.groupingBy(PowerInf::getPowerProducer));
            Map<String, Object> AnalysisMap = new HashMap<>();
            for (String producer : powerProducerMap.keySet()) {
                AnalysisMap.put(producer, powerProducerMap.get(producer).size());
            }
            resPwrdevInfAnalysis.setII(1, true, AnalysisMap, "电源信息");
        } catch (Exception e) {
            resPwrdevInfAnalysis.set(1, false, "发生异常:" + e.getCause());
        }
        res.put("resPwrdevInfAnalysis", resPwrdevInfAnalysis);
        //测试信息
        Response resTestdataInfAnalysis = batttestdataInfService.getTestdataInfAnalysis(userId);
        res.put("resTestdataInfAnalysis", resTestdataInfAnalysis);
        return new Response().set(1, res);
    }

    @OnMessage
    public void onMessage(Session session, String message) {
        UserInf user = (UserInf) this.httpSession.getAttribute("user");
        final int userId = user.getUId().intValue();
        //final int userId = 1087;
        Thread thread = new Thread() {
            @Override
            public void run() {
@@ -78,11 +188,12 @@
                    while (!currentThread().isInterrupted()) {
                        ThreadPoolExecutor poolExecutor = ThreadPoolExecutorFactory.getPoolExecutor();
                        CountDownLatch latch = new CountDownLatch(7);
                        Date date = new Date();
                        poolExecutor.execute(() -> {
                            //蓄电池优劣分析
                            Response resQualityAnalysis = badBattMonService.getQualityAnalysis(userId);
                            res.put("resQualityAnalysis", resQualityAnalysis);
                            //蓄电池组优劣分析(用蓄电池组组后评估的统计)
                            /*Response resGroupAnalysis = badBattMonService.getGroupAnalysis(userId);
                            res.put("resGroupAnalysis", resGroupAnalysis);*/
                            Response resGroupAnalysis = batttestdataInfService.getGroupAnalysis(userId);
                            res.put("resGroupAnalysis", resGroupAnalysis);
                            latch.countDown();
                        });
                        poolExecutor.execute(() -> {
@@ -95,13 +206,14 @@
                            //站点信息
                            Response resStationAnalysis = new Response();
                            try {
                                List<Battinf> battinfList = battInfService.getSateAnalysis(userId);
                                Map<Integer, List<Battinf>> nodeStationMap = battinfList.stream().collect(Collectors.groupingBy(Battinf::getNodeStation));
                                Map<String, List<Battinf>> stationTypeMap = battinfList.stream().collect(Collectors.groupingBy(Battinf::getStationType));
                                //List<Battinf> battinfList = battInfService.getSateAnalysis(userId);
                                List<StationInf> sinfList = stationInfService.getSateAnalysis(userId);
                                Map<Integer, List<StationInf>> nodeStationMap = sinfList.stream().collect(Collectors.groupingBy(StationInf::getNodeStation));
                                Map<String, List<StationInf>> stationTypeMap = sinfList.stream().collect(Collectors.groupingBy(StationInf::getStationType));
                                Map<Integer, Object> nodeMap = new HashMap<>();
                                Map<String, Object> stationMap = new HashMap<>();
                                Map<String, Object> battGroupInfoMap = new HashMap<>();
                                int battSize = battinfList.size();
                                int battSize = sinfList.size();
                                for (Integer node : nodeStationMap.keySet()) {
                                    nodeMap.put(node, nodeStationMap.get(node).size());
                                }
@@ -174,9 +286,9 @@
                            res.put("resTestdataInfAnalysis", resTestdataInfAnalysis);
                            latch.countDown();
                        });
                        latch.await(10, TimeUnit.SECONDS);
                        latch.await(10, TimeUnit.MINUTES);
                        session.getBasicRemote().sendObject(new Response().set(1, res));
                        sleep(4000);
                        sleep(5000);
                    }
                } catch (Exception e) {
                    this.interrupt();
@@ -188,20 +300,22 @@
    }

    @OnClose
    public void onClose(CloseReason closeReason) throws IOException {
        //System.err.println("closeReason = " + closeReason);
        if (session.isOpen()) {
            session.close();
    public void onClose(CloseReason closeReason) {
        System.err.println("closeReason = " + closeReason);
        runFlag = false;
        if (thread != null && thread.isAlive()) {
            thread.interrupt();
        }
        threadMap.remove(session.getId());
    }

    @OnError
    public void onError(Throwable error) throws IOException {
        //error.printStackTrace();
        thread.isInterrupted();
        if (session.isOpen()) {
            session.close();
    public void onError(Throwable error) {
        error.printStackTrace();
        if (thread != null && thread.isAlive()) {
            thread.interrupt();
        }
        threadMap.remove(session.getId());
    }

}