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

根据onMessagejavadoc， 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），这表明客户端没有使用协议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 所述）。

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