Introduce the obkv heed codec

src/heed_codec/mod.rs

@@ -2,6 +2,7 @@ mod beu32_str_codec;
 mod bo_roaring_bitmap_codec;
 mod cbo_roaring_bitmap_codec;
 mod csv_string_record_codec;
+mod obkv_codec;
 mod roaring_bitmap_codec;
 mod str_str_u8_codec;
 
@@ -9,5 +10,6 @@ pub use self::beu32_str_codec::BEU32StrCodec;
 pub use self::bo_roaring_bitmap_codec::BoRoaringBitmapCodec;
 pub use self::cbo_roaring_bitmap_codec::CboRoaringBitmapCodec;
 pub use self::csv_string_record_codec::CsvStringRecordCodec;
+pub use self::obkv_codec::ObkvCodec;
 pub use self::roaring_bitmap_codec::RoaringBitmapCodec;
 pub use self::str_str_u8_codec::StrStrU8Codec;

src/heed_codec/obkv_codec.rs

@@ -0,0 +1,20 @@
+use std::borrow::Cow;
+use obkv::{KvReader, KvWriter};
+
+pub struct ObkvCodec;
+
+impl<'a> heed::BytesDecode<'a> for ObkvCodec {
+    type DItem = KvReader<'a>;
+
+    fn bytes_decode(bytes: &'a [u8]) -> Option<Self::DItem> {
+        Some(KvReader::new(bytes))
+    }
+}
+
+impl heed::BytesEncode<'_> for ObkvCodec {
+    type EItem = KvWriter<Vec<u8>>;
+
+    fn bytes_encode(item: &Self::EItem) -> Option<Cow<[u8]>> {
+        item.clone().into_inner().map(Cow::Owned).ok()
+    }
+}