WebSocket 是一种用于实现双向通信的网络协议,它基于 TCP 协议,可以实现服务器和客户端之间的实时消息传递。 在 Java 中,我们可以使用 Java WebSocket API 来实现 WebSocket。
以下是使用 Java WebSocket API 实现 WebSocket 消息推送的示例:
1、建立 WebSocket 服务端 要建立 WebSocket 服务端,需要实现 javax.websocket.Endpoint 类。在类中,重写 onOpen、onMessage、onClose 和 onError 方法,并在 onOpen 方法中注册客户端。下面是一个简单的示例:
代码语言:javascript复制@ServerEndpoint("/websocket")
public class WebSocketServer {
private static Set<Session> clients = Collections.synchronizedSet(new HashSet<Session>());
@OnOpen
public void onOpen(Session session) {
// 添加客户端
clients.add(session);
}
@OnMessage
public void onMessage(String message, Session session) {
// 处理消息
}
@OnClose
public void onClose(Session session) {
// 移除客户端
clients.remove(session);
}
@OnError
public void onError(Throwable error, Session session) {
// 错误处理
}
}
2、发布消息 要向客户端发布消息,可以使用 javax.websocket.Session.sendText() 方法。将消息发送给客户端时,需要遍历所有连接的客户端并将消息发送给每个客户端。下面是实现方法:
代码语言:javascript复制public static void sendMessage(String message) {
for (Session session : clients) {
try {
session.getBasicRemote().sendText(message);
} catch (IOException e) {
// 发送消息错误处理
}
}
}
3、客户端实现 WebSocket 连接 要在客户端创建 WebSocket 连接,需要使用 javax.websocket.ClientEndpoint 注解来声明一个 Endpoint 类,并重写 onOpen、onMessage、onClose 和 onError 方法。下面是一个简单的示例:
代码语言:javascript复制@ClientEndpoint
public class WebSocketClient {
@OnOpen
public void onOpen(Session session) {
// 打开通信通道时发生
}
@OnMessage
public void onMessage(String message, Session session) {
// 处理接收到的消息
}
@OnClose
public void onClose(Session session, CloseReason reason) {
// 关闭连接
}
@OnError
public void onError(Session session, Throwable error) {
// 错误处理
}
}
4、连接服务器 对于客户端连接到服务器,可以使用 javax.websocket.ContainerProvider 提供的静态方法 getWebSocketContainer() 获得 javax.websocket.WebSocketContainer 实例,来执行连接和开启通信通道的操作。下面是一个代码示例:
代码语言:javascript复制public static void connect(String endpointURI) {
try {
WebSocketContainer container = ContainerProvider.getWebSocketContainer();
container.connectToServer(WebSocketClient.class, URI.create(endpointURI));
} catch (DeploymentException | IOException e) {
// 错误处理
}
}
5、发送消息 发送消息可以在客户端类中定义一个 send() 方法,调用 javax.websocket.Session.sendText() 向服务端发送信息。下面是代码示例:
代码语言:javascript复制public void send(String message) {
try {
session.getBasicRemote().sendText(message);
} catch (IOException e) {
// 发送信息错误处理
}
}