代码优化

yangys

2024-05-18 cc0bdfb33ef638dfafe3185c92c7076d815e1c9b

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package com.qianwen.core.websocket.config;
 
import cn.hutool.json.JSONUtil;
import java.io.IOException;
import java.util.Collection;
import java.util.Set;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.qianwen.core.tool.utils.Func;
import com.qianwen.core.websocket.holder.WebSocketSessionHolder;
import com.qianwen.core.websocket.message.JsonWebSocketMessage;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
 
public class WebSocketMessageSender {
    private static final Logger log = LoggerFactory.getLogger(WebSocketMessageSender.class);
 
    public static void broadcast(String message) {
        Collection<WebSocketSession> sessions = WebSocketSessionHolder.getSessions();
        for (WebSocketSession session : sessions) {
            send(session, message);
        }
    }
 
    public static boolean send(Object sessionKey, String message) {
        Set<WebSocketSession> session = WebSocketSessionHolder.getSession(sessionKey);
        if (Func.isEmpty(session)) {
            log.info("[send] 当前 sessionKey：{} 对应 session 不在本服务中", sessionKey);
            return false;
        }
        session.forEach(item -> {
            send(item, message);
        });
        return true;
    }
 
    public static void send(WebSocketSession session, JsonWebSocketMessage message) {
        send(session, JSONUtil.toJsonStr(message));
    }
 
    public static boolean send(WebSocketSession session, String message) {
        if (session == null) {
            log.error("[send] session 为 null");
            return false;
        } else if (!session.isOpen()) {
            log.error("[send] session 已经关闭");
            return false;
        } else {
            try {
                synchronized (session) {
                    session.sendMessage(new TextMessage(message));
                }
                return true;
            } catch (IOException e) {
                log.error("[send] session({}) 发送消息({}) 异常", new Object[]{session, message, e});
                return false;
            }
        }
    }
}