studio/tuwunel
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Optimize watcher range collection on notify.

Browse Source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk
2025-10-20 20:44:45 +00:00
parent 7d8f7cbe5d
commit fc0f04defa
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/database/map/watch.rs
View File

@@ -8,11 +8,12 @@ use std::{
use futures::pin_mut; use futures::pin_mut;
use serde::Serialize; use serde::Serialize;
use tokio::sync::watch::{Sender, channel}; use tokio::sync::watch::{Sender, channel};
use tuwunel_core::implement; use tuwunel_core::{implement, smallvec::SmallVec};
use crate::keyval::{KeyBuf, serialize_key}; use crate::keyval::{KeyBuf, serialize_key};
type Watchers = Mutex<BTreeMap<KeyBuf, Sender<()>>>; type Watchers = Mutex<BTreeMap<KeyBuf, Sender<()>>>;
type KeyVec = SmallVec<[KeyBuf; 1]>;
#[derive(Default)] #[derive(Default)]
pub(super) struct Watch { pub(super) struct Watch {
@@ -71,7 +72,7 @@ where
.take_while(|(k, _)| key.as_ref().starts_with(k)) .take_while(|(k, _)| key.as_ref().starts_with(k))
.filter_map(|(k, tx)| tx.send(()).is_err().then_some(k)) .filter_map(|(k, tx)| tx.send(()).is_err().then_some(k))
.cloned() .cloned()
.collect::<Vec<_>>() .collect::<KeyVec>()
.into_iter() .into_iter()
.for_each(|k| { .for_each(|k| {
watchers.remove(&k); watchers.remove(&k);