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

use crate::extensions::{ErrorLogger, ResolveInfo};
use crate::parser::types::Field;
use crate::{ContextSelectionSet, OutputType, PathSegment, Positioned, ServerResult, Type, Value};

/// Resolve an list by executing each of the items concurrently.
pub async fn resolve_list<'a, T: OutputType + 'a>(
    ctx: &ContextSelectionSet<'a>,
    field: &Positioned<Field>,
    iter: impl IntoIterator<Item = T>,
    len: Option<usize>,
) -> ServerResult<Value> {
    let mut futures = len.map(Vec::with_capacity).unwrap_or_default();

    for (idx, item) in iter.into_iter().enumerate() {
        let ctx_idx = ctx.with_index(idx);
        futures.push(async move {
            if ctx_idx.query_env.extensions.is_empty() {
                OutputType::resolve(&item, &ctx_idx, field)
                    .await
                    .map_err(|e| e.path(PathSegment::Index(idx)))
                    .log_error(&ctx_idx.query_env.extensions)
            } else {
                let resolve_info = ResolveInfo {
                    resolve_id: ctx_idx.resolve_id,
                    path_node: ctx_idx.path_node.as_ref().unwrap(),
                    parent_type: &Vec::<T>::type_name(),
                    return_type: &T::qualified_type_name(),
                };

                ctx_idx.query_env.extensions.resolve_start(&resolve_info);

                let res = OutputType::resolve(&item, &ctx_idx, field)
                    .await
                    .map_err(|e| e.path(PathSegment::Index(idx)))
                    .log_error(&ctx_idx.query_env.extensions)?;

                ctx_idx.query_env.extensions.resolve_end(&resolve_info);
                Ok(res)
            }
        });
    }

    Ok(Value::List(
        futures_util::future::try_join_all(futures).await?,
    ))
}