added rkyv

Author: maccesch

CHANGELOG.md

@@ -5,4 +5,4 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [0.1.0] - 2023-07-07
 
-Initial release.
+Initial release.

Cargo.toml

@@ -15,6 +15,7 @@ repository = "https://github.com/Synphonyte/codee"
 base64 = { version = "0.21", optional = true }
 bincode = { version = "1", optional = true }
 prost = { version = "0.12", optional = true }
+rkyv = { version = "0.7", optional = true, features = ["validation", "strict"] }
 rmp-serde = { version = "1.1", optional = true }
 serde = { version = "1", optional = true }
 serde_json = { version = "1", optional = true }

src/binary/mod.rs

@@ -5,6 +5,8 @@ mod from_to_bytes;
 mod msgpack_serde;
 #[cfg(feature = "prost")]
 mod prost;
+#[cfg(feature = "rkyv")]
+mod rkyv;
 
 #[cfg(feature = "bincode_serde")]
 pub use bincode_serde::*;
@@ -14,3 +16,5 @@ pub use from_to_bytes::*;
 pub use msgpack_serde::*;
 #[cfg(feature = "prost")]
 pub use prost::*;
+#[cfg(feature = "rkyv")]
+pub use rkyv::*;

src/binary/rkyv.rs

@@ -0,0 +1,60 @@
+use crate::{Decoder, Encoder};
+use rkyv::de::deserializers::SharedDeserializeMap;
+use rkyv::ser::serializers::AllocSerializer;
+use rkyv::validation::validators::DefaultValidator;
+use rkyv::{Archive, CheckBytes, Deserialize, Fallible, Serialize};
+use std::error::Error;
+use std::sync::Arc;
+
+/// A codec that relies on `rkyv` to encode data in the msgpack format.
+///
+/// This is only available with the **`rkyv` feature** enabled.
+pub struct RkyvCodec;
+
+impl<T> Encoder<T> for RkyvCodec
+where
+    T: Serialize<AllocSerializer<1024>>,
+{
+    type Error = <AllocSerializer<1024> as Fallible>::Error;
+    type Encoded = Vec<u8>;
+
+    fn encode(val: &T) -> Result<Self::Encoded, Self::Error> {
+        Ok(rkyv::to_bytes::<T, 1024>(val)?.to_vec())
+    }
+}
+
+impl<T> Decoder<T> for RkyvCodec
+where
+    T: Archive,
+    for<'a> T::Archived:
+        'a + CheckBytes<DefaultValidator<'a>> + Deserialize<T, SharedDeserializeMap>,
+{
+    type Error = Arc<dyn Error>;
+    type Encoded = [u8];
+
+    fn decode(val: &Self::Encoded) -> Result<T, Self::Error> {
+        rkyv::from_bytes::<T>(val).map_err(|e| Arc::new(e) as Arc<dyn Error>)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_rkyv_codec() {
+        #[derive(Clone, Debug, PartialEq, rkyv::Archive, rkyv::Serialize, rkyv::Deserialize)]
+        #[archive(check_bytes)]
+        struct Test {
+            s: String,
+            i: i32,
+        }
+        let t = Test {
+            s: String::from("party time 🎉"),
+            i: 42,
+        };
+        let enc = RkyvCodec::encode(&t).unwrap();
+        let dec: Test = RkyvCodec::decode(&enc).unwrap();
+        assert_eq!(dec, t);
+    }
+}