44.日常问题整理[2022/11/17]Quarkus之websockets的使用

最新推荐文章于 2023-01-03 11:31:51 发布

火火埋头干活

最新推荐文章于 2023-01-03 11:31:51 发布

阅读量288

点赞数

分类专栏： java 文章标签： websocket 网络协议网络

原文链接：https://quarkus.pro/guides/websockets.html

版权

java 专栏收录该内容

51 篇文章 0 订阅

订阅专栏

package org.acme.websockets;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import javax.enterprise.context.ApplicationScoped;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import javax.websocket.Session;

@ServerEndpoint("/chat/{username}")         
@ApplicationScoped
public class ChatSocket {

    Map<String, Session> sessions = new ConcurrentHashMap<>(); 

    //打开连接触发事件
    @OnOpen
    public void onOpen(Session session, @PathParam("username") String username) {
        sessions.put(username, session);
        broadcast("User " + username + " joined");
    }

    //关闭连接触发事件
    @OnClose
    public void onClose(Session session, @PathParam("username") String username) {
        sessions.remove(username);
        broadcast("User " + username + " left");
    }

    //传输错误事件触发事件
    @OnError
    public void onError(Session session, @PathParam("username") String username, Throwable throwable) {
        sessions.remove(username);
        broadcast("User " + username + " left on error: " + throwable);
    }

    //收到信息触发事件
    @OnMessage
    public void onMessage(String message, @PathParam("username") String username) {
        broadcast(">> " + username + ": " + message);
    }

   //群发消息
    private void broadcast(String message) {
        sessions.values().forEach(s -> {
            s.getAsyncRemote().sendObject(message, result ->  {
                if (result.getException() != null) {
                    System.out.println("Unable to send message: " + result.getException());
                }
            });
        });
    }
  //单独向某个连接发消息
  public void sendMessage(String username,String message) {
        Session session = sessions.get(username);
        session.getAsyncRemote().sendObject(message);
    }

}