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 bytes::Bytes;

pub struct ResponsesStorage {
    responses: Vec<Bytes>,
}

impl Default for ResponsesStorage {
    fn default() -> Self {
        Self::new()
    }
}

impl ResponsesStorage {
    pub fn new() -> Self {
        Self {
            responses: Vec::new(),
        }
    }

    pub fn len(&self) -> usize {
        self.responses.len()
    }

    pub fn is_empty(&self) -> bool {
        self.responses.is_empty()
    }

    pub fn add_response(&mut self, response: Bytes) {
        self.responses.push(response);
    }

    pub fn estimate_final_response_size(&self) -> usize {
        let total_size: usize = self.responses.iter().map(|r| r.len()).sum();
        // Add a 20% buffer to account for JSON syntax, escaping, and other overhead.
        // I tested a bunch of numbers and it was the best from the bunch.
        (total_size as f64 * 1.2) as usize
    }
}