From cd182daf35eae3739511f3751843ad01f0d489b3 Mon Sep 17 00:00:00 2001 From: Till Hoeppner Date: Thu, 16 Apr 2015 23:32:00 +0200 Subject: Add untested support for binary serialization with rustc_serialize and bincode --- src/format/binary.rs | 33 +++++++++++++++++++++++++++++++++ src/format/mod.rs | 3 ++- src/format/weechat3.rs | 2 +- src/lib.rs | 6 +++++- src/log.rs | 2 +- 5 files changed, 42 insertions(+), 4 deletions(-) create mode 100644 src/format/binary.rs diff --git a/src/format/binary.rs b/src/format/binary.rs new file mode 100644 index 0000000..aae760c --- /dev/null +++ b/src/format/binary.rs @@ -0,0 +1,33 @@ +use std::io::{ self, BufRead, Write }; +use std::borrow::ToOwned; +use std::iter::{ Iterator }; + +use log::Event; +use format::{ Encode, Decode }; + +use bincode::{ self, SizeLimit }; + +pub struct Binary; + +pub struct Iter where R: BufRead { + input: R, +} + +impl Iterator for Iter where R: BufRead { + type Item = ::Result; + fn next(&mut self) -> Option<::Result> { + Some(bincode::decode_from(&mut self.input, SizeLimit::Infinite).map_err(|_| ::IlcError::BincodeDecode)) + } +} + +impl Encode for Binary where W: Write { + fn encode(&self, mut output: W, event: &Event) -> ::Result<()> { + bincode::encode_into(event, &mut output, SizeLimit::Infinite).map_err(|_| ::IlcError::BincodeEncode) + } +} + +impl Decode> for Binary where R: BufRead { + fn decode(&mut self, input: R) -> Iter { + Iter { input: input } + } +} diff --git a/src/format/mod.rs b/src/format/mod.rs index 9f6d30d..2c271bd 100644 --- a/src/format/mod.rs +++ b/src/format/mod.rs @@ -7,9 +7,10 @@ use std::io::{ self, BufRead, Write }; use log::Event; pub mod weechat3; +pub mod binary; pub trait Encode where W: Write { - fn encode(&self, output: W, event: &Event) -> io::Result<()>; + fn encode(&self, output: W, event: &Event) -> ::Result<()>; } pub trait Decode where R: BufRead, O: Iterator> { diff --git a/src/format/weechat3.rs b/src/format/weechat3.rs index 0904ced..59bb072 100644 --- a/src/format/weechat3.rs +++ b/src/format/weechat3.rs @@ -95,7 +95,7 @@ impl Decode> for Weechat3 where R: BufRead { } impl Encode for Weechat3 where W: Write { - fn encode(&self, mut output: W, event: &Event) -> io::Result<()> { + fn encode(&self, mut output: W, event: &Event) -> ::Result<()> { fn date(t: i64) -> String { format!("{}", UTC.timestamp(t, 0).format(TIME_DATE_FORMAT)) } diff --git a/src/lib.rs b/src/lib.rs index a160e4b..aabaae3 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -1,9 +1,11 @@ -#![feature(plugin, slice_patterns, core)] +#![feature(plugin, slice_patterns, core, custom_derive)] #![plugin(regex_macros)] extern crate regex; extern crate chrono; #[macro_use] extern crate log as l; +extern crate rustc_serialize; +extern crate bincode; pub mod log; pub mod format; @@ -19,6 +21,8 @@ pub type Result = result::Result; pub enum IlcError { Parse(String), Chrono(ParseError), + BincodeDecode, + BincodeEncode, Io(io::Error) } diff --git a/src/log.rs b/src/log.rs index b15d128..c0faaae 100644 --- a/src/log.rs +++ b/src/log.rs @@ -9,7 +9,7 @@ pub struct Log { /// All representable events, such as messages, quits, joins /// and topic changes. -#[derive(Debug)] +#[derive(Debug, RustcEncodable, RustcDecodable)] pub enum Event { Connect { time: i64 -- cgit v1.2.3