Small fix in p2p.

This commit is contained in:
Revertron
2021-04-01 15:43:19 +02:00
parent 8d7f1b2c6b
commit 39accf7bd3
+11 -3
View File
@@ -171,9 +171,17 @@ fn handle_connection_event(context: Arc<Mutex<Context>>, peers: &mut Peers, regi
if event.is_readable() {
let data = {
let peer = peers.get_mut_peer(&event.token()).expect("Error getting peer for connection");
let mut stream = peer.get_stream();
read_message(&mut stream)
let token = event.token();
match peers.get_mut_peer(&token) {
None => {
error!("Error getting peer for connection {}", token.0);
return false;
}
Some(peer) => {
let mut stream = peer.get_stream();
read_message(&mut stream)
}
}
};
if data.is_ok() {