andras/vault-link
andras/vault-link
Watch
1
0
Fork
You've already forked vault-link
 0
Code Issues Pull requests 18 Projects Releases 55 Packages Wiki Activity Actions 36

Fix pool opening race and closing stutter

Browse source
This commit is contained in:
Andras Schmelczer 2026-09-22 20:04:19 +01:00
commit 81b4a08a59
1 changed files with 90 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

28
sync-server/src/app_state/websocket/broadcasts.rs
View file

@ -1,5 +1,11 @@
use super::models::CursorPositionFromServer;
use crate::{app_state::database::models::VaultId, config::server_config::ServerConfig};
use crate::{
app_state::{
database::models::VaultId,
weak_slots::get_or_create,
},
config::server_config::ServerConfig,
};
use std::{collections::HashMap, sync::Arc};
use tokio::sync::{Mutex, broadcast};
@ -25,19 +31,13 @@ impl Broadcasts {
}
}
pub fn try_admit(&self, vault: &VaultId) -> Option<tokio::sync::OwnedSemaphorePermit> {
let mut limits = self.admission.lock().unwrap();
limits.retain(|_, limit| limit.strong_count() > 0);
let limit = limits
.get(vault)
.and_then(std::sync::Weak::upgrade)
.unwrap_or_else(|| {
let limit = Arc::new(tokio::sync::Semaphore::new(self.capacity));
limits.insert(vault.clone(), Arc::downgrade(&limit));
limit
});
pub async fn try_admit(&self, vault: &VaultId) -> Option<tokio::sync::OwnedSemaphorePermit> {
let limit = get_or_create(
async { self.admission.lock().unwrap() },
vault.clone(),
|| Arc::new(tokio::sync::Semaphore::new(self.capacity)),
)
.await;
limit.try_acquire_owned().ok()
}