1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use std::any::{Any, TypeId};
use std::collections::HashMap;
use std::sync::Arc;

#[derive(Clone)]
pub struct Data {
	inner: Arc<HashMap<TypeId, Box<dyn Any + Send + Sync>>>,
}

impl Data {
	pub(crate) fn new() -> Self {
		Self {
			inner: Arc::new(HashMap::new()),
		}
	}

	pub fn exists<D>(&self) -> bool
	where
		D: Any,
	{
		self.inner.contains_key(&TypeId::of::<D>())
	}

	/// returns true if the data already existed
	pub(crate) fn insert<D>(&mut self, data: D) -> bool
	where
		D: Any + Send + Sync,
	{
		let map = Arc::get_mut(&mut self.inner).unwrap();
		map.insert(data.type_id(), Box::new(data)).is_some()
	}

	pub fn get<D>(&self) -> Option<&D>
	where
		D: Any,
	{
		self.inner
			.get(&TypeId::of::<D>())
			.and_then(|a| a.downcast_ref())
	}
}

#[cfg(feature = "graphql")]
impl juniper::Context for Data {}