77 lines
2.1 KiB
Rust
77 lines
2.1 KiB
Rust
|
use std::borrow::Cow;
|
||
|
use std::cmp::Eq;
|
||
|
use std::hash::Hash;
|
||
|
|
||
|
use hashbrown::HashSet;
|
||
|
|
||
|
use crate::parser::types::Field;
|
||
|
use crate::resolver_utils::resolve_list;
|
||
|
use crate::{
|
||
|
registry, ContextSelectionSet, InputType, InputValueError, InputValueResult, OutputType,
|
||
|
Positioned, Result, ServerResult, Value,
|
||
|
};
|
||
|
|
||
|
impl<T: InputType + Hash + Eq> InputType for HashSet<T> {
|
||
|
type RawValueType = Self;
|
||
|
|
||
|
fn type_name() -> Cow<'static, str> {
|
||
|
Cow::Owned(format!("[{}]", T::qualified_type_name()))
|
||
|
}
|
||
|
|
||
|
fn qualified_type_name() -> String {
|
||
|
format!("[{}]!", T::qualified_type_name())
|
||
|
}
|
||
|
|
||
|
fn create_type_info(registry: &mut registry::Registry) -> String {
|
||
|
T::create_type_info(registry);
|
||
|
Self::qualified_type_name()
|
||
|
}
|
||
|
|
||
|
fn parse(value: Option<Value>) -> InputValueResult<Self> {
|
||
|
match value.unwrap_or_default() {
|
||
|
Value::List(values) => values
|
||
|
.into_iter()
|
||
|
.map(|value| InputType::parse(Some(value)))
|
||
|
.collect::<Result<_, _>>()
|
||
|
.map_err(InputValueError::propagate),
|
||
|
value => Ok({
|
||
|
let mut result = Self::default();
|
||
|
result.insert(InputType::parse(Some(value)).map_err(InputValueError::propagate)?);
|
||
|
result
|
||
|
}),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn to_value(&self) -> Value {
|
||
|
Value::List(self.iter().map(InputType::to_value).collect())
|
||
|
}
|
||
|
|
||
|
fn as_raw_value(&self) -> Option<&Self::RawValueType> {
|
||
|
Some(self)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[async_trait::async_trait]
|
||
|
impl<T: OutputType + Hash + Eq> OutputType for HashSet<T> {
|
||
|
fn type_name() -> Cow<'static, str> {
|
||
|
Cow::Owned(format!("[{}]", T::qualified_type_name()))
|
||
|
}
|
||
|
|
||
|
fn qualified_type_name() -> String {
|
||
|
format!("[{}]!", T::qualified_type_name())
|
||
|
}
|
||
|
|
||
|
fn create_type_info(registry: &mut registry::Registry) -> String {
|
||
|
T::create_type_info(registry);
|
||
|
Self::qualified_type_name()
|
||
|
}
|
||
|
|
||
|
async fn resolve(
|
||
|
&self,
|
||
|
ctx: &ContextSelectionSet<'_>,
|
||
|
field: &Positioned<Field>,
|
||
|
) -> ServerResult<Value> {
|
||
|
resolve_list(ctx, field, self, Some(self.len())).await
|
||
|
}
|
||
|
}
|