fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-30 18:21:33 +00:00
Code Issues Projects Releases Wiki Activity

Produce an error if a node is unresponsive (#814)

Browse Source
This commit is contained in:
Pierre Krieger
2019-01-03 17:45:37 +01:00
committed by GitHub
parent 9ca215b1a3
commit ba1b3dc5c7
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
protocols/ping/src/dial_handler.rs
View File

@ -38,7 +38,7 @@ use void::{Void, unreachable};
/// Protocol handler that handles pinging the remote at a regular period. /// Protocol handler that handles pinging the remote at a regular period.
/// ///
/// If the remote doesn't respond, produces `Unresponsive` and closes the connection. /// If the remote doesn't respond, produces an error that closes the connection.
pub struct PeriodicPingHandler<TSubstream> { pub struct PeriodicPingHandler<TSubstream> {
/// Configuration for the ping protocol. /// Configuration for the ping protocol.
ping_config: Ping<Instant>, ping_config: Ping<Instant>,
@ -113,9 +113,6 @@ enum OutState<TSubstream> {
/// Event produced by the periodic pinger. /// Event produced by the periodic pinger.
#[derive(Debug, Copy, Clone)] #[derive(Debug, Copy, Clone)]
pub enum OutEvent { pub enum OutEvent {
/// The node has been determined to be unresponsive.
Unresponsive,
/// Started pinging the remote. This can be used to print a diagnostic message in the logs. /// Started pinging the remote. This can be used to print a diagnostic message in the logs.
PingStart, PingStart,
@ -262,8 +259,7 @@ where
}, },
Ready => { Ready => {
self.out_state = OutState::Shutdown; self.out_state = OutState::Shutdown;
let ev = OutEvent::Unresponsive; Err(io::Error::new(io::ErrorKind::Other, "unresponsive node"))
Ok(Async::Ready(ProtocolsHandlerEvent::Custom(ev)))
}, },
}), }),
@ -297,8 +293,7 @@ where
}, },
Ready => { Ready => {
self.out_state = OutState::Shutdown; self.out_state = OutState::Shutdown;
let ev = OutEvent::Unresponsive; Err(io::Error::new(io::ErrorKind::Other, "unresponsive node"))
Ok(Async::Ready(ProtocolsHandlerEvent::Custom(ev)))
}, },
}) })
} }