msg-wire 0.1.6

Wire formats for msg-rs
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
35
36
37
38
39
40

use bytes::Bytes;
use std::io;
use zstd::{decode_all, stream::encode_all};

use super::{CompressionType, Compressor, Decompressor};

pub struct ZstdCompressor {
    level: i32,
}

impl ZstdCompressor {
    /// Creates a new zstd compressor with the given compression level (0-9).
    /// Default used by zstd is 3.
    pub fn new(level: i32) -> Self {
        Self { level }
    }
}

impl Compressor for ZstdCompressor {
    fn compression_type(&self) -> CompressionType {
        CompressionType::Zstd
    }

    fn compress(&self, data: &[u8]) -> Result<Bytes, io::Error> {
        let compressed = encode_all(data, self.level)?;

        Ok(Bytes::from(compressed))
    }
}

#[derive(Debug, Default)]
pub struct ZstdDecompressor;

impl Decompressor for ZstdDecompressor {
    fn decompress(&self, data: &[u8]) -> Result<Bytes, io::Error> {
        let decompressed = decode_all(data)?;

        Ok(Bytes::from(decompressed))
    }
}