From 7052d2454a2370ab9583f63711df89f3bd7bec83 Mon Sep 17 00:00:00 2001
From: Paolo Barbolini <paolo.barbolini@m4ss.net>
Date: Tue, 23 Apr 2024 21:55:34 +0200
Subject: [PATCH] Reuse capacity when possible in <BytesMut as Buf>::advance
 impl

---
 src/bytes_mut.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

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