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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
use std::time::Duration;
use crate::authenticators::Authenticator;
/// Cluster configuration that holds per node TCP configs
pub struct ClusterTcpConfig<'a, A: Authenticator + Sized>(pub Vec<NodeTcpConfig<'a, A>>);
/// Single node TCP connection config.
#[derive(Clone)]
pub struct NodeTcpConfig<'a, A> {
pub addr: &'a str,
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,
}
/// Builder structure that helps to configure TCP connection for node.
pub struct NodeTcpConfigBuilder<'a, A> {
addr: &'a str,
authenticator: A,
max_size: Option<u32>,
min_idle: Option<u32>,
max_lifetime: Option<Duration>,
idle_timeout: Option<Duration>,
connection_timeout: Option<Duration>,
}
impl<'a, A: Authenticator + Sized> NodeTcpConfigBuilder<'a, A> {
const DEFAULT_MAX_SIZE: u32 = 10;
const DEFAULT_CONNECTION_TIMEOUT: Duration = Duration::from_secs(30);
/// `NodeTcpConfigBuilder` constructor function. It receivesthread::spawn(move || {
/// * node socket address as a string
/// * authenticator
pub fn new<'b>(addr: &'b str, authenticator: A) -> NodeTcpConfigBuilder<'b, A> {
NodeTcpConfigBuilder {
addr,
authenticator,
max_size: None,
min_idle: None,
max_lifetime: None,
idle_timeout: None,
connection_timeout: None,
}
}
/// Sets the maximum number of connections managed by the pool.
/// Defaults to 10.
pub fn max_size(mut self, size: u32) -> Self {
self.max_size = Some(size);
self
}
/// Sets the minimum idle connection count maintained by the pool.
/// If set, the pool will try to maintain at least this many idle
/// connections at all times, while respecting the value of `max_size`.
/// Defaults to None (equivalent to the value of `max_size`).
pub fn min_idle(mut self, min_idle: Option<u32>) -> Self {
self.min_idle = min_idle;
self
}
/// Sets the maximum lifetime of connections in the pool.
/// If set, connections will be closed after existing for at most 30 seconds beyond this duration.
/// If a connection reaches its maximum lifetime while checked out it will be closed when it is returned to the pool.
/// Defaults to 30 minutes.
pub fn max_lifetime(mut self, max_lifetime: Option<Duration>) -> Self {
self.max_lifetime = max_lifetime;
self
}
/// Sets the idle timeout used by the pool.
/// If set, connections will be closed after sitting idle for at most 30 seconds beyond this duration.
/// Defaults to 10 minutes.
pub fn idle_timeout(mut self, idle_timeout: Option<Duration>) -> Self {
self.idle_timeout = idle_timeout;
self
}
/// Sets the connection timeout used by the pool.
/// Defaults to 30 seconds.
pub fn connection_timeout(mut self, connection_timeout: Duration) -> Self {
self.connection_timeout = Some(connection_timeout);
self
}
/// Sets new authenticator.
pub fn authenticator(mut self, authenticator: A) -> Self {
self.authenticator = authenticator;
self
}
/// Finalizes building process and returns `NodeSslConfig`
pub fn build(self) -> NodeTcpConfig<'a, A> {
NodeTcpConfig {
addr: self.addr,
authenticator: self.authenticator,
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),
}
}
}