Java-WebSocket

1、Java-WebSocket

• 官网地址：https://github.com/TooTallNate/Java-WebSocket
• API文档：https://www.javadoc.io/doc/org.java-websocket/Java-WebSocket/latest/index.html

2、WebSocketClient + WebSocketServer基本用法

• 以下是WebSocket写的客户端和服务端，实现两者互相收发消息
• 服务端：WebSocketServer

package com.arthur;

import org.java_websocket.WebSocket;
import org.java_websocket.drafts.Draft;
import org.java_websocket.drafts.Draft_6455;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.InetSocketAddress;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import java.util.Collections;

public class MyWebSocketServer extends WebSocketServer {

    public MyWebSocketServer(int port) throws UnknownHostException {
        super(new InetSocketAddress(port));
        System.out.println("MyWebSocketServer(int port) port = " + port);
    }

    public MyWebSocketServer(InetSocketAddress address) {
        super(address);
        System.out.println("MyWebSocketServer(InetSocketAddress address)");
    }

    public MyWebSocketServer(int port, Draft_6455 draft) {
        super(new InetSocketAddress(port), Collections.<Draft>singletonList(draft));
        System.out.println("MyWebSocketServer(int port, Draft_6455 draft) port = " + port);
    }

    /**
     * WebSocketServer启动成功的回调，不代表客户端和服务端连接成功
     */
    @Override
    public void onStart() {
        System.out.println("onStart");
        setConnectionLostTimeout(100);
    }

    @Override
    public void onOpen(WebSocket conn, ClientHandshake handshake) {
        System.out.println("onOpen : " + conn.getRemoteSocketAddress().getAddress().getHostAddress() + " entered the room!");
    }

    @Override
    public void onMessage(WebSocket conn, String message) {
        System.out.println("onMessage : " + conn.getRemoteSocketAddress().getAddress().getHostAddress() + " String = " + message);
    }

    @Override
    public void onMessage(WebSocket conn, ByteBuffer message) {
        System.out.println("onMessage : " + conn.getRemoteSocketAddress().getAddress().getHostAddress() + " ByteBuffer = " + message.toString());
    }

    @Override
    public void onClose(WebSocket conn, int code, String reason, boolean remote) {
        System.out.println("onClose : " + conn.getRemoteSocketAddress().getAddress().getHostAddress()
                + " code = " + code
                + " reason = " + reason
                + " remote = " + remote);
    }

    @Override
    public void onError(WebSocket conn, Exception ex) {
        System.out.println("onError : " + conn.getRemoteSocketAddress().getAddress().getHostAddress()
                + " Exception = " + ex.toString());
    }

    public static void main(String[] args) throws InterruptedException, IOException {

        int port = 8887;
        MyWebSocketServer webSocketServer = new MyWebSocketServer(port);
        webSocketServer.start();
        System.out.println("MyWebSocketServer start");

        BufferedReader sysin = new BufferedReader(new InputStreamReader(System.in));
        while (true) {
            String in = sysin.readLine();
            // 发送两次数据
            webSocketServer.broadcast(in);
            webSocketServer.broadcast(in.getBytes());
            if (in.equals("exit")) {
                webSocketServer.stop(1000);
                break;
            }
        }
    }
}

• 客户端：WebSocketClient

package com.arthur;

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

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URI;
import java.net.URISyntaxException;
import java.nio.ByteBuffer;

/**
 * 客户端
 */
public class MyWebSocketClient extends WebSocketClient {

    /**
     * 构造方法
     *
     * @param serverUri new URI("ws://127.0.0.1:8887/")
     */
    public MyWebSocketClient(URI serverUri) {
        super(serverUri);
        System.out.println("MyWebSocketClient");
    }

    /**
     * 连接成功的回调
     *
     * @param serverHandshake The handshake of the websocket instance
     */
    @Override
    public void onOpen(ServerHandshake serverHandshake) {
        System.out.println("onOpen: getContent = " + serverHandshake.getContent()
                + ", getHttpStatus = " + serverHandshake.getHttpStatus()
                + ", getHttpStatusMessage = " + serverHandshake.getHttpStatusMessage());
    }

    /**
     * 收到来自服务端的消息 字符串
     *
     * @param s The UTF-8 decoded message that was received.
     */
    @Override
    public void onMessage(String s) {
        System.out.println("onMessage: String = " + s);
    }

    /**
     * 收到来自服务端的消息 字节数组
     * @param bytes The binary message that was received.
     */
    @Override
    public void onMessage(ByteBuffer bytes) {
        super.onMessage(bytes);
        System.out.println("onMessage: ByteBuffer = " + bytes.toString());
    }

    /**
     * 连接关闭的回调
     * @param code   The codes can be looked up here: {@link org.java_websocket.framing.CloseFrame}
     * @param reason Additional information string
     * @param remote Returns whether or not the closing of the connection was initiated by the remote
     *               host.
     */
    @Override
    public void onClose(int code, String reason, boolean remote) {
        System.out.println("onClose: code = " + code
                + ", reason = " + reason
                + ", remote = " + remote);
    }

    /**
     * 连接错误的回调
     * @param e The exception causing this error
     */
    @Override
    public void onError(Exception e) {
        System.out.println("onError: " + e.toString());
    }

    public static void main(String[] args) throws IOException {

        try {
            //实例WebSocketClient对象，并连接到WebSocket服务端
            MyWebSocketClient client = new MyWebSocketClient(new URI("ws://127.0.0.1:8887/"));
            client.connect();
            //等待服务端响应


            while (!client.getReadyState().equals(ReadyState.OPEN)) {
                System.out.println("连接中···请稍后");
                Thread.sleep(1000);
            }

            BufferedReader sysin = new BufferedReader(new InputStreamReader(System.in));
            while (true) {
                String in = sysin.readLine();
                if (in.equals("exit")) {
                    client.close();
                    break;
                }
                client.send(in);
            }

        } catch (URISyntaxException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

• 分别运行客户端和服务端代码