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
use serde::{Deserialize, Serialize};
use std::{fmt, net::SocketAddr};
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq, Hash)]
pub enum Peer {
Node(SocketAddr),
Client(SocketAddr),
}
impl Peer {
pub fn peer_addr(&self) -> SocketAddr {
match *self {
Self::Node(addr) | Self::Client(addr) => addr,
}
}
pub(crate) fn is_node(&self) -> bool {
match *self {
Peer::Node { .. } => true,
Peer::Client { .. } => false,
}
}
}
impl fmt::Display for Peer {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
Self::Node(addr) => write!(f, "Peer::Node({})", addr),
Self::Client(addr) => write!(f, "Peer::Client({})", addr),
}
}
}