From e1418018c6488eb4c09ec2e2b8a95cf42458ea19 Mon Sep 17 00:00:00 2001
From: Nikolay Kim <fafhrd91@gmail.com>
Date: Wed, 24 Oct 2018 10:43:30 -0700
Subject: [PATCH] add BytesCodec

---
 src/codec/bcodec.rs | 33 +++++++++++++++++++++++++++++++++
 src/codec/mod.rs    |  2 ++
 2 files changed, 35 insertions(+)
 create mode 100644 src/codec/bcodec.rs

diff --git a/src/codec/bcodec.rs b/src/codec/bcodec.rs
new file mode 100644
index 00000000..d7523bf0
--- /dev/null
+++ b/src/codec/bcodec.rs
@@ -0,0 +1,33 @@
+use std::io;
+
+use bytes::{Bytes, BytesMut};
+use tokio_codec::{Decoder, Encoder};
+
+/// Bytes codec.
+///
+/// Reads/Writes chunks of bytes from a stream.
+#[derive(Debug, Copy, Clone)]
+pub struct BytesCodec;
+
+impl Encoder for BytesCodec {
+    type Item = Bytes;
+    type Error = io::Error;
+
+    fn encode(&mut self, item: Bytes, dst: &mut BytesMut) -> Result<(), Self::Error> {
+        dst.extend_from_slice(&item[..]);
+        Ok(())
+    }
+}
+
+impl Decoder for BytesCodec {
+    type Item = BytesMut;
+    type Error = io::Error;
+
+    fn decode(&mut self, src: &mut BytesMut) -> Result<Option<Self::Item>, Self::Error> {
+        if src.is_empty() {
+            Ok(None)
+        } else {
+            Ok(Some(src.take()))
+        }
+    }
+}
diff --git a/src/codec/mod.rs b/src/codec/mod.rs
index 54aa35ff..140e77bb 100644
--- a/src/codec/mod.rs
+++ b/src/codec/mod.rs
@@ -12,11 +12,13 @@
 
 #![deny(missing_docs, missing_debug_implementations, warnings)]
 
+mod bcodec;
 mod framed;
 mod framed2;
 mod framed_read;
 mod framed_write;
 
+pub use self::bcodec::BytesCodec;
 pub use self::framed::{Framed, FramedParts};
 pub use self::framed2::{Framed2, FramedParts2};
 pub use self::framed_read::FramedRead;