From 3c58b0c75cd6b89bd67570a604f667295d157945 Mon Sep 17 00:00:00 2001
From: Carl Lerche <me@carllerche.com>
Date: Wed, 21 Sep 2016 20:41:21 -0700
Subject: [PATCH] Add take fn to Buf & MutBuf

---
 src/buf/mod.rs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/buf/mod.rs b/src/buf/mod.rs
index b5cd568..acc2888 100644
--- a/src/buf/mod.rs
+++ b/src/buf/mod.rs
@@ -4,7 +4,7 @@ pub mod byte;
 pub mod ring;
 pub mod take;
 
-use {Bytes};
+use {Bytes, Take};
 use byteorder::ByteOrder;
 use std::{cmp, fmt, io, ptr, usize};
 
@@ -165,6 +165,11 @@ pub trait Buf {
         self.read_slice(&mut buf);
         T::read_f64(&buf)
     }
+
+    /// Create an adapter which will limit at most `limit` bytes from it.
+    fn take(self, limit: usize) -> Take<Self> where Self: Sized {
+        Take::new(self, limit)
+    }
 }
 
 /// A trait for values that provide sequential write access to bytes.
@@ -329,6 +334,11 @@ pub trait MutBuf {
         T::write_f64(&mut buf, n);
         self.write_slice(&buf)
     }
+
+    /// Create an adapter which will limit at most `limit` bytes from it.
+    fn take(self, limit: usize) -> Take<Self> where Self: Sized {
+        Take::new(self, limit)
+    }
 }
 
 /*