使用开源的 Java Websocket 库来实现 Websocket 通信

曾精点

当使用 Java Websocket 库来实现 Websocket 通信时，需要实现两部分代码：服务端和客户端。

• 服务端代码实现：
  

服务端主要处理两个方面的问题：和客户端建立 Websocket 连接，并进行消息的接收和处理。服务端主要由两个组成部分：一个是服务器类，另一个是消息处理类。
下面是使用 Java Websocket 实现服务端逻辑的示例代码：
import java.net.InetSocketAddress;

import org.java_websocket.WebSocket;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

public class MyWebSocketServer extends WebSocketServer {
    public MyWebSocketServer(int port) {
        super(new InetSocketAddress(port));
    }

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        System.out.println("New connection from " + conn.getRemoteSocketAddress());
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        System.out.println("Closed connection from " + conn.getRemoteSocketAddress());
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
        System.out.println("Received message from " + conn.getRemoteSocketAddress() + ": " + message);
        // 处理接收到的消息
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        System.out.println("Error occurred on connection from " + conn.getRemoteSocketAddress() + ": " + ex.getMessage());
    }

    public static void main(String[] args) throws InterruptedException {
        int port = 8080; // 设置 WebSocket 服务器端口
        MyWebSocketServer server = new MyWebSocketServer(port);
        server.start(); // 启动 WebSocket 服务器
        System.out.println("WebSocket server started on port " + server.getPort());
    }
}在上面代码中，首先定义了一个 MyWebSocketServer 类继承 WebSocketServer。然后，在该类中重写了 onOpen()、onClose()、onMessage() 和 onError() 四个方法，分别代表 WebSocket 连接建立、断开、接收消息和错误处理的事件。
此外，在 main() 方法中，定义了一个 WebSocket 服务器 server 并通过 start() 方法启动。在 WebSocket 服务器启动之后，就可以监听客户端的连接和消息，进行相应的处理了。

• 客户端代码实现：
  

客户端有两个核心的功能：建立 Websocket 连接和发送消息。下面是使用 Java Websocket 实现客户端逻辑的示例代码：
import java.net.URI;
import java.net.URISyntaxException;

import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;

public class MyWebSocketClient extends WebSocketClient {
    public MyWebSocketClient(URI uri) {
        super(uri);
    }

    @Override
    public void onOpen(ServerHandshake handshakedata) {
        System.out.println("Connected to server: " + getURI());
    }

    @Override
    public void onMessage(String message) {
        System.out.println("Received message from server: " + message);
        // 处理接收到的消息
    }

    @Override
    public void onClose(int code, String reason, boolean remote) {
        System.out.println("Disconnected from server: " + reason);
    }

    @Override
    public void onError(Exception ex) {
        System.out.println("Error occurred: " + ex.getMessage());
    }

    public static void main(String[] args) throws InterruptedException, URISyntaxException {
        String serverURI = "ws://localhost:8080"; // 连接的服务器 URI
        MyWebSocketClient client = new MyWebSocketClient(new URI(serverURI));
        client.connect(); // 连接 WebSocket 服务器
        System.out.println("Connecting to server...");
        while (!client.getReadyState().equals(WebSocket.READYSTATE.OPEN)) {
            Thread.sleep(1000); // 等待连接成功
            System.out.println("Connecting...");
        }
        System.out.println("Connected.");
        // 发送消息
        String message = "Hello, this is a message.";
        client.send(message);
        System.out.println("Sent message to server: " + message);
        client.close(); // 关闭 WebSocket 连接
        System.out.println("Connection closed.");
    }
}在上面代码中，首先定义了一个 MyWebSocketClient 类继承 WebSocketClient。然后，在该类中重写了 onOpen()、onMessage()、onClose() 和 onError() 四个方法，分别代表 WebSocket 连接建立、接收消息、断开和错误处理的事件。
接着，在 main() 方法中，定义了一个 WebSocket 客户端 client 并通过 connect() 方法连接到服务器。在连接成功之前，使用 Thread.sleep() 方法来等待连接成功。getReadyState().equals(WebSocket.READYSTATE.OPEN) 用于检查当前连接状态是否已经为 OPEN，如果是则表示连接已经建立。
最后，在连接成功之后，使用 client.send() 方法向 WebSocket 服务器发送消息，并通过 client.close() 方法关闭连接。