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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
use std;
use std::collections::HashMap;
use std::net::SocketAddr;
use std::sync::{Arc};
use jsonrpc::futures::{Stream, Poll, Async, Sink, Future};
use jsonrpc::futures::sync::mpsc;
use parking_lot::Mutex;
pub type SenderChannels = Mutex<HashMap<SocketAddr, mpsc::Sender<String>>>;
pub struct PeerMessageQueue<S: Stream> {
up: S,
receiver: mpsc::Receiver<String>,
_addr: SocketAddr,
}
impl<S: Stream> PeerMessageQueue<S> {
pub fn new(
response_stream: S,
receiver: mpsc::Receiver<String>,
addr: SocketAddr,
) -> Self {
PeerMessageQueue {
up: response_stream,
receiver: receiver,
_addr: addr,
}
}
}
#[derive(Debug)]
pub enum PushMessageError {
NoSuchPeer,
Send(mpsc::SendError<String>)
}
impl From<mpsc::SendError<String>> for PushMessageError {
fn from(send_err: mpsc::SendError<String>) -> Self {
PushMessageError::Send(send_err)
}
}
pub struct Dispatcher {
channels: Arc<SenderChannels>,
}
impl Dispatcher {
pub fn new(channels: Arc<SenderChannels>) -> Self {
Dispatcher {
channels: channels,
}
}
pub fn push_message(&self, peer_addr: &SocketAddr, msg: String) -> Result<(), PushMessageError> {
let mut channels = self.channels.lock();
match channels.get_mut(peer_addr) {
Some(channel) => {
try!(channel.send(msg).wait().map_err(|e| PushMessageError::from(e)));
Ok(())
},
None => {
return Err(PushMessageError::NoSuchPeer);
}
}
}
pub fn is_connected(&self, socket_addr: &SocketAddr) -> bool {
self.channels.lock().contains_key(socket_addr)
}
pub fn peer_count(&self) -> usize {
self.channels.lock().len()
}
}
impl<S: Stream<Item=String, Error=std::io::Error>> Stream for PeerMessageQueue<S> {
type Item = String;
type Error = std::io::Error;
fn poll(&mut self) -> Poll<Option<String>, std::io::Error> {
match self.up.poll() {
Ok(Async::Ready(Some(val))) => {
return Ok(Async::Ready(Some(val)));
},
Ok(Async::Ready(None)) => {
return Ok(Async::Ready(None));
},
_ => {}
}
match self.receiver.poll() {
Ok(result) => Ok(result),
Err(send_err) => {
warn!("MPSC send error: {:?}", send_err);
Err(std::io::Error::from(std::io::ErrorKind::Other))
}
}
}
}