spin_factor_key_value/
runtime_config.rs

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
pub mod spin;

use std::{collections::HashMap, sync::Arc};

use crate::StoreManager;

/// Runtime configuration for all key value stores.
#[derive(Default, Clone)]
pub struct RuntimeConfig {
    /// Map of store names to store managers.
    store_managers: HashMap<String, Arc<dyn StoreManager>>,
}

impl RuntimeConfig {
    /// Adds a store manager for the store with the given label to the runtime configuration.
    ///
    /// If a store manager already exists for the given label, it will be replaced.
    pub fn add_store_manager(&mut self, label: String, store_manager: Arc<dyn StoreManager>) {
        self.store_managers.insert(label, store_manager);
    }

    /// Returns whether a store manager exists for the store with the given label.
    pub fn has_store_manager(&self, label: &str) -> bool {
        self.store_managers.contains_key(label)
    }

    /// Returns the store manager for the store with the given label.
    pub fn get_store_manager(&self, label: &str) -> Option<Arc<dyn StoreManager>> {
        self.store_managers.get(label).cloned()
    }
}

impl IntoIterator for RuntimeConfig {
    type Item = (String, Arc<dyn StoreManager>);
    type IntoIter = std::collections::hash_map::IntoIter<String, Arc<dyn StoreManager>>;

    fn into_iter(self) -> Self::IntoIter {
        self.store_managers.into_iter()
    }
}