feat: BufMut with custom allocator

YoloDev · Nov 3, 2024 · 4a86774 · 4a86774
1 parent 8310013
commit 4a86774
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/crates/aligned-buffer/src/bytes.rs b/crates/aligned-buffer/src/bytes.rs
@@ -1,8 +1,10 @@
-use crate::{raw::RawAlignedBuffer, UniqueAlignedBuffer};
+use crate::{alloc::BufferAllocator, raw::RawAlignedBuffer, UniqueAlignedBuffer};
 use bytes::{buf::UninitSlice, Buf, BufMut};
 use std::ptr;
 
-unsafe impl<const ALIGNMENT: usize> BufMut for UniqueAlignedBuffer<ALIGNMENT> {
+unsafe impl<const ALIGNMENT: usize, A: BufferAllocator<ALIGNMENT>> BufMut
+	for UniqueAlignedBuffer<ALIGNMENT, A>
+{
 	#[inline]
 	fn remaining_mut(&self) -> usize {
 		RawAlignedBuffer::<ALIGNMENT>::MAX_CAPACITY - self.len()