From 5e5d0ddc71c957e9593b493e1984d0ca554b0cce Mon Sep 17 00:00:00 2001 From: whycxzp <glperry@163.com> Date: 星期一, 31 三月 2025 01:34:00 +0800 Subject: [PATCH] 电池站点温度实时 --- src/main/java/com/whyc/webSocket/BattStationTemAlarmSocket.java | 102 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 102 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/whyc/webSocket/BattStationTemAlarmSocket.java b/src/main/java/com/whyc/webSocket/BattStationTemAlarmSocket.java new file mode 100644 index 0000000..a7610fa --- /dev/null +++ b/src/main/java/com/whyc/webSocket/BattStationTemAlarmSocket.java @@ -0,0 +1,102 @@ +package com.whyc.webSocket; + +import com.whyc.config.WebSocketConfig; +import com.whyc.dto.PageAlarm; +import com.whyc.service.BattStationTempAlarmService; +import com.whyc.service.BattStationTempService; +import com.whyc.util.ActionUtil; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Component; + +import javax.websocket.*; +import javax.websocket.server.ServerEndpoint; +import java.util.HashMap; +import java.util.Map; + + +/** + * 鐢垫睜鍛婅 + */ +@Component +@ServerEndpoint(value = "/battStationTempAlarm",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) +public class BattStationTemAlarmSocket { + private Session session; + + private Thread thread; + + private static BattStationTempAlarmService service; + + private volatile boolean runFlag = true; + + private volatile Map<String, Thread> threadMap = new HashMap<>(); + + private volatile Map<Long,Boolean> threadFlagMap = new HashMap<>(); + + @Autowired + public void setService(BattStationTempAlarmService service) { + BattStationTemAlarmSocket.service = service; + } + + @OnOpen + public void onOpen(Session session, EndpointConfig config) { + this.session = session; + } + + @OnMessage + public void onMessage(Session session, String message) { + PageAlarm dto = ActionUtil.getGson("yyyy-MM-dd HH:mm:ss").fromJson(message, PageAlarm.class); + + thread = new Thread("Thread_BattStationTemAlarmSocket") { + @Override + public void run() { + while (runFlag && !isInterrupted()) { + Thread thread = currentThread(); + threadFlagMap.put(thread.getId(), true); + try { + if (session.isOpen()) { + //鎺ㄩ�佷俊鎭� + synchronized (session) { + session.getBasicRemote().sendObject(service.getPage(dto.getPageNum(), dto.getPageSize(), dto.getAlmLevel())); + } + 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); + } + + + @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()); + } +} \ No newline at end of file -- Gitblit v1.9.1