From 392b42c4891cf2e6beda57ab32c51598f290f4b7 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期五, 14 三月 2025 20:56:27 +0800
Subject: [PATCH] bug修改

---
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WebSocketController.java |   24 ++++++++++++++++++++++++
 1 files changed, 24 insertions(+), 0 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WebSocketController.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WebSocketController.java
index 8a721a0..84aeed5 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WebSocketController.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/controller/WebSocketController.java
@@ -40,4 +40,28 @@
         WebSocketUsers.sendMessageToUsersByText(message);
         return R.ok();
     }
+
+    /**
+     * 批量向指定用户推送消息
+     * @param dto
+     * @return
+     */
+    @InnerAuth
+    @PostMapping("/push-batch")
+    public R<?> pushBatch(@RequestBody WebsocketMessageDTO dto) {
+        WebSocketUsers.sendMessageToUserById(dto.getUserId(), dto.getMessage());
+        return R.ok();
+    }
+    /**
+     * 通过用户id推送消息
+     * @param type
+     * @param msg
+     * @return
+     */
+    @GetMapping("/push-by-user-id/{userId}/{msg}")
+    public R<?> pushByUserId(@PathVariable("userId") Long userId,
+            @PathVariable("msg") String msg) {
+        WebSocketUsers.sendMessageToUserById(userId, msg);
+        return R.ok();
+    }
 }
\ No newline at end of file

--
Gitblit v1.7.1