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() { if event.is_readable() {
let data = { let data = {
let peer = peers.get_mut_peer(&event.token()).expect("Error getting peer for connection"); let token = event.token();
let mut stream = peer.get_stream(); match peers.get_mut_peer(&token) {
read_message(&mut stream) 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() { if data.is_ok() {