obeli-sk-wasm-workers 0.26.2

Internal package of obeli-sk
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

use std::{pin::Pin, sync::Arc, time::Duration};

pub trait DeadlineTracker: Send + Sync {
    fn track(&self) -> Option<Pin<Box<dyn Future<Output = ()> + Send>>>;
}

pub trait DeadlineTrackerFactory: Send + Sync {
    fn create(&self, deadline_duration: Duration) -> Arc<dyn DeadlineTracker>;
}

pub(crate) struct DeadlineTrackerTokio {
    deadline: tokio::time::Instant,
}
impl DeadlineTrackerTokio {
    pub(crate) fn new(deadline_duration: Duration) -> DeadlineTrackerTokio {
        let deadline = tokio::time::Instant::now() + deadline_duration;
        DeadlineTrackerTokio { deadline }
    }
}
impl DeadlineTracker for DeadlineTrackerTokio {
    fn track(&self) -> Option<Pin<Box<dyn Future<Output = ()> + Send>>> {
        if self.deadline <= tokio::time::Instant::now() {
            None
        } else {
            Some(Box::pin(tokio::time::sleep_until(self.deadline)))
        }
    }
}

#[derive(Clone, Copy)]
pub struct DeadlineTrackerFactoryTokio;

impl DeadlineTrackerFactory for DeadlineTrackerFactoryTokio {
    fn create(&self, deadline_duration: Duration) -> Arc<dyn DeadlineTracker> {
        Arc::new(DeadlineTrackerTokio::new(deadline_duration))
    }
}