From 01d6fa48a0de7a21988e89f71721b6b85e53b517 Mon Sep 17 00:00:00 2001 From: mitao <2763622819@qq.com> Date: 星期四, 06 三月 2025 16:33:53 +0800 Subject: [PATCH] 去掉资讯图片必填限制 --- ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/websocket/WebSocketUsers.java | 15 ++++++++++++++- 1 files changed, 14 insertions(+), 1 deletions(-) diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/websocket/WebSocketUsers.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/websocket/WebSocketUsers.java index 9b162c2..0b2f9cf 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/websocket/WebSocketUsers.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/websocket/WebSocketUsers.java @@ -20,11 +20,14 @@ private static Map<String, Session> USERS = new ConcurrentHashMap<>(); private static Map<String, Integer> USER_TYPES = new ConcurrentHashMap<>(); + // 保存用户id与session映射 + private static Map<String, Session> USER_ID_MAP = new ConcurrentHashMap<>(); - public static void put(String key, Session session, Integer clientType) + public static void put(String key, Session session, Integer clientType, Long userId) { USERS.put(key, session); USER_TYPES.put(key, clientType); + USER_ID_MAP.put(userId.toString(), session); } public static boolean remove(Session session) @@ -120,4 +123,14 @@ } } } + + /** + * 根据用户id进行点对点推送 + * @param userId 用户id + * @param message 消息内容 + */ + public static void sendMessageToUserById(Long userId, String message) { + Session session = USER_ID_MAP.get(userId.toString()); + sendMessageToUserByText(session, message); + } } -- Gitblit v1.7.1