nimiq-network 0.2.0

Nimiq's network implementation in Rust
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

#[cfg(feature = "metrics")]
use std::sync::Arc;

use network_primitives::address::net_address::NetAddress;

#[cfg(feature = "metrics")]
use crate::network_metrics::NetworkMetrics;
use crate::websocket::NimiqMessageStream;

/// This struct stores public information about the stream.
#[derive(Clone, Debug)]
pub struct PublicStreamInfo {
    // Constant info.
    pub net_address: NetAddress,
    pub outbound: bool,

    #[cfg(feature = "metrics")]
    pub network_metrics: Arc<NetworkMetrics>,
}

impl<'a> From<&'a PublicStreamInfo> for PublicStreamInfo {
    fn from(state: &'a PublicStreamInfo) -> Self {
        state.clone()
    }
}

impl<'a> From<&'a NimiqMessageStream> for PublicStreamInfo {
    fn from(stream: &'a NimiqMessageStream) -> Self {
        stream.public_state.clone()
    }
}

impl PublicStreamInfo {
    pub fn new(net_address: NetAddress, outbound: bool) -> Self {
        PublicStreamInfo {
            net_address,
            outbound,

            #[cfg(feature = "metrics")]
            network_metrics: Arc::new(NetworkMetrics::default()),
        }
    }
}