feat(direct-io): Use take to avoid allocation when releasing ioBuffer (…

…#21)
zuston · Jan 7, 2025 · e3e6100 · e3e6100
1 parent 546000c
commit e3e6100
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/store/alignment/io_buffer_pool.rs b/src/store/alignment/io_buffer_pool.rs
@@ -101,7 +101,8 @@ impl<'a> RecycledIoBuffer<'a> {
 impl Drop for RecycledIoBuffer<'_> {
     fn drop(&mut self) {
         if let Some(pool) = self.pool_ref {
-            pool.release(self.internal.clone());
+            let taked = std::mem::take(&mut self.internal);
+            pool.release(taked);
         }
     }
 }

diff --git a/src/store/alignment/io_bytes.rs b/src/store/alignment/io_bytes.rs
@@ -48,6 +48,12 @@ impl IoBuffer {
     }
 }
 
+impl Default for IoBuffer {
+    fn default() -> Self {
+        IoBuffer::new(16 * 1024 * 1024)
+    }
+}
+
 impl Deref for IoBuffer {
     type Target = BoxA<[u8], &'static AlignedAllocator<ALIGN>>;