From eed5fdb54cc261756c828d9c6bd1114e16ec6aa8 Mon Sep 17 00:00:00 2001 From: hozan23 Date: Sat, 22 Jun 2024 15:53:22 +0200 Subject: jsonrpc/client: close the subscription channel when calling unsubscribe --- jsonrpc/src/client/subscriptions.rs | 4 +++- jsonrpc/src/server/mod.rs | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) (limited to 'jsonrpc/src') diff --git a/jsonrpc/src/client/subscriptions.rs b/jsonrpc/src/client/subscriptions.rs index fe66f96..3583b33 100644 --- a/jsonrpc/src/client/subscriptions.rs +++ b/jsonrpc/src/client/subscriptions.rs @@ -78,7 +78,9 @@ impl Subscriptions { /// Unsubscribe from the provided subscription id. pub(super) async fn unsubscribe(&self, id: &SubscriptionID) { - self.subs.lock().await.remove(id); + if let Some(sub) = self.subs.lock().await.remove(id) { + sub.close(); + } } /// Notifies the subscription about the given notification. diff --git a/jsonrpc/src/server/mod.rs b/jsonrpc/src/server/mod.rs index ddebeb9..bde351e 100644 --- a/jsonrpc/src/server/mod.rs +++ b/jsonrpc/src/server/mod.rs @@ -83,11 +83,11 @@ impl Server { match selfc.listener.accept().await { Ok(conn) => { if let Err(err) = selfc.handle_conn(conn).await { - error!("Failed to handle a new conn: {err}") + error!("Handle a new connection: {err}") } } Err(err) => { - error!("Failed to accept a new conn: {err}") + error!("Accept a new connection: {err}") } } } @@ -249,7 +249,7 @@ impl Server { trace!("--> new request {msg}"); let on_complete = |result: TaskResult>| async move { if let TaskResult::Completed(Err(err)) = result { - error!("Failed to handle a request: {err}"); + error!("Handle a new request: {err}"); } }; let selfc = self.clone(); -- cgit v1.2.3