rari 0.13.6

rari: Runtime Accelerated Rendering Infrastructure - A high-performance React Server Components framework powered by Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

pub mod stream;

pub use stream::{CompressionEncoding, compress_stream};

use bytes::Bytes;
use futures::StreamExt;

pub async fn compress_body(
    body: Bytes,
    encoding: CompressionEncoding,
) -> (Bytes, CompressionEncoding) {
    if matches!(encoding, CompressionEncoding::Identity) || body.is_empty() {
        return (body, CompressionEncoding::Identity);
    }

    let body_clone = body.clone();
    let input_stream =
        futures::stream::once(async move { Ok::<Bytes, std::io::Error>(body_clone) });
    let mut compressed_stream = compress_stream(input_stream, encoding);

    let mut compressed = Vec::new();
    while let Some(chunk) = compressed_stream.next().await {
        match chunk {
            Ok(data) => compressed.extend_from_slice(&data),
            Err(_) => return (body, CompressionEncoding::Identity),
        }
    }

    if compressed.len() < body.len() {
        (Bytes::from(compressed), encoding)
    } else {
        (body, CompressionEncoding::Identity)
    }
}