init

ruoyi-websocket/src/main/java/com/ruoyi/websocket/dto/WsRMsgGen.java

@@ -49,4 +49,14 @@ public class WsRMsgGen {
         ok.setMsg("提示！");
         return ok;
     }
+
+    public static WsR hangup(String message, String roomId, Integer hangUpType) {
+        Map<String,Object> map = new HashMap<>();
+        map.put("roomid","roomId");
+        map.put("type","hangUpType");
+        WsR<Map<String, Object>> ok = WsR.ok(map);
+        ok.setMethod("hangup");
+        ok.setMsg(message);
+        return ok;
+    }
 }

ruoyi-websocket/src/main/java/com/ruoyi/websocket/util/WebSocketUtils.java

@@ -1,6 +1,7 @@
 package com.ruoyi.websocket.util;
 
 import com.ruoyi.websocket.holder.WebSocketSessionHolder;
+import jdk.nashorn.internal.runtime.regexp.JoniRegExp;
 import lombok.AccessLevel;
 import lombok.NoArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -31,6 +32,19 @@ public class WebSocketUtils {
         sendMessage(session, message);
     }
 
+    public static boolean close(String sessionKey) {
+        WebSocketSession sessions = WebSocketSessionHolder.getSessions(sessionKey);
+        if(sessions != null){
+            try {
+                sessions.close();
+                return true;
+            } catch (IOException e) {
+                log.error("关闭ws失败，sessionKey={}",sessionKey,e);
+            }
+        }
+        return false;
+    }
+
     public static void sendPongMessage(WebSocketSession session) {
         sendMessage(session, new PongMessage());
     }