From 1469fdca0a5f50344876cccdf743ca9c831d009f Mon Sep 17 00:00:00 2001
From: whyczh <hzjl@qq.com>
Date: 星期五, 25 二月 2022 09:52:55 +0800
Subject: [PATCH] websocket接口更新

---
 src/main/java/com/whyc/webSocket/DeviceWorkStateSocket.java |   50 +++++++++++++++++++++++++++++++-------------------
 1 files changed, 31 insertions(+), 19 deletions(-)

diff --git a/src/main/java/com/whyc/webSocket/DeviceWorkStateSocket.java b/src/main/java/com/whyc/webSocket/DeviceWorkStateSocket.java
index 3387b0d..f3bd16e 100644
--- a/src/main/java/com/whyc/webSocket/DeviceWorkStateSocket.java
+++ b/src/main/java/com/whyc/webSocket/DeviceWorkStateSocket.java
@@ -1,5 +1,6 @@
 package com.whyc.webSocket;
 
+import com.whyc.dto.Response;
 import com.whyc.pojo.Fbs9100State;
 import com.whyc.service.Fbs9100StateService;
 import com.whyc.util.ActionUtil;
@@ -25,53 +26,64 @@
 
     private static final int executeTime = 15000;
 
-    private static boolean exit=false;
-
     private static Fbs9100StateService service;
 
     private static Map<String,Thread> threadMap = new HashMap<>();
+
+    private volatile Map<Long,Boolean> threadFlagMap = new HashMap<>();
+
+    private Session session;
 
     @Autowired
     public void setService(Fbs9100StateService service) {
         DeviceWorkStateSocket.service = service;
     }
 
+    @OnOpen
+    public void onOpen(Session session){
+        this.session=session;
+    }
+
     @OnMessage
-    public void onMessage(Session session,String msg){
-        //鍋滄褰撳墠socket鐨勭嚎绋�
-        Thread threadBefore = threadMap.get(session.getId());
-        if(threadBefore !=null && threadBefore.isAlive()){
-            threadBefore.interrupt();
-            exit = true;
-        }
+    public synchronized void onMessage(Session session,String msg){
         Fbs9100State stateParam = ActionUtil.getGson().fromJson(msg, Fbs9100State.class);
         this.thread = new Thread("Thread_DeviceWorkState") {
             public void run() {
-                exit=false;
-                while (!thread.isInterrupted()&&!exit) {
+                while (true) {
+                    Thread thread = currentThread();
+                    threadFlagMap.put(thread.getId(), true);
                     try {
                         if (session.isOpen()) {
-                            session.getBasicRemote().sendObject(service.getList(stateParam.getDevId().intValue(),stateParam.getDevWorkstate(),Long.parseLong(stateParam.getNote())));
+                            //鎺ㄩ�佷俊鎭�
+                            synchronized (session) {
+                                session.getBasicRemote().sendObject(service.getList(stateParam.getDevId().intValue(),stateParam.getDevWorkstate(),Long.parseLong(stateParam.getNote())));
+                            }
+                            threadFlagMap.put(thread.getId(),false);
                         }
                         sleep(executeTime);
-                    } catch (IOException | InterruptedException | EncodeException e) {
+                        //} catch (IOException | InterruptedException | EncodeException e) {
+                    } catch (Exception e) {
                         interrupt();
                     }
                 }
             }
         };
         this.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);
     }
 
-    @OnOpen
-    public void onOpen(Session session){
-
-    }
-
     @OnClose
-    public void onClose(){
+    public void onClose(CloseReason closeReason){
+        System.err.println("closeReason = " + closeReason);
         if (thread != null && thread.isAlive()) {
             thread.interrupt();
         }

--
Gitblit v1.9.1