mirror of https://github.com/KaiserY/trpl-zh-cn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
936 B
38 lines
936 B
10 months ago
|
extern crate trpl; // required for mdbook test
|
||
|
|
|
||
|
|
use std::{future::Future, time::Duration};
|
||
|
|
|
||
|
|
// ANCHOR: implementation
|
||
|
|
use trpl::Either;
|
||
|
|
|
||
|
|
// --snip--
|
||
|
|
// ANCHOR: implementation
|
||
|
|
|
||
|
|
fn main() {
|
||
|
|
trpl::run(async {
|
||
|
|
let slow = async {
|
||
|
|
trpl::sleep(Duration::from_secs(5)).await;
|
||
|
|
"Finally finished"
|
||
|
|
};
|
||
|
|
|
||
|
|
match timeout(slow, Duration::from_secs(2)).await {
|
||
|
|
Ok(message) => println!("Succeeded with '{message}'"),
|
||
|
|
Err(duration) => {
|
||
|
|
println!("Failed after {} seconds", duration.as_secs())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn timeout<F: Future>(
|
||
|
|
future_to_try: F,
|
||
|
|
max_time: Duration,
|
||
|
|
) -> Result<F::Output, Duration> {
|
||
|
|
// ANCHOR: implementation
|
||
|
|
match trpl::race(future_to_try, trpl::sleep(max_time)).await {
|
||
|
|
Either::Left(output) => Ok(output),
|
||
|
|
Either::Right(_) => Err(max_time),
|
||
|
|
}
|
||
|
|
// ANCHOR_END: implementation
|
||
|
|
}
|