Java websocket的简单介绍和使用-CSDN博客

本文链接：https://blog.csdn.net/qq_51767561/article/details/123934517

一 . 什么是websocket?

websocket就是运行在"web",也就是http上的socket通信规范,提供了与"tcp socket"类似的功能,使用它就可以像"tcp socket"一样调用下层协议栈,任意地收发数据.

二 . websocket session 发送文本消息两个方法方法的区别：

session.getAsyncRemote()表示异步发送

session.getBasicRemote()表示同步发送

三 . ConcurrentHashMap

ConcurrentHashMap是HashMap的一个线程安全的、支持高效并发的版本。默认理想状态下，ConcurrentHashMap可以支持16个线程执行并发写操作及任意数量线程的读操作

实现websocket连接的代码：

@Slf4j
@Component
@ServerEndpoint("/websocket/{sessionId}")
public class ProjectConsoleWs {
    public static ProjectConsoleWs service = new ProjectConsoleWs();
    /**
     * 存放所有在线的客户端 [sessionId, session]
     */
    private static final Map<String, Session> clients = new ConcurrentHashMap<>();

    public void sendMessage(String sessionId, String message) {
        if (!clients.containsKey(sessionId)) {
            log.error("未找到指定客户端{}的websocket连接!", sessionId);
            return;
        }

        this.onMessage(message, clients.get(sessionId));
        log.info("准备向客户端程序{}发送消息:{}", sessionId, message);
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        try {
            session.getBasicRemote().sendText(message);
        } catch (IOException e) {
            log.error("消息发送失败!", e);
        }
    }

    @OnOpen
    public void onOpen(@PathParam("sessionId") String sessionId, Session session) {
        if (clients.containsKey(sessionId)) {
            log.warn("客户端程序{}已有连接,无需建立连接", sessionId);
            return;
        }
        session.setMaxIdleTimeout(0);
        clients.put(sessionId, session);
        log.info("客户端程序{}建立连接成功!------>当前在线人数为：{}", sessionId,getOnlineCount());
    }

    @OnClose
    public void onClose(@PathParam("sessionId") String sessionId, Session session) {
        if (!clients.containsKey(sessionId)) {
            log.warn("客户端程序{}没有连接,无需断开连接", sessionId);
            return;
        }

        clients.remove(sessionId);
        log.info("客户端程序{}断开连接成功!------>当前在线人数为：{}", sessionId,getOnlineCount());
    }

    @OnError
    public void onError(Session session, Throwable throwable) {
        log.info("连接{}发生错误!", session.getId());
        throwable.printStackTrace();
    }

    public boolean sessionIsOpen(String sessionId){
        return clients.get(sessionId).isOpen();
    }

    public synchronized int getOnlineCount(){
        return clients.size();
    }

}

@Configuration
public class WebSocketConfig {

    @Bean("serverEndpointExporter")
    public ServerEndpointExporter serverEndpointExporter(){
        return new ServerEndpointExporter();
    }

}