From 2b5c034f133fe55aaf9e5d69c98cafcfee139a11 Mon Sep 17 00:00:00 2001 From: Till Hoeppner Date: Tue, 28 Jul 2015 00:26:03 +0200 Subject: Add msgpack output --- src/format/mod.rs | 3 +++ src/format/msgpack.rs | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 56 insertions(+) create mode 100644 src/format/msgpack.rs (limited to 'src/format') diff --git a/src/format/mod.rs b/src/format/mod.rs index 558de90..f918287 100644 --- a/src/format/mod.rs +++ b/src/format/mod.rs @@ -25,6 +25,7 @@ use context::Context; pub mod weechat3; pub mod energymech; pub mod binary; +pub mod msgpack; pub trait Encode<'a, W> where W: Write { fn encode(&'a self, context: &'a Context, output: W, event: &'a Event) -> ::Result<()>; @@ -53,6 +54,7 @@ pub fn decoder<'a>(format: &str) -> Option>> "energymech" => Some(Box::new(energymech::Energymech)), "weechat3" => Some(Box::new(weechat3::Weechat3)), "binary" => Some(Box::new(binary::Binary)), + "msgpack" => Some(Box::new(msgpack::Msgpack)), _ => None } } @@ -62,6 +64,7 @@ pub fn encoder<'a>(format: &str) -> Option>> { "energymech" => Some(Box::new(energymech::Energymech)), "weechat3" => Some(Box::new(weechat3::Weechat3)), "binary" => Some(Box::new(binary::Binary)), + "msgpack" => Some(Box::new(msgpack::Msgpack)), _ => None } } diff --git a/src/format/msgpack.rs b/src/format/msgpack.rs new file mode 100644 index 0000000..3519b76 --- /dev/null +++ b/src/format/msgpack.rs @@ -0,0 +1,53 @@ +// Copyright 2015 Till Höppner +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +use std::io::{ BufRead, Write }; +use std::iter::Iterator; +use std::marker::PhantomData; + +use event::Event; +use context::Context; +use format::{ Encode, Decode }; + +use rustc_serialize::{ Encodable, Decodable }; +use msgpack::{ Encoder, Decoder }; + +pub struct Msgpack; + +pub struct Iter<'a, R: 'a> where R: BufRead { + _phantom: PhantomData<&'a ()>, + input: R +} + +impl<'a, R: 'a> Iterator for Iter<'a, R> where R: BufRead { + type Item = ::Result>; + fn next(&mut self) -> Option<::Result>> { + Some(Decodable::decode(&mut Decoder::new(&mut self.input)) + .map_err(|_| ::IlcError::BincodeDecode)) + } +} + +impl<'a, W> Encode<'a, W> for Msgpack where W: Write { + fn encode(&'a self, _context: &'a Context, mut output: W, event: &'a Event) -> ::Result<()> { + event.encode(&mut Encoder::new(&mut output)) + .map_err(|_| ::IlcError::BincodeEncode) + } +} + +impl<'a, R: 'a> Decode<'a, R> for Msgpack where R: BufRead { + type Output = Iter<'a, R>; + fn decode(&'a mut self, _context: &'a Context, input: R) -> Iter { + Iter { _phantom: PhantomData, input: input } + } +} -- cgit v1.2.3