fix: the cpuload 100% in case the tcpstream WouldBlock (#3029)

* fix: the cpuload 100% in case the tcpstream WouldBlock

* extending the sleep on WouldBlock to all read/write

* rustfmt
This commit is contained in:
Gary Yu 2019-09-10 07:55:57 +08:00 committed by GitHub
parent 7a07541c26
commit 4faac470d4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -58,9 +58,10 @@ macro_rules! try_break {
($inner:expr) => {
match $inner {
Ok(v) => Some(v),
Err(Error::Connection(ref e))
if e.kind() == io::ErrorKind::WouldBlock || e.kind() == io::ErrorKind::TimedOut =>
{
Err(Error::Connection(ref e)) if e.kind() == io::ErrorKind::TimedOut => None,
Err(Error::Connection(ref e)) if e.kind() == io::ErrorKind::WouldBlock => {
// to avoid the heavy polling which will consume CPU 100%
thread::sleep(Duration::from_millis(10));
None
}
Err(Error::Store(_))
@ -375,7 +376,7 @@ where
})?;
let writer_thread = thread::Builder::new()
.name("peer_read".to_string())
.name("peer_write".to_string())
.spawn(move || {
let mut retry_send = Err(());
loop {
@ -394,7 +395,7 @@ where
}
debug!(
"Shutting down reader connection with {}",
"Shutting down writer connection with {}",
writer
.peer_addr()
.map(|a| a.to_string())