blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

handle channel disconnect (#24036)

This commit is contained in:
HaoranYi 2022-04-01 13:47:06 -05:00 committed by GitHub
parent df4d92f9cf
commit c9a476e24d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
rpc/src/rpc_completed_slots_service.rs
View File

@ -1,5 +1,6 @@
use {
crate::rpc_subscriptions::RpcSubscriptions,
crossbeam_channel::RecvTimeoutError,
solana_client::rpc_response::SlotUpdate,
solana_ledger::blockstore::CompletedSlotsReceiver,
solana_sdk::timing::timestamp,
@ -25,19 +26,26 @@ impl RpcCompletedSlotsService {
Builder::new()
.name("solana-rpc-completed-slots-service".to_string())
.spawn(move || loop {
// shutdown the service
// received exit signal, shutdown the service
if exit.load(Ordering::Relaxed) {
break;
}
if let Ok(slots) = completed_slots_receiver
match completed_slots_receiver
.recv_timeout(Duration::from_millis(COMPLETE_SLOT_REPORT_SLEEP_MS))
{
for slot in slots {
rpc_subscriptions.notify_slot_update(SlotUpdate::Completed {
slot,
timestamp: timestamp(),
});
Err(RecvTimeoutError::Timeout) => {}
Err(RecvTimeoutError::Disconnected) => {
info!("RpcCompletedSlotService channel disconnected, exiting.");
break;
}
Ok(slots) => {
for slot in slots {
rpc_subscriptions.notify_slot_update(SlotUpdate::Completed {
slot,
timestamp: timestamp(),
});
}
}
}
})