Websocket(Jetty):如何处理服务器端的二进制数据

Question

我需要设置一个websocket服务器,它可以接收客户端发送的音频数据.我正在使用码头.

处理程序代码:

{
    @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) {}方法.

谢谢,
Shantanu

Answer 1

按照onMessage javadoc的,都onMessage(byte[] b)和onMessage(InputStream is)将接收整个消息:

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

• byte []或ByteBuffer接收整个消息
• InputStream将整个消息作为阻塞流接收

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

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

另请注意,您可能只能在Handler类中同时定义其中一种方法:

每个websocket端点对于每个本地websocket消息格式可能只有一个消息处理方法:text,binary和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实例(如此处所述).

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