Reuse capacity when possible in <BytesMut as Buf>::advance impl

src/bytes_mut.rs

@@ -1066,12 +1066,21 @@ 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`: {:?} <= {:?}",
             cnt,
             self.remaining(),
         );
+
         unsafe {
             // SAFETY: We've checked that `cnt` <= `self.remaining()` and we know that
             // `self.remaining()` <= `self.cap`.