Websocket（Jetty）：如何在服务器端处理以块形式出现的二进制数据

Question

我需要设置一个web套接字服务器，它可以接收客户端发送的音频数据。我用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）{}方法之间的区别是什么？

Answer 1

根据onMessagejavadoc，onMessage（字节[]b）和onMessage（InputStream is）都将接收整个消息：

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

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

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

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

另外请注意，在Handler类中，您可能只能同时定义其中一个方法：

对于每种本机WebSocket消息格式，每个webSocketendpoint可能只有一种消息处理方法：文本、二进制和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-protocol功能，或者实现客户端使用的相同“应用程序级协议”。

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

@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
        );
    }

}

此实现还意味着默认ServerEndpoint Config。使用Configurator，因此每个连接正好有一个Endpoint实例（如本文所述）。

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