use std::fmt::{Debug, Formatter, Result as FmtResult};
use std::time::Duration;
use stream_reconnect::ReconnectOptions;
use super::policy::BEBIterator;
#[derive(Clone)]
pub struct RetryConfig(ReconnectOptions);
impl RetryConfig {
pub fn new<F, I, IN>(duration_generator: F) -> Self
where
F: 'static + Send + Sync + Fn() -> IN,
I: 'static + Send + Sync + Iterator<Item = Duration>,
IN: IntoIterator<IntoIter = I, Item = Duration>,
{
Self(ReconnectOptions::new().with_retries_generator(duration_generator))
}
}
impl From<RetryConfig> for ReconnectOptions {
fn from(o: RetryConfig) -> Self {
o.0
}
}
impl Debug for RetryConfig {
fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
f.debug_tuple("RetryConfig")
.field(&"<ReconnectOptions>")
.finish()
}
}
impl Default for RetryConfig {
fn default() -> Self {
Self::new(BEBIterator::default)
}
}