|
9 | 9 | //! As a workaround, zerogc introduces a `GcDeserialize` type,
|
10 | 10 | //! indicating an implementation of [serde::Deserialize]
|
11 | 11 | //! that requires a [GcContext].
|
| 12 | +use std::collections::{HashMap, HashSet}; |
12 | 13 | use std::marker::PhantomData;
|
| 14 | +use std::hash::{BuildHasher, Hash}; |
13 | 15 |
|
14 |
| -use crate::array::{GcArray, GcString}; |
15 |
| -use serde::{Serialize, de::{self, Deserializer, Visitor, DeserializeSeed}, ser::SerializeSeq}; |
| 16 | +use serde::Serialize; |
| 17 | +use serde::de::{self, Deserializer, Visitor, DeserializeSeed, MapAccess, SeqAccess}; |
| 18 | +use serde::ser::SerializeSeq; |
| 19 | + |
| 20 | +#[cfg(feature = "indexmap")] |
| 21 | +use indexmap::{IndexMap, IndexSet}; |
16 | 22 |
|
| 23 | +use crate::array::{GcArray, GcString}; |
17 | 24 | use crate::prelude::*;
|
18 | 25 |
|
19 | 26 | #[doc(hidden)]
|
@@ -161,3 +168,107 @@ impl<'de, 'gc, Id: CollectorId, T: GcDeserialize<'gc, 'de, Id>> DeserializeSeed<
|
161 | 168 | T::deserialize_gc(self.context, deserializer)
|
162 | 169 | }
|
163 | 170 | }
|
| 171 | + |
| 172 | +macro_rules! impl_for_map { |
| 173 | + ($target:ident <K, V, S> $(where $($bounds:tt)*)?) => { |
| 174 | + impl<'gc, 'de, Id: CollectorId, |
| 175 | + K: Eq + Hash + GcDeserialize<'gc, 'de, Id>, |
| 176 | + V: GcDeserialize<'gc, 'de, Id>, |
| 177 | + S: BuildHasher + Default |
| 178 | + > GcDeserialize<'gc, 'de, Id> for $target<K, V, S> $(where $($bounds)*)* { |
| 179 | + fn deserialize_gc<D: Deserializer<'de>>(ctx: &'gc <Id::System as GcSystem>::Context, deserializer: D) -> Result<Self, D::Error> { |
| 180 | + struct MapVisitor< |
| 181 | + 'gc, 'de, Id: CollectorId, |
| 182 | + K: GcDeserialize<'gc, 'de, Id>, |
| 183 | + V: GcDeserialize<'gc, 'de, Id>, |
| 184 | + S: BuildHasher + Default |
| 185 | + > { |
| 186 | + ctx: &'gc <Id::System as GcSystem>::Context, |
| 187 | + marker: PhantomData<(&'de S, K, V)> |
| 188 | + } |
| 189 | + impl<'gc, 'de, Id: CollectorId, |
| 190 | + K: Eq + Hash + GcDeserialize<'gc, 'de, Id>, |
| 191 | + V: GcDeserialize<'gc, 'de, Id>, |
| 192 | + S: BuildHasher + Default |
| 193 | + > Visitor<'de> for MapVisitor<'gc, 'de, Id, K, V, S> { |
| 194 | + type Value = $target<K, V, S>; |
| 195 | + fn expecting(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { |
| 196 | + f.write_str(concat!("a ", stringify!($target))) |
| 197 | + } |
| 198 | + #[inline] |
| 199 | + fn visit_map<A>(self, mut access: A) -> Result<Self::Value, A::Error> |
| 200 | + where A: MapAccess<'de>, { |
| 201 | + let mut values = $target::<K, V, S>::with_capacity_and_hasher( |
| 202 | + access.size_hint().unwrap_or(0).min(1024), |
| 203 | + S::default() |
| 204 | + ); |
| 205 | + while let Some((key, value)) = access.next_entry_seed( |
| 206 | + GcDeserializeSeed::new(self.ctx), |
| 207 | + GcDeserializeSeed::new(self.ctx) |
| 208 | + )? { |
| 209 | + values.insert(key, value); |
| 210 | + } |
| 211 | + |
| 212 | + Ok(values) |
| 213 | + } |
| 214 | + } |
| 215 | + let visitor: MapVisitor<Id, K, V, S> = MapVisitor { ctx, marker: PhantomData }; |
| 216 | + deserializer.deserialize_map(visitor) |
| 217 | + } |
| 218 | + } |
| 219 | + }; |
| 220 | +} |
| 221 | + |
| 222 | +macro_rules! impl_for_set { |
| 223 | + ($target:ident <T, S> $(where $($bounds:tt)*)?) => { |
| 224 | + impl<'gc, 'de, Id: CollectorId, |
| 225 | + T: Eq + Hash + GcDeserialize<'gc, 'de, Id>, |
| 226 | + S: BuildHasher + Default |
| 227 | + > GcDeserialize<'gc, 'de, Id> for $target<T, S> $(where $($bounds)*)* { |
| 228 | + fn deserialize_gc<D: Deserializer<'de>>(ctx: &'gc <Id::System as GcSystem>::Context, deserializer: D) -> Result<Self, D::Error> { |
| 229 | + struct SetVisitor< |
| 230 | + 'gc, 'de, Id: CollectorId, |
| 231 | + T: GcDeserialize<'gc, 'de, Id>, |
| 232 | + S: BuildHasher + Default |
| 233 | + > { |
| 234 | + ctx: &'gc <Id::System as GcSystem>::Context, |
| 235 | + marker: PhantomData<fn(&'de (), S) -> T> |
| 236 | + } |
| 237 | + impl<'gc, 'de, Id: CollectorId, |
| 238 | + T: Eq + Hash + GcDeserialize<'gc, 'de, Id>, |
| 239 | + S: BuildHasher + Default |
| 240 | + > Visitor<'de> for SetVisitor<'gc, 'de, Id, T, S> { |
| 241 | + type Value = $target<T, S>; |
| 242 | + fn expecting(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { |
| 243 | + f.write_str(concat!("a ", stringify!($target))) |
| 244 | + } |
| 245 | + #[inline] |
| 246 | + fn visit_seq<A>(self, mut access: A) -> Result<Self::Value, A::Error> |
| 247 | + where A: SeqAccess<'de>, { |
| 248 | + let mut values = $target::<T, S>::with_capacity_and_hasher( |
| 249 | + access.size_hint().unwrap_or(0).min(1024), |
| 250 | + S::default() |
| 251 | + ); |
| 252 | + while let Some(value) = access.next_element_seed( |
| 253 | + GcDeserializeSeed::new(self.ctx) |
| 254 | + )? { |
| 255 | + values.insert(value); |
| 256 | + } |
| 257 | + |
| 258 | + Ok(values) |
| 259 | + } |
| 260 | + } |
| 261 | + let visitor: SetVisitor<Id, T, S> = SetVisitor { ctx, marker: PhantomData }; |
| 262 | + deserializer.deserialize_seq(visitor) |
| 263 | + } |
| 264 | + } |
| 265 | + }; |
| 266 | +} |
| 267 | + |
| 268 | + |
| 269 | +impl_for_map!(HashMap<K, V, S> where K: TraceImmutable, S: 'static); |
| 270 | +impl_for_set!(HashSet<T, S> where T: TraceImmutable, S: 'static); |
| 271 | +#[cfg(feature = "indexmap")] |
| 272 | +impl_for_map!(IndexMap<K, V, S> where K: TraceImmutable, S: 'static); |
| 273 | +#[cfg(feature = "indexmap")] |
| 274 | +impl_for_set!(IndexSet<T, S> where T: TraceImmutable, S: 'static); |
0 commit comments