Skip to content

Commit 4c233d1

Browse files
author
Palmer Cox
committed
Update LimitReader to take the Reader to wrap by value
1 parent d4dd4c6 commit 4c233d1

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

src/libstd/io/util.rs

+8-7
Original file line numberDiff line numberDiff line change
@@ -14,19 +14,20 @@ use io;
1414
use vec::bytes::MutableByteVector;
1515

1616
/// Wraps a `Reader`, limiting the number of bytes that can be read from it.
17-
pub struct LimitReader<'a, R> {
17+
pub struct LimitReader<R> {
1818
priv limit: uint,
19-
priv inner: &'a mut R
19+
priv inner: R
2020
}
2121

22-
impl<'a, R: Reader> LimitReader<'a, R> {
22+
impl<R: Reader> LimitReader<R> {
2323
/// Creates a new `LimitReader`
24-
pub fn new<'a>(r: &'a mut R, limit: uint) -> LimitReader<'a, R> {
24+
pub fn new(r: R, limit: uint) -> LimitReader<R> {
2525
LimitReader { limit: limit, inner: r }
2626
}
27+
pub fn unwrap(self) -> R { self.inner }
2728
}
2829

29-
impl<'a, R: Reader> Reader for LimitReader<'a, R> {
30+
impl<R: Reader> Reader for LimitReader<R> {
3031
fn read(&mut self, buf: &mut [u8]) -> io::IoResult<uint> {
3132
if self.limit == 0 {
3233
return Err(io::standard_error(io::EndOfFile));
@@ -192,7 +193,7 @@ mod test {
192193
fn test_bounded_reader_unlimited() {
193194
let mut r = MemReader::new(~[0, 1, 2]);
194195
{
195-
let mut r = LimitReader::new(&mut r, 4);
196+
let mut r = LimitReader::new(r.by_ref(), 4);
196197
assert_eq!(~[0, 1, 2], r.read_to_end().unwrap());
197198
}
198199
}
@@ -201,7 +202,7 @@ mod test {
201202
fn test_bound_reader_limited() {
202203
let mut r = MemReader::new(~[0, 1, 2]);
203204
{
204-
let mut r = LimitReader::new(&mut r, 2);
205+
let mut r = LimitReader::new(r.by_ref(), 2);
205206
assert_eq!(~[0, 1], r.read_to_end().unwrap());
206207
}
207208
assert_eq!(~[2], r.read_to_end().unwrap());

0 commit comments

Comments
 (0)