A tool that helps you retry your future :) Well, Futures and Streams, to be precise.
It's quite a common task when you need to repeat some action if you've got an error, be it a connection timeout or some temporary OS error.
When you do stuff in a synchronous manner it's quite easy to implement the attempts logic, but when it comes to asynchronous programming you suddenly need to write a fool load of a boilerplate code, introduce state machines and everything.
This library aims to make your life easier and let you write more straightword and nice code, concentrating on buisness logic rathen than on handling all the mess.
I was inspired to write this library after coming over a hyper
issue, and I came to an understanding that the
problem is more common than I used to think.
For examples have a look in the examples/ folder in the git repo.
Suggestions and critiques are welcome!
// ...
use futures_retry::{RetryPolicy, StreamRetryExt};
// In this example we use a free function to handle errors, while in your project you have
// more options: for simple cases a simple closure will do, for complex cases you might go
// as far as implementing an `ErrorHandler` trait for a custom type with some complex logic.
fn handle_error(e: io::Error) -> RetryPolicy<io::Error> {
match e.kind() {
io::ErrorKind::Interrupted => RetryPolicy::Repeat,
io::ErrorKind::PermissionDenied => RetryPolicy::ForwardError(e),
_ => RetryPolicy::WaitRetry(Duration::from_millis(5)),
}
}
async fn serve_connection(stream: TcpStream) {
// ...
}
#[tokio::main]
async fn main() {
let addr = //...
# "127.0.0.1:12345";
let mut listener = TcpListener::bind(addr).await.unwrap();
let server = listener.incoming()
.retry(handle_error) // Magic happens here
.and_then(|(stream, _attempt)| {
tokio::spawn(serve_connection(stream));
ok(())
})
.try_for_each(|_| ok(()))
.map_err(|(e, _attempt)| eprintln!("Caught an error {}", e));
# // This nasty hack is required to exit immediately when running the doc tests.
# let server = select(ok::<_, ()>(()), server).map(|_| ());
server.await
}Licensed under either of
- Apache License, Version 2.0 (LICENSE-APACHE or http://www.apache.org/licenses/LICENSE-2.0)
- MIT license (LICENSE-MIT or http://opensource.org/licenses/MIT)
at your option.
Unless you explicitly state otherwise, any contribution intentionally submitted for inclusion in the work by you, as defined in the Apache-2.0 license, shall be dual licensed as above, without any additional terms or conditions.
License: MIT/Apache-2.0