From 9dbb5c37c273091d8cc6762a464392309413d0a0 Mon Sep 17 00:00:00 2001
From: Titouan Rigoudy <titouan.rigoudy@gmail.com>
Date: Sat, 22 Sep 2018 23:40:24 +0000
Subject: [PATCH] Implement tokio_codec::Decoder for all message types.

---
 src/proto/codec.rs | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/src/proto/codec.rs b/src/proto/codec.rs
index 91f979a..7863430 100644
--- a/src/proto/codec.rs
+++ b/src/proto/codec.rs
@@ -4,7 +4,7 @@ use std::marker;
 use tokio_codec;
 use bytes::BytesMut;
 
-use super::base_codec::{ProtoEncode, ProtoEncoder};
+use super::base_codec::{Decode, ProtoEncode, ProtoEncoder};
 use super::server::{ServerRequest,ServerResponse};
 use super::peer::Message;
 
@@ -30,3 +30,21 @@ impl<T: ProtoEncode> tokio_codec::Encoder for Encoder<T> {
 pub type ServerRequestEncoder = Encoder<ServerRequest>;
 pub type ServerResponseEncoder = Encoder<ServerResponse>;
 pub type PeerMessageEncoder = Encoder<Message>;
+
+struct Decoder<T> {
+    data: marker::PhantomData<T>
+}
+
+impl<T> tokio_codec::Decoder for Decoder<T>
+where BytesMut: Decode<T> {
+    type Item = T;
+    type Error = io::Error;
+
+    fn decode(&mut self, src: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
+        Ok(Some(src.decode()?))
+    }
+}
+
+pub type ServerRequestDecoder = Decoder<ServerRequest>;
+pub type ServerResponseDecoder = Decoder<ServerResponse>;
+pub type PeerMessageDecoder = Decoder<Message>;