WebSocket是HTML5引入的一种新型协议，用于在客户端和服务器之间建立长连接。它与HTTP一样可以在标准的Web端口上运行，也能穿越防火墙和代理服务器，并且已经被大量使用于实时通信和推送通知等场景。
Java作为一门强大的编程语言，也提供了各种WebSocket相关的API和库，本文将介绍如何使用Java开发Websocket应用程序，并提供具体的代码示例。
一、WebSocket API简介
Java EE 7提供了JSR-356 Java API for WebSocket，该API包含了WebSocket的客户端和服务器端相关的接口。你可以使用Java EE 7的容器（如GlassFish、Tomcat等）或第三方库（如Jetty、Tyrus等）来实现WebSocket功能。
立即学习“Java免费学习笔记（深入）”；
Java WebSocket API的核心接口和类如下：
- Session：表示WebSocket连接会话，通过它可以发送和接收消息。
- Endpoint：表示WebSocket的端点，实现了WebSocket的核心逻辑，包括连接、消息传输、错误处理等。
- MessageHandler：表示WebSocket的消息处理器，用于接收和处理各种类型的消息。
使用Java WebSocket API可以很方便地实现WebSocket应用程序，接下来我们将介绍具体示例。
二、WebSocket应用程序示例
本文将从以下几个方面来介绍WebSocket应用程序的实现：
- 服务器端的WebSocket端点实现
- 客户端的WebSocket实现
- 服务器端广播消息的实现
2.1 服务器端的WebSocket端点实现
在服务器端实现WebSocket应用程序，我们需要创建一个WebSocket端点。简单的端点代码如下：
import javax.websocket.OnMessage;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
@ServerEndpoint(value = "/chat")
public class ChatEndpoint {
@OnMessage
public void onMessage(Session session, String message) throws IOException {
// 处理收到的消息
session.getBasicRemote().sendText("You said: " + message);
}
}通过@ServerEndpoint注解指定端点的URI路径，这里是"/chat"。在端点中，实现了@OnMessage注解，用于接收和处理客户端发送的消息。在该方法中，我们可以对接收到的消息进行处理，并给客户端发送响应消息。
2.2 客户端的WebSocket实现
客户端通过Java实现WebSocket可以使用javax.websocket客户端API，下面给出一个简单的Java WebSocket客户端示例：
import javax.websocket.ClientEndpoint;
import javax.websocket.OnMessage;
import javax.websocket.Session;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.concurrent.CountDownLatch;
@ClientEndpoint
public class ChatClientEndpoint {
private CountDownLatch latch;
public ChatClientEndpoint(CountDownLatch latch) {
this.latch = latch;
}
@OnMessage
public void onMessage(Session session, String message) {
System.out.println("Received message: " + message);
latch.countDown();
}
public static void main(String[] args) throws Exception {
final int messageCount = 5;
final CountDownLatch latch = new CountDownLatch(messageCount);
URI uri = new URI("ws://localhost:8080/chat");
ChatClientEndpoint client = new ChatClientEndpoint(latch);
WebSocketContainer container = ContainerProvider.getWebSocketContainer();
container.connectToServer(client, uri);
for (int i = 0; i < messageCount; i++) {
String message = "Hello " + i;
client.sendMessage(message);
System.out.println("Sent message: " + message);
}
latch.await();
container.close();
}
public void sendMessage(String message) {
try {
session.getBasicRemote().sendText(message);
} catch (IOException e) {
e.printStackTrace();
}
}
}在客户端代码中，我们使用@ClientEndpoint注解标记客户端的WebSocket端点。在main方法中，我们使用WebSocketContainer连接到服务器端，并发送5条消息。当接收到服务器端响应后，会调用onMessage方法进行处理，同时借助CountDownLatch实现程序的同步等待。
2.3 服务器端广播消息的实现
WebSocket应用程序中，有时候需要服务器端向所有客户端广播消息。下面是一个简单的示例：
import javax.websocket.OnMessage;
import javax.websocket.Session;
import javax.websocket.server.ServerEndpoint;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
@ServerEndpoint(value = "/chat")
public class ChatEndpoint {
private static Set<Session> sessions = Collections.synchronizedSet(new HashSet<Session>());
@OnMessage
public void onMessage(Session session, String message) throws IOException {
// 处理收到的消息
broadcast("User " + session.getId() + " said: " + message);
}
private void broadcast(String message) throws IOException {
for (Session session : sessions) {
if (session.isOpen()) {
session.getBasicRemote().sendText(message);
}
}
}
@OnOpen
public void onOpen(Session session) {
sessions.add(session);
}
@OnClose
public void onClose(Session session) {
sessions.remove(session);
}
}在该代码中，我们维护了所有WebSocket会话的列表sessions，在@OnOpen方法中添加新的会话，@OnClose方法中移除会话。同时，在broadcast方法中向所有WebSocket会话发送广播消息。
三、总结
WebSocket是一种非常强大的新型协议，可以让程序建立实时通信和实时推送功能。Java对WebSocket也提供了丰富的API和库支持。本文介绍了如何使用Java开发WebSocket应用程序，并给出了具体的代码示例。希望本文能够对Java开发WebSocket应用程序有所帮助。
