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

//! Types for the bootstrap server
use crate::bin_types::{KitsuneBinType, KitsuneSpace};
use std::sync::Arc;

/// The number of random agent infos we want to collect from the bootstrap service when we want to
/// populate an empty local space.
/// @todo expose this to network config.
const RANDOM_LIMIT_DEFAULT: u32 = 16;

/// Struct to be encoded for the `random` op.
#[derive(serde::Deserialize, serde::Serialize, Clone)]
pub struct RandomQuery {
    /// The space to get random agents from.
    pub space: Arc<KitsuneSpace>,
    /// The maximum number of random agents to retrieve for this query.
    pub limit: RandomLimit,
}

impl Default for RandomQuery {
    fn default() -> Self {
        Self {
            // This is useless, it's here as a placeholder so that ..Default::default() syntax
            // works for limits, not because you'd actually ever want a "default" space.
            space: Arc::new(KitsuneSpace::new(vec![0; 36])),
            limit: RandomLimit::default(),
        }
    }
}

#[derive(serde::Deserialize, serde::Serialize, derive_more::From, derive_more::Into, Clone)]
/// Limit of random peers to return.
pub struct RandomLimit(pub u32);

impl Default for RandomLimit {
    fn default() -> Self {
        Self(RANDOM_LIMIT_DEFAULT)
    }
}