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

Using broadcast channels instead of notify

This commit is contained in:
godmodegalactus 2024-03-28 16:03:03 +01:00
parent 91cf06436a
commit ca3fa46139
No known key found for this signature in database
GPG Key ID: 22DA4A30887FDA3C
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cluster-endpoints/src/grpc_subscription.rs
View File

@ -275,7 +275,7 @@ pub fn create_block_processing_task(
grpc_x_token: Option<String>,
block_sx: async_channel::Sender<SubscribeUpdateBlock>,
commitment_level: CommitmentLevel,
exit_notfier: Arc<Notify>,
mut exit_notify: broadcast::Receiver<()>,
) -> AnyhowJoinHandle {
tokio::spawn(async move {
loop {
@ -293,7 +293,8 @@ pub fn create_block_processing_task(
// connect to grpc
let mut client =
connect_with_timeout_hacked(grpc_addr.clone(), grpc_x_token.clone()).await?;
let mut stream = client
let mut stream = tokio::select! {
res = client
.subscribe_once(
HashMap::new(),
Default::default(),
@ -304,8 +305,13 @@ pub fn create_block_processing_task(
Some(commitment_level),
Default::default(),
None,
)
.await?;
) => {
res?
},
_ = exit_notify.recv() => {
break;
}
};
loop {
tokio::select! {
@ -338,8 +344,8 @@ pub fn create_block_processing_task(
}
};
},
_ = exit_notfier.notified() => {
break;
_ = exit_notify.recv() => {
break 'main_loop;
}
}
}