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
use crate::{ registry, ContextSelectionSet, InputValueResult, InputValueType, OutputValueType, Positioned, Result, Type, Value, }; use async_graphql_parser::query::Field; use std::borrow::Cow; impl<T: Type> Type for Vec<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() } } impl<T: InputValueType> InputValueType for Vec<T> { fn parse(value: Option<Value>) -> InputValueResult<Self> { match value.unwrap_or_default() { Value::List(values) => { let mut result = Vec::new(); for elem_value in values { result.push(InputValueType::parse(Some(elem_value))?); } Ok(result) } value => Ok(vec![InputValueType::parse(Some(value))?]), } } fn to_value(&self) -> Value { Value::List(self.iter().map(InputValueType::to_value).collect()) } } #[allow(clippy::ptr_arg)] #[async_trait::async_trait] impl<T: OutputValueType + Send + Sync> OutputValueType for Vec<T> { async fn resolve( &self, ctx: &ContextSelectionSet<'_>, field: &Positioned<Field>, ) -> Result<serde_json::Value> { let mut futures = Vec::with_capacity(self.len()); for (idx, item) in self.iter().enumerate() { let ctx_idx = ctx.with_index(idx); futures.push(async move { OutputValueType::resolve(item, &ctx_idx, field).await }); } Ok(futures::future::try_join_all(futures).await?.into()) } } impl<'a, T: Type + 'a> Type for &'a [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_trait::async_trait] impl<T: OutputValueType + Send + Sync> OutputValueType for &[T] { async fn resolve( &self, ctx: &ContextSelectionSet<'_>, field: &Positioned<Field>, ) -> Result<serde_json::Value> { let mut futures = Vec::with_capacity(self.len()); for (idx, item) in (*self).iter().enumerate() { let ctx_idx = ctx.with_index(idx); futures.push(async move { OutputValueType::resolve(item, &ctx_idx, field).await }); } Ok(futures::future::try_join_all(futures).await?.into()) } } #[cfg(test)] mod tests { use crate::Type; #[test] fn test_list_type() { assert_eq!(Vec::<i32>::type_name(), "[Int!]"); assert_eq!(Vec::<Option<i32>>::type_name(), "[Int]"); assert_eq!(Option::<Vec::<Option<i32>>>::type_name(), "[Int]"); assert_eq!(Vec::<i32>::qualified_type_name(), "[Int!]!"); assert_eq!(Vec::<Option<i32>>::qualified_type_name(), "[Int]!"); assert_eq!(Option::<Vec::<Option<i32>>>::qualified_type_name(), "[Int]"); assert_eq!(<&[i32] as Type>::qualified_type_name(), "[Int!]!"); } }