java - 网络套接字( jetty ): How to handle the binary data on server side which is coming in chunks

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) {
    }
}

由于音频文件很大，客户端以 block 的形式发送这些文件。现在对于每个 block ，都会调用 onMessage(bytes [] b) {} 方法。

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

此外，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 将自动重新组合 block 并将整个消息传递给您的方法(作为 byte[] 数组或 InputStream).

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

另请注意，您一次只能在 Handler 类中定义其中一种方法:

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
}

并手动缓冲您的数据(在内存或磁盘文件等中)

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

---

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

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

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

@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 )。

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