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/HeartAcOutSocket.java | 112 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 112 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/whyc/webSocket/HeartAcOutSocket.java b/src/main/java/com/whyc/webSocket/HeartAcOutSocket.java new file mode 100644 index 0000000..a4c0e6d --- /dev/null +++ b/src/main/java/com/whyc/webSocket/HeartAcOutSocket.java @@ -0,0 +1,112 @@ +package com.whyc.webSocket; + +import com.whyc.config.WebSocketConfig; +import com.whyc.dto.Real.PwrHisRealAcInDto; +import com.whyc.dto.Real.PwrHisRealDcoutInDto; +import com.whyc.dto.Response; +import com.whyc.pojo.db_param.PowerheartParam; +import com.whyc.service.PowerheartParamService; +import com.whyc.service.PwrdevAcdcdataService; +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 = "/acout",encoders = WebSocketEncoder.class,configurator = WebSocketConfig.class) +public class HeartAcOutSocket { + private Session session; + + private Thread thread; + + private static PwrdevAcdcdataService acdcdataService; + + 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 setPwrdevAcdcdataService(PwrdevAcdcdataService service) { + HeartAcOutSocket.acdcdataService = service; + } + + @Autowired + public void setPowerheartParamService(PowerheartParamService service) { + HeartAcOutSocket.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) { + Integer powerId = Integer.valueOf(message); + thread = new Thread("Thread_HeartacoutSocket") { + @Override + public void run() { + while (runFlag && !isInterrupted()) { + Thread thread = currentThread(); + threadFlagMap.put(thread.getId(), true); + try { + PwrHisRealDcoutInDto dcoutDto = acdcdataService.getPwrRealDcOut(powerId); + if (session.isOpen()) { + //鎺ㄩ�佷俊鎭� + synchronized (session) { + session.getBasicRemote().sendObject(new Response<>().setII(1,dcoutDto!=null,dcoutDto,"鐢垫簮鐩存祦杈撳嚭鏁版嵁")); + } + threadFlagMap.put(thread.getId(), false); + } + PowerheartParam powerheartParam=pwrHeartService.getHeartParamByPowerId(powerId); + sleep(1000*60*powerheartParam.getAcoutInterverCfg()); + } 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