hive-router 0.2.0

GraphQL router for Federation, part of the Hive platform
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

use ahash::AHashMap;

type SubgraphName<'a> = &'a str;
type FetchStepHash = u64;
type SubgraphCost = u64;

pub struct SubgraphResponseCostTracker<'a> {
    records: AHashMap<(SubgraphName<'a>, FetchStepHash), SubgraphCost>,
}

impl<'a> Default for SubgraphResponseCostTracker<'a> {
    fn default() -> Self {
        Self::new()
    }
}

impl<'a> SubgraphResponseCostTracker<'a> {
    pub fn new() -> Self {
        Self {
            records: AHashMap::default(),
        }
    }

    pub fn track(
        &mut self,
        subgraph: SubgraphName<'a>,
        fetch_step_hash: FetchStepHash,
        cost: SubgraphCost,
    ) {
        self.records.insert((subgraph, fetch_step_hash), cost);
    }

    pub fn total(&self) -> u64 {
        self.records
            .values()
            .fold(0, |acc, cost| acc.saturating_add(*cost))
    }
}