kitsune2_core 0.3.0-dev.2

p2p / dht communication framework core and testing modules
Documentation
use kitsune2_api::*;
use std::collections::HashMap;
use std::sync::Arc;

#[cfg(test)]
mod test;

/// Factory for creating core gossip instances.
///
/// This factory returns stub gossip instances that do nothing.
#[derive(Debug)]
pub struct CoreGossipStubFactory;

impl CoreGossipStubFactory {
    /// Construct a new CoreGossipFactory.
    pub fn create() -> DynGossipFactory {
        Arc::new(CoreGossipStubFactory)
    }
}

impl GossipFactory for CoreGossipStubFactory {
    fn default_config(&self, _config: &mut Config) -> K2Result<()> {
        Ok(())
    }

    fn validate_config(&self, _config: &Config) -> K2Result<()> {
        Ok(())
    }

    fn create(
        &self,
        _builder: Arc<Builder>,
        _space_id: SpaceId,
        _peer_store: DynPeerStore,
        _local_agent_store: DynLocalAgentStore,
        _peer_meta_store: DynPeerMetaStore,
        _op_store: DynOpStore,
        _transport: DynTransport,
        _fetch: DynFetch,
    ) -> BoxFut<'static, K2Result<DynGossip>> {
        let out: DynGossip = Arc::new(CoreGossipStub);
        Box::pin(async move { Ok(out) })
    }
}

/// A stub gossip implementation that does nothing.
///
/// This is useful for constructing a Kitsune2 instance that does not require gossip, such as for
/// testing.
#[derive(Debug, Clone)]
pub struct CoreGossipStub;

impl Gossip for CoreGossipStub {
    fn inform_ops_stored(
        &self,
        ops: Vec<StoredOp>,
    ) -> BoxFut<'_, K2Result<()>> {
        Box::pin(async move {
            drop(ops);
            Ok(())
        })
    }

    fn get_state_summary(
        &self,
        _request: GossipStateSummaryRequest,
    ) -> BoxFut<'_, K2Result<GossipStateSummary>> {
        Box::pin(async move {
            Ok(GossipStateSummary {
                initiated_round: None,
                accepted_rounds: Vec::with_capacity(0),
                dht_summary: HashMap::with_capacity(0),
                peer_meta: HashMap::with_capacity(0),
            })
        })
    }
}

impl TxBaseHandler for CoreGossipStub {}
impl TxModuleHandler for CoreGossipStub {}