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

use crate::common::Named;
use crate::scheduler::listener::Listener;
use crate::scheduler::{Scheduler, SchedulerImpl};

#[allow(missing_docs, clippy::missing_errors_doc)]
pub trait HasScheduler<'s> {
    fn scheduler(&self) -> &SchedulerImpl<'s>;

    fn scheduler_mut(&mut self) -> &mut SchedulerImpl<'s>;

    fn get_stack_size(&self) -> usize {
        self.scheduler().get_stack_size()
    }

    fn set_stack_size(&self, stack_size: usize) {
        self.scheduler().set_stack_size(stack_size);
    }

    fn try_resume(&self, co_name: &str) -> std::io::Result<()> {
        self.scheduler().try_resume(co_name)
    }

    fn try_schedule(&self) -> std::io::Result<()> {
        self.scheduler().try_schedule()
    }

    fn try_timed_schedule(&self, dur: std::time::Duration) -> std::io::Result<u64> {
        self.scheduler().try_timed_schedule(dur)
    }

    fn try_timeout_schedule(&self, timeout_time: u64) -> std::io::Result<u64> {
        self.scheduler().try_timeout_schedule(timeout_time)
    }

    fn try_get_co_result(&self, co_name: &str) -> Option<Result<Option<usize>, &'s str>> {
        self.scheduler().try_get_co_result(co_name)
    }

    fn is_empty(&self) -> bool {
        self.scheduler().is_empty()
    }

    fn size(&self) -> usize {
        self.scheduler().size()
    }

    fn add_listener(&mut self, listener: impl Listener + 's) {
        self.scheduler_mut().add_listener(listener);
    }
}

impl<'s, HasSchedulerImpl: HasScheduler<'s>> Named for HasSchedulerImpl {
    fn get_name(&self) -> &str {
        Box::leak(Box::from(self.scheduler().get_name()))
    }
}