graphql-composition 0.12.2

An implementation of GraphQL federated schema composition
Documentation 
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

use super::Context;
use crate::subgraphs::*;

/// See [crate::subgraphs::Keys::nested_key_fields].
pub(super) fn ingest_nested_key_fields(ctx: &mut Context<'_>) {
    let subgraph_id = ctx.subgraph_id;
    ctx.subgraphs.with_nested_key_fields(|subgraphs, nested_key_fields| {
        for definition in subgraph_id.definitions(subgraphs) {
            for key in definition.id.keys(subgraphs) {
                for selection in &key.selection_set {
                    let Selection::Field(field) = selection else { continue };

                    let Some(field_type) = definition.id.field_by_name(subgraphs, field.field).and_then(|field| {
                        subgraphs.definition_by_name_id(field.r#type.definition_name_id, subgraph_id)
                    }) else {
                        continue;
                    };

                    for subselection_field in &field.subselection {
                        ingest_nested_key_fields_rec(subgraphs, field_type, subselection_field, nested_key_fields);
                    }
                }
            }
        }
    });
}

fn ingest_nested_key_fields_rec(
    subgraphs: &Subgraphs,
    parent_definition: DefinitionId,
    selection: &Selection,
    nested_key_fields: &mut NestedKeyFields,
) {
    let Selection::Field(FieldSelection {
        field,
        arguments: _,
        subselection,
        has_directives: _,
    }) = selection
    else {
        return;
    };

    let Some(field) = parent_definition.field_by_name(subgraphs, *field) else {
        return;
    };

    nested_key_fields.insert(field.record);

    if subselection.is_empty() {
        return;
    }

    let Some(selection_field_type) = subgraphs.definition_by_name_id(
        field.r#type.definition_name_id,
        subgraphs.at(parent_definition).subgraph_id,
    ) else {
        return;
    };

    for subselection in subselection {
        ingest_nested_key_fields_rec(subgraphs, selection_field_type, subselection, nested_key_fields);
    }
}