Websocket(Jetty)：如何在服务器端处理成块的二进制数据

Question

我需要设置一个 websocket 服务器来接收客户端发送的音频数据。为此，我正在使用 Jetty。

我的处理程序代码：

{
    @OnWebSocketClose
    public void onClose(int statusCode, String reason) {
    }

    @OnWebSocketError
    public void onError(Throwable t) {
    }

    @OnWebSocketConnect
    public void onConnect(Session session) {

    }

    @OnWebSocketMessage
    public void onMessage(String message) {
    }

    @OnWebSocketMessage
    public void onMessage(bytes [] b) {
    }

    @OnWebSocketMessage
    public void inputStream(InputStream is) {
    }
}

由于音频文件很大，客户端正在分块发送。现在对于每个块，都会调用 onMessage(bytes [] b) {} 方法。

在服务器端，我需要添加这些块并处理音频。我该怎么做？

此外，onMessage(bytes [] b) {} 和 onMessage(InputStream is) {} 方法有什么区别？

Answer 1

根据 onMessage javadoc, onMessage(byte[] b) 和 onMessage(InputStream is) 都将收到整个消息：

if the method is handling binary messages:

byte[] or ByteBuffer to receive the whole message

InputStream to receive the whole message as a blocking stream

因此，如果您使用其中一种方法，Jetty 将自动重新组合块并将整个消息传递给您的方法（作为 byte[] 数组或 InputStream）。

您可以通过这种方式接收的二进制消息的最大大小由 setMaxBinaryMessageSize

设置

另请注意，您一次只能在处理程序 class 中定义其中一种方法：

Each websocket endpoint may only have one message handling method for each of the native websocket message formats: text, binary and pong.

如果您想按原样处理数据，您应该改用以下方法签名：

@OnMessage
public void processUpload(byte[] b, boolean last, Session session) {
    // process partial data here, which check on last to see if these is more on the way
}

并手动缓冲您的数据（在内存或磁盘文件中 e.t.c。）

反过来，客户端应使用 sendPartialBytes 方法之一逐块发送数据。

如果您观察到客户端发送的每个数据块都调用了“完整消息”方法（即 onMessage(byte[] b），则表明客户端未使用协议 capabilities 发送分块消息，而是拆分输入数据，然后将其部分作为独立 WS 消息发送，有效地创建它自己的应用程序级协议来分块传输数据。

在这种情况下，您必须更新客户端（如果可以的话），使其使用常规 WS 协议功能，或实现客户端正在使用的相同“应用程序级协议”。

下面是一个超级简单的端点示例，它将缓冲输入数据直到 WS 套接字关闭（这意味着客户端将在发送完所有块后关闭连接）。

@ServerEndpoint("/data")
public static class Handler {

    private ByteArrayOutputStream buffer = new ByteArrayOutputStream();

    @OnMessage
    public void onMessage(byte[] message) throws IOException {
        buffer.write(message);
    }

    @OnClose
    public void onClose(Session session) throws IOException {
        System.out.println(
            buffer.toByteArray().length
        );
    }

}

此实现还意味着使用默认 ServerEndpointConfig.Configurator，因此每个连接只有一个 Endpoint 实例（如记录 here）。

更复杂的实现可能需要重用套接字来发送多个文件，并指定一种机制来表示每次传输的开始和结束（例如，使用特殊格式的消息），但这完全取决于您的客户端如何已实施。

Websocket(Jetty)：如何在服务器端处理成块的二进制数据

Websocket(Jetty) : How to handle the binary data on server side which is coming in chunks

java

jetty

websocket

java-websocket

server