Add comparable

src/buffer.rs

@@ -1,9 +1,12 @@
 use core::{
+  borrow::Borrow,
   marker::PhantomData,
   ptr::{self, NonNull},
   slice,
 };
 
+use crossbeam_skiplist::{Comparable, Equivalent};
+
 /// Returns when the bytes are too large to be written to the vacant buffer.
 #[derive(Debug, Default, Clone, Copy)]
 pub struct TooLarge {
@@ -166,9 +169,44 @@ impl<'a> AsMut<[u8]> for VacantBuffer<'a> {
   }
 }
 
-impl<'a> PartialEq<[u8]> for VacantBuffer<'a> {
-  fn eq(&self, other: &[u8]) -> bool {
-    self.as_ref().eq(other)
+impl<'a, Q> PartialEq<Q> for VacantBuffer<'a>
+where
+  [u8]: Borrow<Q>,
+  Q: ?Sized + Eq,
+{
+  fn eq(&self, other: &Q) -> bool {
+    self.as_ref().borrow().eq(other)
+  }
+}
+
+impl<'a, Q> PartialOrd<Q> for VacantBuffer<'a>
+where
+  [u8]: Borrow<Q>,
+  Q: ?Sized + Ord,
+{
+  fn partial_cmp(&self, other: &Q) -> Option<core::cmp::Ordering> {
+    #[allow(clippy::needless_borrow)]
+    Some(self.as_ref().borrow().cmp(&other))
+  }
+}
+
+impl<'a, Q> Equivalent<Q> for VacantBuffer<'a>
+where
+  [u8]: Borrow<Q>,
+  Q: ?Sized + Eq,
+{
+  fn equivalent(&self, key: &Q) -> bool {
+    self.as_ref().borrow().eq(key)
+  }
+}
+
+impl<'a, Q> Comparable<Q> for VacantBuffer<'a>
+where
+  [u8]: Borrow<Q>,
+  Q: ?Sized + Ord,
+{
+  fn compare(&self, other: &Q) -> core::cmp::Ordering {
+    self.as_ref().borrow().compare(other)
   }
 }
 
@@ -178,21 +216,33 @@ impl<'a> PartialEq<VacantBuffer<'a>> for [u8] {
   }
 }
 
+impl<'a> PartialOrd<VacantBuffer<'a>> for [u8] {
+  fn partial_cmp(&self, other: &VacantBuffer<'a>) -> Option<core::cmp::Ordering> {
+    Some(self.as_ref().cmp(other.as_ref()))
+  }
+}
+
 impl<'a> PartialEq<[u8]> for &VacantBuffer<'a> {
   fn eq(&self, other: &[u8]) -> bool {
     self.as_ref().eq(other)
   }
 }
 
+impl<'a> PartialOrd<[u8]> for &VacantBuffer<'a> {
+  fn partial_cmp(&self, other: &[u8]) -> Option<core::cmp::Ordering> {
+    Some(self.as_ref().cmp(other))
+  }
+}
+
 impl<'a> PartialEq<&VacantBuffer<'a>> for [u8] {
   fn eq(&self, other: &&VacantBuffer<'a>) -> bool {
     self.eq(other.as_ref())
   }
 }
 
-impl<'a, const N: usize> PartialEq<[u8; N]> for VacantBuffer<'a> {
-  fn eq(&self, other: &[u8; N]) -> bool {
-    self.as_ref().eq(other.as_ref())
+impl<'a> PartialOrd<&VacantBuffer<'a>> for [u8] {
+  fn partial_cmp(&self, other: &&VacantBuffer<'a>) -> Option<core::cmp::Ordering> {
+    Some(self.cmp(other.as_ref()))
   }
 }
 
@@ -202,6 +252,12 @@ impl<'a, const N: usize> PartialEq<VacantBuffer<'a>> for [u8; N] {
   }
 }
 
+impl<'a, const N: usize> PartialOrd<VacantBuffer<'a>> for [u8; N] {
+  fn partial_cmp(&self, other: &VacantBuffer<'a>) -> Option<core::cmp::Ordering> {
+    Some(self.as_ref().cmp(other.as_ref()))
+  }
+}
+
 impl<'a, const N: usize> PartialEq<&VacantBuffer<'a>> for [u8; N] {
   fn eq(&self, other: &&VacantBuffer<'a>) -> bool {
     self.as_ref().eq(other.as_ref())

src/tests.rs

@@ -6,7 +6,6 @@ use wal::ImmutableWal;
 
 const MB: usize = 1024 * 1024;
 
-#[macro_export]
 macro_rules! common_unittests {
   ($prefix:ident::$wal:ident) => {
     paste::paste! {