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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use std::fmt;
use std::hash::Hash;
use std::hash::Hasher;
use std::sync::Arc;

use hash::Blake2bHash;
use network_primitives::address::net_address::NetAddress;
use network_primitives::address::peer_address::PeerAddress;

use crate::peer_channel::PeerChannel;

#[derive(Clone, Debug)]
pub struct Peer {
    pub channel: Arc<PeerChannel>,
    pub version: u32,
    pub head_hash: Blake2bHash,
    pub time_offset: i64,
    pub user_agent: Option<String>,
}

impl Peer {
    pub fn new(channel: Arc<PeerChannel>, version: u32, head_hash: Blake2bHash, time_offset: i64, user_agent: Option<String>) -> Self {
        Peer {
            channel,
            version,
            head_hash,
            time_offset,
            user_agent
        }
    }

    pub fn peer_address(&self) -> Arc<PeerAddress> {
        // If a peer object exists, peer_address should be set.
        self.channel.address_info.peer_address().unwrap()
    }

    pub fn net_address(&self) -> Option<Arc<NetAddress>> {
        self.channel.address_info.net_address()
    }
}

impl fmt::Display for Peer {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}", self.peer_address())
    }
}

impl PartialEq for Peer {
    fn eq(&self, other: &Peer) -> bool {
        self.channel == other.channel
    }
}

impl Eq for Peer {}

impl Hash for Peer {
    fn hash<H: Hasher>(&self, state: &mut H) {
        self.channel.hash(state);
    }
}