diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/common.rs | 1 | ||||
| -rw-r--r-- | src/share.rs | 3 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/common.rs b/src/common.rs index 4e32edc..b888ed7 100644 --- a/src/common.rs +++ b/src/common.rs @@ -81,6 +81,7 @@ macro_rules! impl_share { } pub async fn close(self) { + self.share.bomb.lock().await.defuse(); self.share.close_tx.send(true).ok(); let mut tasks = self.share.tasks.lock().await; diff --git a/src/share.rs b/src/share.rs index ad63290..a2afc4c 100644 --- a/src/share.rs +++ b/src/share.rs @@ -1,4 +1,5 @@ use super::*; +use drop_bomb::DropBomb; use std::{borrow::Cow, collections::HashMap, io, sync::Arc, time::Duration}; use tokio::{ sync::{mpsc, watch, Mutex, RwLock}, @@ -26,6 +27,7 @@ pub(crate) struct RudpShare<S: UdpSender> { pub(crate) udp_tx: S, pub(crate) close_tx: watch::Sender<bool>, pub(crate) tasks: Mutex<JoinSet<()>>, + pub(crate) bomb: Mutex<DropBomb>, } pub async fn new<S: UdpSender, R: UdpReceiver>( @@ -51,6 +53,7 @@ pub async fn new<S: UdpSender, R: UdpReceiver>( }) .collect(), tasks: Mutex::new(JoinSet::new()), + bomb: Mutex::new(DropBomb::new("rudp connection must be explicitly closed")), }); let mut tasks = share.tasks.lock().await; |