Skip to content

Commit

Permalink
Reuse capacity when possible in <BytesMut as Buf>::advance impl (#698)
Browse files Browse the repository at this point in the history
  • Loading branch information
paolobarbolini authored Apr 25, 2024
1 parent ce09d7d commit baa5053
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions src/bytes_mut.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1066,6 +1066,14 @@ impl Buf for BytesMut {

#[inline]
fn advance(&mut self, cnt: usize) {
// Advancing by the length is the same as resetting the length to 0,
// except this way we get to reuse the full capacity.
if cnt == self.remaining() {
// SAFETY: Zero is not greater than the capacity.
unsafe { self.set_len(0) };
return;
}

assert!(
cnt <= self.remaining(),
"cannot advance past `remaining`: {:?} <= {:?}",
Expand Down

0 comments on commit baa5053

Please sign in to comment.