Skip to content

Commit

Permalink
Rollup merge of rust-lang#23847 - bcoopers:read_clarification, r=sfac…
Browse files Browse the repository at this point in the history
…kler

This introduces no functional changes except for reducing a few unnecessary operations and variables.  Vec has the behavior that, if you request space past the capacity with reserve(), it will round up to the nearest power of 2.  What that effectively means is that after the first call to reserve(16), we are doubling our capacity every time.  So using the DEFAULT_BUF_SIZE and doubling cap_size() here is meaningless and has no effect on the call to reserve().

Note that with rust-lang#23842 implemented this will hopefully have a clearer API and less of a need for commenting.  If rust-lang#23842 is not implemented then the most clear implementation would be to call reserve_exact(buf.capacity()) at every step (and making sure that buf.capacity() is not zero at the beginning of the function of course).

Edit- functional change now introduced.  We will now zero 16 bytes of the vector first, then double to 32, then 64, etc. until we read 64kB.  This stops us from zeroing the entire vector when we double it, some of which may be wasted work.  Reallocation still follows the doubling strategy, but the responsibility has been moved to vec.extend(), which calls reserve() and push_back().
  • Loading branch information
Manishearth committed Apr 1, 2015
2 parents 1d17e6e + 240734c commit abd747c
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions src/libstd/io/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,18 +101,14 @@ fn append_to_string<F>(buf: &mut String, f: F) -> Result<usize>
fn read_to_end<R: Read + ?Sized>(r: &mut R, buf: &mut Vec<u8>) -> Result<usize> {
let start_len = buf.len();
let mut len = start_len;
let mut cap_bump = 16;
let mut new_write_size = 16;
let ret;
loop {
if len == buf.len() {
if buf.capacity() == buf.len() {
if cap_bump < DEFAULT_BUF_SIZE {
cap_bump *= 2;
}
buf.reserve(cap_bump);
if new_write_size < DEFAULT_BUF_SIZE {
new_write_size *= 2;
}
let new_area = buf.capacity() - buf.len();
buf.extend(iter::repeat(0).take(new_area));
buf.extend(iter::repeat(0).take(new_write_size));
}

match r.read(&mut buf[len..]) {
Expand Down

0 comments on commit abd747c

Please sign in to comment.