use core::time::Duration;
use std::sync::Arc;
use std::net;
use crate::authenticators::Authenticator;
pub struct ClusterRustlsConfig<A: Authenticator + Sized>(pub Vec<NodeRustlsConfig<A>>);
#[derive(Clone)]
pub struct NodeRustlsConfig<A> {
pub addr: net::SocketAddr,
pub dns_name: webpki::DNSName,
pub authenticator: A,
pub max_size: u32,
pub min_idle: Option<u32>,
pub max_lifetime: Option<Duration>,
pub idle_timeout: Option<Duration>,
pub connection_timeout: Duration,
pub config: Arc<rustls::ClientConfig>,
}
pub struct NodeRustlsConfigBuilder<A> {
addr: net::SocketAddr,
dns_name: webpki::DNSName,
authenticator: A,
max_size: Option<u32>,
min_idle: Option<u32>,
max_lifetime: Option<Duration>,
idle_timeout: Option<Duration>,
connection_timeout: Option<Duration>,
config: Arc<rustls::ClientConfig>,
}
impl<A: Authenticator + Sized> NodeRustlsConfigBuilder<A> {
const DEFAULT_MAX_SIZE: u32 = 10;
const DEFAULT_CONNECTION_TIMEOUT: Duration = Duration::from_secs(30);
pub fn new(addr: net::SocketAddr, dns_name: webpki::DNSName, authenticator: A, config: Arc<rustls::ClientConfig>) -> Self {
NodeRustlsConfigBuilder {
addr,
dns_name,
authenticator,
max_size: None,
min_idle: None,
max_lifetime: None,
idle_timeout: None,
connection_timeout: None,
config,
}
}
pub fn max_size(mut self, size: u32) -> Self {
self.max_size = Some(size);
self
}
pub fn min_idle(mut self, min_idle: Option<u32>) -> Self {
self.min_idle = min_idle;
self
}
pub fn max_lifetime(mut self, max_lifetime: Option<Duration>) -> Self {
self.max_lifetime = max_lifetime;
self
}
pub fn idle_timeout(mut self, idle_timeout: Option<Duration>) -> Self {
self.idle_timeout = idle_timeout;
self
}
pub fn connection_timeout(mut self, connection_timeout: Duration) -> Self {
self.connection_timeout = Some(connection_timeout);
self
}
pub fn authenticator(mut self, authenticator: A) -> Self {
self.authenticator = authenticator;
self
}
pub fn build(self) -> NodeRustlsConfig<A> {
NodeRustlsConfig {
addr: self.addr,
dns_name: self.dns_name,
authenticator: self.authenticator,
config: self.config,
max_size: self.max_size.unwrap_or(Self::DEFAULT_MAX_SIZE),
min_idle: self.min_idle,
max_lifetime: self.max_lifetime,
idle_timeout: self.idle_timeout,
connection_timeout: self
.connection_timeout
.unwrap_or(Self::DEFAULT_CONNECTION_TIMEOUT),
}
}
}