gelf 0.2.2

A library for logging GELF messages to a Graylog compatible server
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
41
42
43
44
45
46
47

use std::io;

use libflate::gzip;
use libflate::zlib;

use errors::{Result, ErrorKind, ResultExt};
use message::WireMessage;

/// MessageCompression represents all possible compression algorithms in GELF.
#[derive(PartialEq, Clone, Copy)]
pub enum MessageCompression {
    None,
    Gzip,
    Zlib,
}

impl MessageCompression {
    /// Return the default compression algorithm.
    pub fn default() -> MessageCompression {
        MessageCompression::Gzip
    }

    /// Compress a serialized message with the defined algorithm.
    pub fn compress(&self, message: &WireMessage) -> Result<Vec<u8>> {
        let json = message.to_gelf()?;

        Ok(match *self {
            MessageCompression::None => json.into_bytes(),
            MessageCompression::Gzip => {
                let mut cursor = io::Cursor::new(json);
                gzip::Encoder::new(Vec::new()).and_then(|mut encoder| {
                        io::copy(&mut cursor, &mut encoder)
                            .and_then(|_| encoder.finish().into_result())
                    })
                    .chain_err(|| ErrorKind::CompressMessageFailed("gzip"))?
            }
            MessageCompression::Zlib => {
                let mut cursor = io::Cursor::new(json);
                zlib::Encoder::new(Vec::new()).and_then(|mut encoder| {
                        io::copy(&mut cursor, &mut encoder)
                            .and_then(|_| encoder.finish().into_result())
                    })
                    .chain_err(|| ErrorKind::CompressMessageFailed("zlib"))?
            }
        })
    }
}