Java WebSocket的使用(有代码案例)
import javax.websocket.*;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.CopyOnWriteArraySet;
// 使用 @ServerEndpoint 注解定义一个 WebSocket 服务端点,
// 客户端可以通过 ws://{host}:{port}/websocket/{path} 连接到这个端点。
@ServerEndpoint("/websocket/chat")
public class ChatEndpoint {
// 使用一个 concurrent 的集合来存储所有的当前连接会话。
private static final CopyOnWriteArraySet<Session> sessions = new CopyOnWriteArraySet<>();
// 当新的客户端连接时,会触发 @OnOpen 注解的方法。
@OnOpen
public void onOpen(Session session) {
sessions.add(session);
System.out.println("新连接加入:" + session.getId());
}
// 当客户端发送消息时,会触发 @OnMessage 注解的方法。
@OnMessage
public void onMessage(String message, Session session) {
System.out.println("收到消息:" + message);
// 广播消息给所有的客户端。
for (Session peer : sessions) {
if (!peer.equals(session)) {
try {
peer.getBasicRemote().sendText(message);
} catch (IOException e) {
System.out.println("发送消息出错:" + e.getMessage());
}
}
}
}
// 当连接关闭时,会触发 @OnClose 注解的方法。
@OnClose
public void onClose(Session session) {
sessions.remove(session);
System.out.println("连接关闭:" + session.getId());
}
// 当出现错误时,会触发 @OnError 注解的方法。
@OnError
public void onError(Session session, Throwable throwable) {
System.out.println("发生错误:" + throwable.getMessage());
}
}
这段代码定义了一个简单的 WebSocket 服务端,当有新的客户端连接时,它会将会话添加到一个集合中;当接收到客户端的消息时,它会将这个消息广播给所有其他的客户端。同时,它还处理了连接关闭和错误情况。这个例子展示了如何使用 Java WebSocket API 来创建一个简单的聊天服务器。
评论已关闭