Function copy_reader_to_ws

crates/openshell-server/src/ws_tunnel.rs:93–124

Copy bytes from an async reader into WebSocket binary frames.

(
    mut reader: R,
    mut ws_sink: futures::stream::SplitSink<axum::extract::ws::WebSocket, Message>,
)

/// Copy bytes from an async reader into WebSocket binary frames.
async fn copy_reader_to_ws<R>(
    mut reader: R,
    mut ws_sink: futures::stream::SplitSink<axum::extract::ws::WebSocket, Message>,
) -> Result<(), Box<dyn std::error::Error + Send + Sync>>
where
    R: AsyncRead + Unpin,
{
    let mut buf = vec![0u8; 32 * 1024];
    loop {
        match reader.read(&mut buf).await {
            Ok(0) => {
                let _ = ws_sink.close().await;
                break;
            }
            Ok(n) => {
                if ws_sink
                    .send(Message::Binary(buf[..n].to_vec().into()))
                    .await
                    .is_err()
                {
                    break;
                }
            }
            Err(e) => {
                debug!(error = %e, "WS tunnel: read error");
                let _ = ws_sink.close().await;
                break;
            }
        }
    }
    Ok(())
}

/// Copy bytes from WebSocket binary/text frames into an async writer.
async fn copy_ws_to_writer<W>(