axum-cometd 0.7.0-beta.2

Framework for CometD server creation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use crate::{types::ChannelId, utils::get_wild_names};
use ahash::{AHashMap, AHashSet};
use std::sync::Arc;
use tokio::sync::RwLock;

#[derive(Debug, Default)]
pub(crate) struct WildNamesCache {
    // TODO: Replace on Arc<[String]>
    cache: RwLock<AHashMap<ChannelId, Arc<Vec<String>>>>,
}

impl WildNamesCache {
    pub(crate) async fn fetch_wildnames(&self, name: &str) -> Arc<Vec<ChannelId>> {
        let read_guard = self.cache.read().await;
        if let Some(wildnames) = read_guard.get(name) {
            wildnames.clone()
        } else {
            drop(read_guard);
            let mut write_guard = self.cache.write().await;
            write_guard
                .entry(name.to_string())
                .or_insert_with(|| Arc::new(get_wild_names(name)))
                .clone()
        }
    }

    #[inline(always)]
    pub(crate) async fn remove_wildnames(&self, mut names: AHashSet<ChannelId>) {
        self.cache
            .write()
            .await
            .retain(|name, _| names.remove(name));
    }
}