New file |
| | |
| | | package com.whyc.webSocket; |
| | | |
| | | import com.whyc.config.WebSocketConfig; |
| | | import com.whyc.dto.RealDataStatic.ResRealDataAc; |
| | | import com.whyc.dto.RealDataStatic.SticRealDataAc; |
| | | import com.whyc.dto.Response; |
| | | import com.whyc.pojo.db_param.PowerheartParam; |
| | | import com.whyc.service.BattRtstateService; |
| | | import com.whyc.service.PowerheartParamService; |
| | | import com.whyc.service.PwrdevAcdcdataService; |
| | | import com.whyc.util.ActionUtil; |
| | | 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.util.HashMap; |
| | | import java.util.Map; |
| | | |
| | | /** |
| | | * 实时数据统计推送核容设备,更改频率 |
| | | */ |
| | | @Component |
| | | @ServerEndpoint(value = "/realsticHr",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) |
| | | public class RealDataStaticHtSocket { |
| | | private Session session; |
| | | |
| | | private Thread thread; |
| | | |
| | | private static BattRtstateService rtstateService; |
| | | |
| | | private static PowerheartParamService pwrHeartService; |
| | | |
| | | private static HttpSession httpSession; |
| | | |
| | | private volatile boolean runFlag = true; |
| | | |
| | | private volatile Map<String, Thread> threadMap = new HashMap<>(); |
| | | |
| | | private volatile Map<Long,Boolean> threadFlagMap = new HashMap<>(); |
| | | |
| | | @Autowired |
| | | public void setBattRtstateService(BattRtstateService service) { |
| | | RealDataStaticHtSocket.rtstateService = service; |
| | | } |
| | | |
| | | |
| | | @Autowired |
| | | public void setPowerheartParamService(PowerheartParamService service) { |
| | | RealDataStaticHtSocket.pwrHeartService = service; |
| | | } |
| | | |
| | | @OnOpen |
| | | public void onOpen(Session session, EndpointConfig config) { |
| | | this.session = session; |
| | | this.httpSession = (HttpSession) config.getUserProperties().get("httpSession"); |
| | | } |
| | | @OnMessage |
| | | public void onMessage(Session session, String message) { |
| | | SticRealDataAc realData= ActionUtil.getGson().fromJson(message,SticRealDataAc.class); |
| | | thread = new Thread("Thread_HeartacinSocket") { |
| | | @Override |
| | | public void run() { |
| | | while (runFlag && !isInterrupted()) { |
| | | Thread thread = currentThread(); |
| | | threadFlagMap.put(thread.getId(), true); |
| | | try { |
| | | ResRealDataAc dataAc = rtstateService.getSticRealDataAc(realData); |
| | | if (session.isOpen()) { |
| | | //推送信息 |
| | | synchronized (session) { |
| | | session.getBasicRemote().sendObject(new Response<>().setII(1,dataAc!=null,dataAc,"实时数据统计-核容设备")); |
| | | } |
| | | threadFlagMap.put(thread.getId(), false); |
| | | } |
| | | PowerheartParam powerheartParam=pwrHeartService.getHeartParamByPowerId(realData.getPowerId()); |
| | | sleep(1000*60*powerheartParam.getAcinInterverCfg()); |
| | | } catch (Exception e) { |
| | | interrupt(); |
| | | } |
| | | } |
| | | } |
| | | }; |
| | | thread.start(); |
| | | threadFlagMap.put(thread.getId(),true); |
| | | //停止老的socket线程 |
| | | Thread threadBefore = threadMap.get(session.getId()); |
| | | if(threadBefore !=null && threadBefore.isAlive()){ |
| | | while (threadFlagMap.get(threadBefore.getId())){ |
| | | } |
| | | threadBefore.interrupt(); |
| | | } |
| | | //将线程存储,便于调用定位 |
| | | threadMap.put(session.getId(), this.thread); |
| | | } |
| | | @OnClose |
| | | 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) { |
| | | error.printStackTrace(); |
| | | if (thread != null && thread.isAlive()) { |
| | | thread.interrupt(); |
| | | } |
| | | threadMap.remove(session.getId()); |
| | | } |
| | | } |