bee-network 0.2.2

Networking functionality and types for nodes and clients participating in the IOTA protocol built on top of `libp2p`.
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

// Copyright 2020-2021 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

use super::id::IotaGossipIdentifier;

use futures::{future, AsyncRead, AsyncWrite};
use libp2p::{core::UpgradeInfo, InboundUpgrade, OutboundUpgrade};
use log::*;

use std::{io, iter};

#[derive(Debug, Clone)]
pub struct IotaGossipProtocolUpgrade {
    id: IotaGossipIdentifier,
}

impl IotaGossipProtocolUpgrade {
    pub fn new(id: IotaGossipIdentifier) -> Self {
        Self { id }
    }
}

impl UpgradeInfo for IotaGossipProtocolUpgrade {
    type Info = IotaGossipIdentifier;
    type InfoIter = iter::Once<Self::Info>;

    fn protocol_info(&self) -> Self::InfoIter {
        trace!("gossip upgrade: protocol info query: {}", self.id);

        iter::once(self.id.clone())
    }
}

impl<S> InboundUpgrade<S> for IotaGossipProtocolUpgrade
where
    S: AsyncWrite + AsyncWrite + Unpin + Send,
{
    type Output = S;
    type Error = io::Error;
    type Future = future::Ready<Result<Self::Output, Self::Error>>;

    fn upgrade_inbound(self, stream: S, info: Self::Info) -> Self::Future {
        debug!("gossip upgrade: inbound: {}", info);

        future::ok(stream)
    }
}

impl<S> OutboundUpgrade<S> for IotaGossipProtocolUpgrade
where
    S: AsyncRead + AsyncWrite + Unpin + Send,
{
    type Output = S;
    type Error = io::Error;
    type Future = future::Ready<Result<Self::Output, Self::Error>>;

    fn upgrade_outbound(self, stream: S, info: Self::Info) -> Self::Future {
        debug!("gossip upgrade: outbound: {}", info);

        future::ok(stream)
    }
}