From d8475b8670b6b4cbcd1bc9e57d30a6f433d206ab Mon Sep 17 00:00:00 2001 From: DELL <1525436766@qq.com> Date: 星期一, 23 六月 2025 23:33:54 +0800 Subject: [PATCH] Merge branch 'master' of http://118.89.139.230:10101/r/powerIntelligenceSystem --- src/main/java/com/whyc/webSocket/HeartHrSocket.java | 120 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 120 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/whyc/webSocket/HeartHrSocket.java b/src/main/java/com/whyc/webSocket/HeartHrSocket.java new file mode 100644 index 0000000..2b662d5 --- /dev/null +++ b/src/main/java/com/whyc/webSocket/HeartHrSocket.java @@ -0,0 +1,120 @@ +package com.whyc.webSocket; + +import com.whyc.config.WebSocketConfig; +import com.whyc.dto.Real.PwrHisRealDcoutInDto; +import com.whyc.dto.Real.RealDateDTO; +import com.whyc.dto.Real.RealHeart; +import com.whyc.dto.Response; +import com.whyc.pojo.db_param.PowerheartParam; +import com.whyc.pojo.db_ram_db.BattRtdata; +import com.whyc.pojo.db_ram_db.BattRtstate; +import com.whyc.service.BattRtdataService; +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 = "/hr",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) +public class HeartHrSocket { + 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) { + HeartHrSocket.rtstateService = service; + } + + + @Autowired + public void setPowerheartParamService(PowerheartParamService service) { + HeartHrSocket.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) { + RealHeart heart= ActionUtil.getGson().fromJson(message,RealHeart.class); + thread = new Thread("Thread_HeartacoutSocket") { + @Override + public void run() { + while (runFlag && !isInterrupted()) { + Thread thread = currentThread(); + threadFlagMap.put(thread.getId(), true); + try { + RealDateDTO hrdto=rtstateService.getBattRealHr(heart.getBattgroupId()); + if (session.isOpen()) { + //鎺ㄩ�佷俊鎭� + synchronized (session) { + session.getBasicRemote().sendObject(new Response<>().setII(1,hrdto!=null,hrdto,"鐢垫簮鐩存祦杈撳嚭鏁版嵁")); + } + threadFlagMap.put(thread.getId(), false); + } + PowerheartParam powerheartParam=pwrHeartService.getHeartParamByPowerId(heart.getPowerId()); + sleep(1000*60*powerheartParam.getHrInterverCfg()); + } 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