add init data

tickbh · Sep 12, 2024 · d28ccb9 · d28ccb9
1 parent 277899f
commit d28ccb9
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -3,7 +3,7 @@ members = ["algorithm-macro"]
 
 [package]
 name = "algorithm"
-version = "0.1.9"
+version = "0.1.10"
 edition = "2021"
 authors = ["tickbh <[email protected]>"]
 description = "about algorithm data structure, now has ttl with lru/lru-k/lfu/arc and slab/rbtree/roaring_bitmap/timer_wheelss, 关于算法常用的数据结构"

diff --git a/src/buf/binary_mut.rs b/src/buf/binary_mut.rs
@@ -289,6 +289,13 @@ impl BinaryMut {
         }
     }
 
+    pub fn data_mut(&mut self) -> &mut [u8] {
+        if self.wpos + 128 < self.vec.capacity() {
+            self.reserve(128);
+        }
+        &mut self.vec[self.wpos..]
+    }
+
     #[inline]
     pub unsafe fn try_resort_memory(&mut self) {
         if self.vec.len() < self.resort || self.cursor < self.resort / 2 {