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 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.max_size = 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),
    }
  }
}