rkyv/impls/indexmap/
index_map.rs

1use crate::{
2    collections::index_map::{ArchivedIndexMap, IndexMapResolver},
3    ser::{ScratchSpace, Serializer},
4    Archive, Deserialize, Fallible, Serialize,
5};
6use core::hash::{BuildHasher, Hash};
7use indexmap::IndexMap;
8
9impl<K: Archive, V: Archive, S> Archive for IndexMap<K, V, S> {
10    type Archived = ArchivedIndexMap<K::Archived, V::Archived>;
11    type Resolver = IndexMapResolver;
12
13    unsafe fn resolve(&self, pos: usize, resolver: Self::Resolver, out: *mut Self::Archived) {
14        ArchivedIndexMap::resolve_from_len(self.len(), pos, resolver, out);
15    }
16}
17
18impl<K, V, S, RandomState> Serialize<S> for IndexMap<K, V, RandomState>
19where
20    K: Hash + Eq + Serialize<S>,
21    V: Serialize<S>,
22    S: ScratchSpace + Serializer + ?Sized,
23    RandomState: BuildHasher,
24{
25    fn serialize(&self, serializer: &mut S) -> Result<IndexMapResolver, S::Error> {
26        unsafe {
27            ArchivedIndexMap::serialize_from_iter_index(
28                self.iter(),
29                |k| self.get_index_of(k).unwrap(),
30                serializer,
31            )
32        }
33    }
34}
35
36impl<K, V, D, S> Deserialize<IndexMap<K, V, S>, D> for ArchivedIndexMap<K::Archived, V::Archived>
37where
38    K: Archive + Hash + Eq,
39    K::Archived: Deserialize<K, D>,
40    V: Archive,
41    V::Archived: Deserialize<V, D>,
42    D: Fallible + ?Sized,
43    S: Default + BuildHasher,
44{
45    fn deserialize(&self, deserializer: &mut D) -> Result<IndexMap<K, V, S>, D::Error> {
46        let mut result = IndexMap::with_capacity_and_hasher(self.len(), S::default());
47        for (k, v) in self.iter() {
48            result.insert(k.deserialize(deserializer)?, v.deserialize(deserializer)?);
49        }
50        Ok(result)
51    }
52}
53
54impl<UK, K, UV, V, S> PartialEq<IndexMap<UK, UV, S>> for ArchivedIndexMap<K, V>
55where
56    K: PartialEq<UK>,
57    V: PartialEq<UV>,
58    S: BuildHasher,
59{
60    fn eq(&self, other: &IndexMap<UK, UV, S>) -> bool {
61        self.iter()
62            .zip(other.iter())
63            .all(|((ak, av), (bk, bv))| ak == bk && av == bv)
64    }
65}
66
67#[cfg(test)]
68mod tests {
69    use crate::{
70        archived_root,
71        ser::{serializers::AllocSerializer, Serializer},
72        Deserialize, Infallible,
73    };
74    use indexmap::{indexmap, IndexMap};
75
76    #[test]
77    fn index_map() {
78        let value = indexmap! {
79            String::from("foo") => 10,
80            String::from("bar") => 20,
81            String::from("baz") => 40,
82            String::from("bat") => 80,
83        };
84
85        let mut serializer = AllocSerializer::<4096>::default();
86        serializer.serialize_value(&value).unwrap();
87        let result = serializer.into_serializer().into_inner();
88        let archived = unsafe { archived_root::<IndexMap<String, i32>>(result.as_ref()) };
89
90        assert_eq!(value.len(), archived.len());
91        for (k, v) in value.iter() {
92            let (ak, av) = archived.get_key_value(k.as_str()).unwrap();
93            assert_eq!(k, ak);
94            assert_eq!(v, av);
95        }
96
97        let deserialized: IndexMap<String, i32> = archived.deserialize(&mut Infallible).unwrap();
98        assert_eq!(value, deserialized);
99    }
100
101    #[cfg(feature = "validation")]
102    #[test]
103    fn validate_index_map() {
104        use crate::check_archived_root;
105
106        let value = indexmap! {
107            String::from("foo") => 10,
108            String::from("bar") => 20,
109            String::from("baz") => 40,
110            String::from("bat") => 80,
111        };
112
113        let mut serializer = AllocSerializer::<4096>::default();
114        serializer.serialize_value(&value).unwrap();
115        let result = serializer.into_serializer().into_inner();
116        check_archived_root::<IndexMap<String, i32>>(result.as_ref())
117            .expect("failed to validate archived index map");
118    }
119}
rkyv/impls/indexmap/index_map.rs

rkyv/impls/indexmap/
index_map.rs
