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

use std::io::{self, Cursor, Read};

#[cfg(feature = "message_compress_gzip")]
use flate2::{
    bufread::{GzDecoder, GzEncoder},
    Compression,
};

/// Compress the input bytes.
#[cfg(feature = "message_compress_gzip")]
pub fn pack_gzip<S>(d: S) -> io::Result<Vec<u8>>
where
    S: AsRef<[u8]>,
{
    // ref. "golang/compress/flag.DefaultCompression" is -1 which is level 6
    // "Compression::default()" returns 6
    let mut gz = GzEncoder::new(Cursor::new(d), Compression::new(6));
    let mut encoded = Vec::new();
    gz.read_to_end(&mut encoded)?;
    Ok(encoded)
}

/// Decompress the input bytes.
#[cfg(feature = "message_compress_gzip")]
pub fn unpack_gzip<S>(d: S) -> io::Result<Vec<u8>>
where
    S: AsRef<[u8]>,
{
    let mut gz = GzDecoder::new(Cursor::new(d));
    let mut decoded = Vec::new();
    gz.read_to_end(&mut decoded)?;
    Ok(decoded)
}