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

//! Scheduling.

use std::future::Future;

use crate::error::Error;

/// Async/await scheduler.
pub struct Scheduler<SM>
where
    SM: ScheduleMethod,
{
    schedule_method: SM,
}

impl<SM> Scheduler<SM>
where
    SM: ScheduleMethod,
{
    pub fn new(schedule_method: SM) -> Self {
        Self { schedule_method }
    }

    pub fn schedule_method(&self) -> &SM {
        &self.schedule_method
    }
}

/// Can schedule a task from a future factory and a shared state.
pub trait ScheduleMethod: 'static {
    #[cfg(not(feature = "no-thread-safe-futures"))]
    fn schedule<T>(
        &self,
        future_factory: impl (FnOnce() -> T) + Send + 'static,
    ) -> Result<(), Error>
    where
        T: Future<Output = ()> + Send + 'static;

    #[cfg(feature = "no-thread-safe-futures")]
    fn schedule<T>(
        &self,
        future_factory: impl (FnOnce() -> T) + Send + 'static,
    ) -> Result<(), Error>
    where
        T: Future<Output = ()> + 'static;
}