roundabout 0.2.0

An message oriented concurrent runtime
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

pub mod prelude;

mod handler;
mod message;
mod schedule;
mod util;
mod wait;

use crate::handler::{
    InitMessageHandlerBuilder, MessageGroup, MessageGroupBuilder, OpenMessageHandlerBuilder,
};
use crate::message::MessageRegistry;
use crate::schedule::{MessageScheduler, ShutdownSwitch};
use crate::wait::WaitingStrategy;
use std::sync::Arc;
use std::time::Duration;

pub struct RuntimeBuilder {
    cpu_affinity: bool,
    min_bus_capacity: usize,
    shutdown_timeout: Duration,
    waiting_strategy: WaitingStrategy,
    registry: MessageRegistry,
    // last group runs on the main thread
    groups: Vec<MessageGroup>,
}

impl RuntimeBuilder {
    pub fn with_cpu_affinity(mut self, cpu_affinity: bool) -> Self {
        self.cpu_affinity = cpu_affinity;
        self
    }

    pub fn with_waiting_strategy(mut self, waiting_strategy: WaitingStrategy) -> Self {
        self.waiting_strategy = waiting_strategy;
        self
    }

    pub fn add<T: 'static, FH>(self, handler: FH) -> Self
    where
        FH: FnOnce(OpenMessageHandlerBuilder<T>) -> InitMessageHandlerBuilder<T>,
    {
        self.add_group(|mut group| {
            let hb = group.register(handler);

            group.init(move |recv, mut context| {
                let mut h = hb.finish(&context).unwrap();
                recv.stream(move |message| {
                    h.handle(&mut context, message);
                });
            })
        })
    }

    pub fn add_group<F>(mut self, group: F) -> Self
    where
        F: FnOnce(MessageGroupBuilder) -> MessageGroup,
    {
        let group = group(MessageGroupBuilder::new(&mut self.registry));
        self.groups.push(group);
        self
    }

    pub fn finish_main<T: 'static, FH>(self, handler: FH) -> Runtime
    where
        FH: FnOnce(OpenMessageHandlerBuilder<T>) -> InitMessageHandlerBuilder<T>,
    {
        self.add(handler).finish()
    }

    pub fn finish_main_group<F>(self, group: F) -> Runtime
    where
        F: FnOnce(MessageGroupBuilder) -> MessageGroup,
    {
        self.add_group(group).finish()
    }

    pub fn finish(self) -> Runtime {
        let registry = Arc::new(self.registry);
        let scheduler = MessageScheduler::new(
            registry,
            self.groups,
            self.cpu_affinity,
            self.min_bus_capacity,
            self.shutdown_timeout,
            self.waiting_strategy,
        );

        Runtime { scheduler }
    }
}

pub struct Runtime {
    scheduler: MessageScheduler,
}

impl Runtime {
    pub fn builder(min_bus_capacity: usize) -> RuntimeBuilder {
        RuntimeBuilder {
            cpu_affinity: false,
            min_bus_capacity,
            shutdown_timeout: Duration::from_secs(10),
            waiting_strategy: Default::default(),
            registry: MessageRegistry::default(),
            groups: Default::default(),
        }
    }

    pub fn get_shutdown_switch(&self) -> ShutdownSwitch {
        self.scheduler.get_shutdown_switch()
    }

    pub fn start<E: 'static + Send + Sync>(self, init: E) {
        log::info!("start runtime");
        self.scheduler.schedule(init);
    }
}