diff --git a/sdk/rust/nym-sdk/src/mixnet/native_client.rs b/sdk/rust/nym-sdk/src/mixnet/native_client.rs index c9495680b2..9050ff45a1 100644 --- a/sdk/rust/nym-sdk/src/mixnet/native_client.rs +++ b/sdk/rust/nym-sdk/src/mixnet/native_client.rs @@ -25,6 +25,7 @@ use std::sync::Arc; use std::task::{Context, Poll}; use tokio::sync::RwLockReadGuard; use tokio_util::sync::CancellationToken; +use tokio_util::sync::WaitForCancellationFutureOwned; /// Client connected to the Nym mixnet. pub struct MixnetClient { @@ -273,6 +274,12 @@ impl MixnetClient { } } } + + pub fn cancelled(&self) -> WaitForCancellationFutureOwned { + self.shutdown_handle + .clone_shutdown_token() + .cancelled_owned() + } } #[derive(Clone)]