elfo-core 0.2.0-alpha.21

The core of the elfo system
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
35
36
37
38
39
40

use std::sync::Arc;

use tokio::runtime::Handle;

use crate::actor::ActorMeta;
#[cfg(feature = "unstable-stuck-detection")]
use crate::stuck_detection::StuckDetector;

pub(crate) trait RuntimeFilter: Fn(&ActorMeta) -> bool + Send + Sync + 'static {}
impl<F: Fn(&ActorMeta) -> bool + Send + Sync + 'static> RuntimeFilter for F {}

// === RuntimeManager ===

#[derive(Default, Clone)]
pub(crate) struct RuntimeManager {
    dedicated: Vec<(Arc<dyn RuntimeFilter>, Handle)>,
    #[cfg(feature = "unstable-stuck-detection")]
    stuck_detector: StuckDetector,
}

impl RuntimeManager {
    pub(crate) fn add<F: RuntimeFilter>(&mut self, filter: F, handle: Handle) {
        self.dedicated.push((Arc::new(filter), handle));
    }

    pub(crate) fn get(&self, meta: &ActorMeta) -> Handle {
        for (f, h) in &self.dedicated {
            if f(meta) {
                return h.clone();
            }
        }

        tokio::runtime::Handle::current()
    }

    #[cfg(feature = "unstable-stuck-detection")]
    pub(crate) fn stuck_detector(&self) -> StuckDetector {
        self.stuck_detector.clone()
    }
}