feat(types/buffer): skip copying in to_bytes when NonContiguous c…

…ontains a single `Bytes` (#5388)
apache · Dec 4, 2024 · eaf6c18 · eaf6c18
1 parent 6cf1269
commit eaf6c18
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()
+                }
             }
         }
     }