Core/grin
0
0
Fork 0
mirror of https://github.com/mimblewimble/grin.git synced 2025-05-05 00:31:14 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

optimization of peer state write lock ()

Browse source 
* optimization of peer state write lock
This commit is contained in:
Gary Yu 2018-10-07 13:01:32 +08:00 • committed by GitHub
parent 296e912a1f
commit 463567b19e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 8 deletions

36
p2p/src/peer.rs
View file

@ -326,12 +326,19 @@ impl Peer {
}
fn check_connection(&self) -> bool {
let mut state = self.state.write().unwrap();
match self.connection.as_ref().unwrap().error_channel.try_recv() {
Ok(Error::Serialization(e)) => {
if State::Banned != *state {
*state = State::Disconnected;
info!(
let need_stop = {
let mut state = self.state.write().unwrap();
if State::Banned != *state {
*state = State::Disconnected;
true
} else {
false
}
};
if need_stop {
debug!(
LOGGER,
"Client {} corrupted, will disconnect ({:?}).", self.info.addr, e
);
@ -340,12 +347,25 @@ impl Peer {
false
}
Ok(e) => {
*state = State::Disconnected;
debug!(LOGGER, "Client {} connection lost: {:?}", self.info.addr, e);
self.stop();
let need_stop = {
let mut state = self.state.write().unwrap();
if State::Disconnected != *state {
*state = State::Disconnected;
true
} else {
false
}
};
if need_stop {
debug!(LOGGER, "Client {} connection lost: {:?}", self.info.addr, e);
self.stop();
}
false
}
Err(_) => State::Connected == *state,
Err(_) => {
let state = self.state.read().unwrap();
State::Connected == *state
}
}
}
}