Support rkyv

Cargo.toml

@@ -20,14 +20,17 @@ simdutf8 = { version = "0.1", default-features = false, features = [
 redis = { version = "0.26", optional = true, default-features = false }
 itoa = { version = "1", optional = true }
 ryu = { version = "1", optional = true }
+rkyv = { version = "0.8", optional = true, default-features = false }
 
 [features]
 default = ["std"]
-std = ["bytes/std", "simdutf8/std", "serde?/std"]
+std = ["bytes/std", "simdutf8/std", "serde?/std", "rkyv?/std"]
 serde = ["serde/alloc"]
 serde-unsafe = ["serde"]
 redis = ["std", "dep:redis", "itoa", "ryu"]
 redis-unsafe = ["redis"]
+rkyv = ["rkyv/alloc"]
+
 
 [dev-dependencies]
 static_assertions = { version = "1" }

src/lib.rs

@@ -798,3 +798,6 @@ mod redis;
 
 #[cfg(feature = "serde")]
 mod serde;
+
+#[cfg(feature = "rkyv")]
+mod rkyv;

src/rkyv.rs

@@ -0,0 +1,48 @@
+#[cfg(not(feature = "std"))]
+use alloc::string::String;
+
+use rancor::{Fallible, Source};
+use rkyv::{
+    ser::{Allocator, Writer},
+    string::{ArchivedString, StringResolver},
+    *,
+};
+
+use super::FastStr;
+
+impl Archive for FastStr {
+    type Archived = ArchivedString;
+    type Resolver = StringResolver;
+
+    fn resolve(&self, resolver: Self::Resolver, out: Place<Self::Archived>) {
+        ArchivedString::resolve_from_str(self, resolver, out);
+    }
+}
+
+impl<S> Serialize<S> for FastStr
+where
+    S: Fallible + Allocator + Writer + ?Sized,
+    S::Error: Source,
+{
+    fn serialize(&self, serializer: &mut S) -> Result<Self::Resolver, S::Error> {
+        ArchivedString::serialize_from_str(self, serializer)
+    }
+}
+
+impl<D: Fallible + ?Sized> Deserialize<FastStr, D> for ArchivedString {
+    fn deserialize(&self, _deserializer: &mut D) -> Result<FastStr, D::Error> {
+        Ok(FastStr::new(self.as_str()))
+    }
+}
+
+impl PartialEq<FastStr> for ArchivedString {
+    fn eq(&self, other: &FastStr) -> bool {
+        other.as_str() == self.as_str()
+    }
+}
+
+impl PartialOrd<FastStr> for ArchivedString {
+    fn partial_cmp(&self, other: &FastStr) -> Option<::core::cmp::Ordering> {
+        Some(self.as_str().cmp(other.as_str()))
+    }
+}