Reuse capacity when possible in <BytesMut as Buf>::advance impl (#698)

1 files changed, 8 insertions, 0 deletions

diff --git a/src/bytes_mut.rs b/src/bytes_mut.rs
index 0ea0272..35e1900 100644
--- a/src/bytes_mut.rs
+++ b/src/bytes_mut.rs
@@ -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`: {:?} <= {:?}",