基于Java实现WebSocket协议过程详解

什么是WebSocket

WebSocket是HTML5规范中的协议，它允许在客户端和服务器之间建立一种双向通信的协议，即WebSocket连接。该连接是基于TCP的，它通过在HTTP/1.1之上进行协商升级，可以在客户端和服务器之间创建持久性的连接，实现低延迟、高效率的实时通信。

WebSocket连接的建立流程

WebSocket连接的建立过程可以分为以下几个步骤：

客户端向服务器发起HTTP请求，请求升级为WebSocket协议。
服务器返回HTTP 101状态码，表示已升级到WebSocket协议。
在HTTP头中包含WebSocket协议使用到的参数信息。如：
Upgrade: WebSocket
Connection: Upgrade
Sec-WebSocket-Accept：xx
客户端收到返回后，将连接升级为WebSocket连接，建立双向通讯。

基于Java实现WebSocket

在Java中，可以使用Java API中的javax.websocket来实现WebSocket协议，代码可以分为客户端和服务端两部分。客户端代码如下：

ClientManager client = ClientManager.createClient();
Session session = client.connectToServer(MyClientEndpoint.class, 
     new URI("ws://127.0.0.1:8080/myendpoint"));

服务端代码如下：

@ServerEndpoint("/myendpoint")
public class MyServerEndpoint {

   @OnMessage
   public void onMessage(String message, Session session) {
      System.out.println("Received message: " + message);
   }

   @OnOpen
   public void onOpen(Session session) {
      System.out.println("WebSocket opened: " + session.getId());
   }

   @OnClose
   public void onClose(Session session) {
      System.out.println("WebSocket closed: " + session.getId());
   }

   @OnError
   public void onError(Session session, Throwable t) {
      System.out.println("Error on WebSocket: " + session.getId());
      t.printStackTrace();
   }
}

示例说明

示例一：发送消息

客户端发送消息给服务端的示例代码如下：

session.getBasicRemote().sendText("Hello, server!");

服务端收到消息的代码如下：

@OnMessage
public void onMessage(String message, Session session) {
   System.out.println("Received message: " + message);
}

示例二：群发消息

服务端群发消息的代码如下：

for (Session sess : session.getOpenSessions()) {
   sess.getBasicRemote().sendText("Hello, everyone!");
}

客户端收到消息的代码如下：

session.addMessageHandler(new MessageHandler.Whole<String>() {
   @Override
   public void onMessage(String message) {
      System.out.println("Received message: " + message);
   }
});