gnosis/lighthouse
1
0
Fork 0
mirror of https://github.com/sigp/lighthouse.git synced 2026-03-10 12:11:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve peer status handling

Browse Source
This commit is contained in:
Paul Hauner
2019-03-25 15:30:46 +11:00
parent 708d9b5674
commit ebb9ced0a4
2 changed files with 81 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
beacon_node/network/src/message_handler.rs
View File

@@ -4,7 +4,7 @@ use crate::service::{NetworkMessage, OutgoingMessage};
use crate::sync::SimpleSync;
use crossbeam_channel::{unbounded as channel, Sender};
use eth2_libp2p::{
rpc::{IncomingGossip, RPCRequest, RPCResponse},
rpc::{methods::GoodbyeReason, IncomingGossip, RPCRequest, RPCResponse},
PeerId, RPCEvent,
};
use futures::future;
@@ -199,7 +199,8 @@ impl MessageHandler {
.on_block_gossip(peer_id, message, &mut self.network_context)
}
IncomingGossip::Attestation(message) => {
//
self.sync
.on_attestation_gossip(peer_id, message, &mut self.network_context)
}
}
}
@@ -226,7 +227,8 @@ impl NetworkContext {
}
}
pub fn disconnect(&self, _peer_id: PeerId) {
pub fn disconnect(&mut self, peer_id: PeerId, reason: GoodbyeReason) {
self.send_rpc_request(peer_id, RPCRequest::Goodbye(reason))
// TODO: disconnect peers.
}