From 373f93a629313dccd1858e403e99fb25709947a4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tobias=20Mu=CC=88ller?= Date: Mon, 28 Mar 2016 01:28:03 +0200 Subject: [PATCH] Implement BufRead for Chain --- src/libstd/io/mod.rs | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/src/libstd/io/mod.rs b/src/libstd/io/mod.rs index 28492b30e0f..ce14e5de90a 100644 --- a/src/libstd/io/mod.rs +++ b/src/libstd/io/mod.rs @@ -1442,6 +1442,27 @@ impl Read for Chain { } } +#[stable(feature = "chain_bufread", since = "1.9.0")] +impl BufRead for Chain { + fn fill_buf(&mut self) -> Result<&[u8]> { + if !self.done_first { + match try!(self.first.fill_buf()) { + buf if buf.len() == 0 => { self.done_first = true; } + buf => return Ok(buf), + } + } + self.second.fill_buf() + } + + fn consume(&mut self, amt: usize) { + if !self.done_first { + self.first.consume(amt) + } else { + self.second.consume(amt) + } + } +} + /// Reader adaptor which limits the bytes read from an underlying reader. /// /// This struct is generally created by calling [`take()`][take] on a reader.