From 4edcdffd7a197071d3bb876a3b17fc67744293a3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E8=89=AF=28004796=29?= <zhanglian.zl@fcbox.com>
Date: Mon, 15 Apr 2024 18:28:39 +0800
Subject: [PATCH] 123

---
 .../cai/ws/holder/WebSocketSessionHolder.java |  4 ++++
 .../com/ruoyi/controller/IndexController.java | 12 ++++++++++++
 .../com/ruoyi/controller/TestController.java  | 19 +++++++++++++++++++
 3 files changed, 35 insertions(+)
 create mode 100644 ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/IndexController.java
 create mode 100644 ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/TestController.java

diff --git a/ruoyi-cai/src/main/java/com/ruoyi/cai/ws/holder/WebSocketSessionHolder.java b/ruoyi-cai/src/main/java/com/ruoyi/cai/ws/holder/WebSocketSessionHolder.java
index c0b66c50..ff74dd2c 100644
--- a/ruoyi-cai/src/main/java/com/ruoyi/cai/ws/holder/WebSocketSessionHolder.java
+++ b/ruoyi-cai/src/main/java/com/ruoyi/cai/ws/holder/WebSocketSessionHolder.java
@@ -18,6 +18,10 @@ public class WebSocketSessionHolder {
 
     private static final Map<String, WebSocketSession> USER_SESSION_MAP = new ConcurrentHashMap<>();
 
+    public static int size(){
+        return USER_SESSION_MAP.size();
+    }
+
     public static void addSession(String sessionKey, WebSocketSession session) {
         USER_SESSION_MAP.put(sessionKey, session);
     }
diff --git a/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/IndexController.java b/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/IndexController.java
new file mode 100644
index 00000000..57e8c3bb
--- /dev/null
+++ b/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/IndexController.java
@@ -0,0 +1,12 @@
+package com.ruoyi.controller;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+public class IndexController {
+    @GetMapping("/ok")
+    public String ok(){
+        return "ok";
+    }
+}
diff --git a/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/TestController.java b/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/TestController.java
new file mode 100644
index 00000000..d45b103c
--- /dev/null
+++ b/ruoyi-websocket-boot/src/main/java/com/ruoyi/controller/TestController.java
@@ -0,0 +1,19 @@
+package com.ruoyi.controller;
+
+
+import com.ruoyi.cai.ws.holder.WebSocketSessionHolder;
+import com.ruoyi.cai.ws.util.WebSocketUtils;
+import com.ruoyi.common.core.domain.R;
+import lombok.RequiredArgsConstructor;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/cai/test")
+public class TestController {
+    public R getSession(){
+        int size = WebSocketSessionHolder.size();
+        return R.ok(size);
+    }
+}