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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
use crate::provider::ProviderKey;
use dashmap::DashMap;
use std::sync::Arc;
pub trait Provider: Send + Sync {}
/// Thread-safe registry that stores provider instances keyed by [`ProviderKey`].
///
/// # Examples
///
/// ```
/// use std::sync::Arc;
/// use gsm_core::registry::{Provider, ProviderRegistry};
/// use gsm_core::{Platform, ProviderKey, EnvId, TenantId};
///
/// struct MyProvider;
/// impl Provider for MyProvider {}
///
/// let registry: ProviderRegistry<MyProvider> = ProviderRegistry::default();
/// let key = ProviderKey {
/// platform: Platform::Slack,
/// env: EnvId("dev".into()),
/// tenant: TenantId("acme".into()),
/// team: None,
/// };
/// registry.put(key.clone(), Arc::new(MyProvider));
/// assert!(registry.get(&key).is_some());
/// ```
pub struct ProviderRegistry<P: Provider> {
inner: DashMap<ProviderKey, Arc<P>>,
}
impl<P: Provider> ProviderRegistry<P> {
pub fn new() -> Self {
Self {
inner: DashMap::new(),
}
}
pub fn get(&self, key: &ProviderKey) -> Option<Arc<P>> {
self.inner.get(key).map(|entry| Arc::clone(entry.value()))
}
pub fn put(&self, key: ProviderKey, provider: Arc<P>) {
self.inner.insert(key, provider);
}
}
impl<P: Provider> Default for ProviderRegistry<P> {
fn default() -> Self {
Self::new()
}
}