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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use std::borrow::Cow;
use std::collections::BTreeMap;
use std::fmt::Display;
use std::str::FromStr;
use async_graphql_parser::types::Field;
use async_graphql_parser::Positioned;
use async_graphql_value::{from_value, to_value};
use indexmap::IndexMap;
use serde::de::DeserializeOwned;
use serde::Serialize;
use crate::registry::{MetaType, Registry};
use crate::{
ContextSelectionSet, InputType, InputValueError, InputValueResult, Name, OutputType,
ServerResult, Value,
};
impl<K, V> InputType for BTreeMap<K, V>
where
K: ToString + FromStr + Ord + Send + Sync,
K::Err: Display,
V: Serialize + DeserializeOwned + Send + Sync,
{
type RawValueType = Self;
fn type_name() -> Cow<'static, str> {
Cow::Borrowed("JSONObject")
}
fn create_type_info(registry: &mut Registry) -> String {
registry.create_input_type::<Self, _>(|_| MetaType::Scalar {
name: <Self as InputType>::type_name().to_string(),
description: Some("A scalar that can represent any JSON Object value."),
is_valid: |_| true,
visible: None,
specified_by_url: None,
})
}
fn parse(value: Option<Value>) -> InputValueResult<Self> {
let value = value.unwrap_or_default();
match value {
Value::Object(map) => map
.into_iter()
.map(|(name, value)| {
Ok((
K::from_str(&name).map_err(|err| {
InputValueError::<Self>::custom(format!("object key: {}", err))
})?,
from_value(value).map_err(|err| format!("object value: {}", err))?,
))
})
.collect::<Result<_, _>>()
.map_err(InputValueError::propagate),
_ => Err(InputValueError::expected_type(value)),
}
}
fn to_value(&self) -> Value {
let mut map = IndexMap::new();
for (name, value) in self {
map.insert(
Name::new(name.to_string()),
to_value(value).unwrap_or_default(),
);
}
Value::Object(map)
}
fn as_raw_value(&self) -> Option<&Self::RawValueType> {
Some(self)
}
}
#[async_trait::async_trait]
impl<K, V> OutputType for BTreeMap<K, V>
where
K: ToString + Ord + Send + Sync,
V: Serialize + Send + Sync,
{
fn type_name() -> Cow<'static, str> {
Cow::Borrowed("JSONObject")
}
fn create_type_info(registry: &mut Registry) -> String {
registry.create_output_type::<Self, _>(|_| MetaType::Scalar {
name: <Self as OutputType>::type_name().to_string(),
description: Some("A scalar that can represent any JSON Object value."),
is_valid: |_| true,
visible: None,
specified_by_url: None,
})
}
async fn resolve(
&self,
_ctx: &ContextSelectionSet<'_>,
_field: &Positioned<Field>,
) -> ServerResult<Value> {
let mut map = IndexMap::new();
for (name, value) in self {
map.insert(
Name::new(name.to_string()),
to_value(value).unwrap_or_default(),
);
}
Ok(Value::Object(map))
}
}