apollo-router 1.61.13

A configurable, high-performance routing runtime for Apollo Federation 🚀
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
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

use apollo_compiler::ExecutableDocument;

use crate::plugins::demand_control::DemandControlError;
use crate::plugins::demand_control::strategy::StrategyImpl;
use crate::plugins::demand_control::test::TestError;
use crate::plugins::demand_control::test::TestStage;
use crate::services::execution::Request;
use crate::services::subgraph::Response;

/// Test strategy for demand control.
/// Can be configured to fail at different stages of the request processing.
pub(crate) struct Test {
    pub(crate) stage: TestStage,
    pub(crate) error: TestError,
}

impl StrategyImpl for Test {
    fn on_execution_request(&self, request: &Request) -> Result<(), DemandControlError> {
        match self {
            Test {
                stage: TestStage::ExecutionRequest,
                error,
            } => {
                let error: DemandControlError = error.into();
                request
                    .context
                    .insert_cost_result(error.code().to_string())?;
                Err(error)
            }
            _ => Ok(()),
        }
    }

    fn on_subgraph_request(
        &self,
        request: &crate::services::subgraph::Request,
    ) -> Result<(), DemandControlError> {
        match self {
            Test {
                stage: TestStage::SubgraphRequest,
                error,
            } => {
                let error: DemandControlError = error.into();
                request
                    .context
                    .insert_cost_result(error.code().to_string())?;
                Err(error)
            }
            _ => Ok(()),
        }
    }

    fn on_subgraph_response(
        &self,
        _request: &ExecutableDocument,
        response: &Response,
    ) -> Result<(), DemandControlError> {
        match self {
            Test {
                stage: TestStage::SubgraphResponse,
                error,
            } => {
                let error: DemandControlError = error.into();
                response
                    .context
                    .insert_cost_result(error.code().to_string())?;
                Err(error)
            }
            _ => Ok(()),
        }
    }

    fn on_execution_response(
        &self,
        context: &crate::Context,
        _request: &ExecutableDocument,
        _response: &crate::graphql::Response,
    ) -> Result<(), DemandControlError> {
        match self {
            Test {
                stage: TestStage::ExecutionResponse,
                error,
            } => {
                let error: DemandControlError = error.into();
                context.insert_cost_result(error.code().to_string())?;
                Err(error)
            }
            _ => Ok(()),
        }
    }
}