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)都将收到整个消息：

如果该方法正在处理二进制消息：

• byte[] 或 ByteBuffer 接收整个消息
• InputStream 以阻塞流的形式接收整个消息

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

二进制消息的最大大小，可以这样接收，用setMaxBinaryMessageSize设置

另请注意，您可能一次只能在 Handler 类中定义以下方法之一：

每个 websocket 端点对于每种原生 websocket 消息格式可能只有一种消息处理方法：文本、二进制和 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方法之一， 逐块发送数据。

---

如果您观察到的是为每个数据块调用的“完整消息”方法（即onMessage(byte[] b )，由客户端发送，则表明客户端没有利用协议功能来发送分块消息，而是拆分输入数据，然后将其部分作为独立的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 实例（如此处所述）。

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