perf(types/buffer): skip copying in to_bytes when NonContiguous

contains a single `Bytes`
apache · Dec 4, 2024 · e6900f4 · e6900f4
1 parent ec1c53a
commit e6900f4
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/core/src/types/buffer.rs b/core/src/types/buffer.rs
@@ -287,10 +287,19 @@ impl Buffer {
     pub fn to_bytes(&self) -> Bytes {
         match &self.0 {
             Inner::Contiguous(bytes) => bytes.clone(),
-            Inner::NonContiguous { .. } => {
-                let mut ret = BytesMut::with_capacity(self.len());
-                ret.put(self.clone());
-                ret.freeze()
+            Inner::NonContiguous {
+                parts,
+                size,
+                idx: _,
+                offset,
+            } => {
+                if parts.len() == 1 {
+                    parts[0].slice(*offset..(*offset + *size))
+                } else {
+                    let mut ret = BytesMut::with_capacity(self.len());
+                    ret.put(self.clone());
+                    ret.freeze()
+                }
             }
         }
     }