added transfered event [CI SKIP]

This commit is contained in:
shimunn 2019-04-04 20:34:39 +02:00
parent ba247655bd
commit f3eeac404c
2 changed files with 13 additions and 0 deletions

View File

@ -62,6 +62,15 @@ pub(crate) fn gen_events(
//dbg!(other); //dbg!(other);
} }
} }
if prev.traffic != cur.traffic {
if let ((p_tx, p_rx), (c_tx, c_rx)) = (prev.traffic, cur.traffic) {
listeners.fire(Event::Tranfered {
peer: &cur,
tx: c_tx - p_tx,
rx: c_rx - p_rx,
});
}
}
} }
(None, Some(cur)) => listeners.fire(Event::Added(&cur)), (None, Some(cur)) => listeners.fire(Event::Added(&cur)),
(Some(prev), None) => listeners.fire(Event::Removed(&prev)), (Some(prev), None) => listeners.fire(Event::Removed(&prev)),

View File

@ -13,6 +13,7 @@ pub enum Event<'a> {
Connected(&'a Peer), Connected(&'a Peer),
Disconnected(&'a Peer), Disconnected(&'a Peer),
Roamed(&'a Peer, SocketAddr), Roamed(&'a Peer, SocketAddr),
Tranfered { peer: &'a Peer, tx: u64, rx: u64 },
} }
pub trait EventListener { pub trait EventListener {
@ -40,6 +41,9 @@ impl EventListener for LogListener {
previous_addr, previous_addr,
peer.endpoint.unwrap() peer.endpoint.unwrap()
), ),
Event::Tranfered { peer, tx, rx } => {
println!("{} transfered {} up, {} down", peer.key, tx, rx)
}
_ => unimplemented!(), _ => unimplemented!(),
} }
} }