shimun/wireguard-user
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added transfered event [CI SKIP]

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
wg-event-gen/src/gen.rs
View File

@ -62,6 +62,15 @@ pub(crate) fn gen_events(
//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)),
(Some(prev), None) => listeners.fire(Event::Removed(&prev)),

4
wg-event-gen/src/listener.rs
View File

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