gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix resource hogging when waiting for role transition

Browse Source
This commit is contained in:
Sagar Dhawan
2019-02-06 15:31:41 -08:00
committed by Michael Vines
parent d9e2b94d7a
commit 9681c4d468
1 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/fullnode.rs
View File

@ -403,28 +403,25 @@ impl Fullnode {
}
fn handle_role_transition(&mut self) -> Option<(FullnodeReturnType, u64)> {
loop {
if self.exit.load(Ordering::Relaxed) {
return None;
if self.node_services.tpu.is_leader() {
let should_be_forwarder = self.role_notifiers.1.recv();
match should_be_forwarder {
Ok(TpuReturnType::LeaderRotation(tick_height)) => {
Some((self.leader_to_validator(tick_height), tick_height + 1))
}
Err(_) => None,
}
let should_be_forwarder = self.role_notifiers.1.try_recv();
let should_be_leader = self.role_notifiers.0.try_recv();
} else {
let should_be_leader = self.role_notifiers.0.recv();
match should_be_leader {
Ok(TvuReturnType::LeaderRotation(tick_height, entry_height, last_entry_id)) => {
self.validator_to_leader(tick_height, entry_height, last_entry_id);
return Some((
Some((
FullnodeReturnType::ValidatorToLeaderRotation,
tick_height + 1,
));
))
}
_ => match should_be_forwarder {
Ok(TpuReturnType::LeaderRotation(tick_height)) => {
return Some((self.leader_to_validator(tick_height), tick_height + 1))
}
_ => {
continue;
}
},
Err(_) => None,
}
}
}
@ -438,6 +435,12 @@ impl Fullnode {
let (sender, receiver) = channel();
let exit = self.exit.clone();
spawn(move || loop {
if self.exit.load(Ordering::Relaxed) {
debug!("node shutdown requested");
self.close().expect("Unable to close node");
sender.send(true).expect("Unable to signal exit");
break;
}
let status = self.handle_role_transition();
match status {
None => {