New file |
| | |
| | | package com.whyc.webSocket; |
| | | |
| | | import com.whyc.config.WebSocketConfig; |
| | | import com.whyc.dto.RealTimeDto; |
| | | import com.whyc.dto.Response; |
| | | import com.whyc.pojo.db_user.User; |
| | | 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 = "/real",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) |
| | | public class RealTimeSocket { |
| | | private Session session; |
| | | |
| | | private Thread thread; |
| | | |
| | | //private static LockAlarmService almService; |
| | | |
| | | |
| | | private volatile boolean runFlag = true; |
| | | |
| | | private static HttpSession httpSession; |
| | | |
| | | private volatile Map<String, Thread> threadMap = new HashMap<>(); |
| | | |
| | | private volatile Map<Long,Boolean> threadFlagMap = new HashMap<>(); |
| | | |
| | | /*@Autowired |
| | | public void setLockAlarmService(LockAlarmService almService) { |
| | | LockAlmRtSocket.almService = almService; |
| | | }*/ |
| | | |
| | | |
| | | @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) { |
| | | RealTimeDto realDto= ActionUtil.getGson().fromJson(message,RealTimeDto.class); |
| | | if(httpSession!=null){ |
| | | User user = (User) httpSession.getAttribute("user"); |
| | | realDto.setUid(user.getId()); |
| | | }else{ |
| | | realDto.setUid(99); |
| | | } |
| | | thread = new Thread("Thread_realTimeSocket") { |
| | | @Override |
| | | public void run() { |
| | | while (runFlag && !isInterrupted()) { |
| | | Thread thread = currentThread(); |
| | | threadFlagMap.put(thread.getId(), true); |
| | | try { |
| | | Map<String, Object> map=new HashMap<>(); |
| | | //获取头部信息 |
| | | Response topRes=getTop(realDto); |
| | | map.put("topRes",topRes); |
| | | //根据pageType获取返回界面 |
| | | Response realRes=getRes(realDto); |
| | | map.put("realRes",realRes); |
| | | if (session.isOpen()) { |
| | | //推送信息 |
| | | synchronized (session) { |
| | | session.getBasicRemote().sendObject(new Response<>().set(1,map)); |
| | | } |
| | | threadFlagMap.put(thread.getId(), false); |
| | | } |
| | | sleep(4000); |
| | | } 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); |
| | | } |
| | | //获取头部信息 |
| | | private Response getTop(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | |
| | | //根据pageType获取返回界面 |
| | | private Response getRes(RealTimeDto realDto) { |
| | | Response res=new Response(); |
| | | switch (realDto.getPageType()){ |
| | | case "system":res=getSystemPage(realDto);break; |
| | | case "power":res=getPowerPage(realDto);break; |
| | | case "vol":res=getVolPage(realDto);break; |
| | | case "res":res=getResPage(realDto);break; |
| | | case "tmp":res=getTmpPage(realDto);break; |
| | | case "img":res=getImgPage(realDto);break; |
| | | case "3D":res=get3DPage(realDto);break; |
| | | case "self":res=getSelfPage(realDto);break; |
| | | case "manage":res=getManagePage(realDto);break; |
| | | default:res=new Response().set(1); |
| | | } |
| | | return res; |
| | | } |
| | | //获取管理界面信息 |
| | | private Response getManagePage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取自愈能力界面信息 |
| | | private Response getSelfPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取3D界面信息 |
| | | private Response get3DPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取图片界面信息 |
| | | private Response getImgPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取温度界面信息 |
| | | private Response getTmpPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取内阻界面信息 |
| | | private Response getResPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取电压界面信息 |
| | | private Response getVolPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取电源界面信息 |
| | | private Response getPowerPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | //获取系统概述界面信息 |
| | | private Response getSystemPage(RealTimeDto realDto) { |
| | | return new Response().set(1); |
| | | } |
| | | |
| | | @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()); |
| | | } |
| | | } |