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
|
//! A `tokio` backend.
use futures::Future;
/// A `tokio` executor.
pub type Executor = tokio::runtime::Runtime;
impl crate::Executor for Executor {
fn new() -> Result<Self, futures::io::Error> {
tokio::runtime::Runtime::new()
}
#[allow(clippy::let_underscore_future)]
fn spawn(&self, future: impl Future<Output = ()> + Send + 'static) {
let _ = tokio::runtime::Runtime::spawn(self, future);
}
fn enter<R>(&self, f: impl FnOnce() -> R) -> R {
let _guard = tokio::runtime::Runtime::enter(self);
f()
}
}
pub mod time {
//! Listen and react to time.
use crate::MaybeSend;
use crate::core::time::{Duration, Instant};
use crate::subscription::Subscription;
use futures::stream;
use std::future::Future;
/// Returns a [`Subscription`] that produces messages at a set interval.
///
/// The first message is produced after a `duration`, and then continues to
/// produce more messages every `duration` after that.
pub fn every(duration: Duration) -> Subscription<Instant> {
Subscription::run_with(duration, |duration| {
use futures::stream::StreamExt;
let start = tokio::time::Instant::now() + *duration;
let mut interval = tokio::time::interval_at(start, *duration);
interval.set_missed_tick_behavior(
tokio::time::MissedTickBehavior::Skip,
);
let stream = {
futures::stream::unfold(interval, |mut interval| async move {
Some((interval.tick().await, interval))
})
};
stream.map(tokio::time::Instant::into_std).boxed()
})
}
/// Returns a [`Subscription`] that runs the given async function at a
/// set interval; producing the result of the function as output.
pub fn repeat<F, T>(f: fn() -> F, interval: Duration) -> Subscription<T>
where
F: Future<Output = T> + MaybeSend + 'static,
T: MaybeSend + 'static,
{
Subscription::run_with((f, interval), |(f, interval)| {
let f = *f;
let interval = *interval;
stream::unfold(0, move |i| async move {
if i > 0 {
tokio::time::sleep(interval).await;
}
Some((f().await, i + 1))
})
})
}
}
|
