docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-26 18:36:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

ban time

Browse Source
This commit is contained in:
Lev Kokotov
2022-02-08 09:28:53 -08:00
parent c27a7d30dc
commit b2c7d6bdab
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/pool.rs
View File

@@ -38,6 +38,7 @@ pub struct ConnectionPool {
round_robin: Counter, round_robin: Counter,
banlist: BanList, banlist: BanList,
healthcheck_timeout: u64, healthcheck_timeout: u64,
ban_time: i64,
} }
impl ConnectionPool { impl ConnectionPool {
@@ -74,6 +75,7 @@ impl ConnectionPool {
round_robin: Arc::new(AtomicUsize::new(0)), round_robin: Arc::new(AtomicUsize::new(0)),
banlist: Arc::new(Mutex::new(vec![HashMap::new()])), banlist: Arc::new(Mutex::new(vec![HashMap::new()])),
healthcheck_timeout: HEALTHCHECK_TIMEOUT, healthcheck_timeout: HEALTHCHECK_TIMEOUT,
ban_time: BAN_TIME,
} }
} }
@@ -133,6 +135,7 @@ impl ConnectionPool {
round_robin: Arc::new(AtomicUsize::new(0)), round_robin: Arc::new(AtomicUsize::new(0)),
banlist: Arc::new(Mutex::new(banlist)), banlist: Arc::new(Mutex::new(banlist)),
healthcheck_timeout: config.general.healthcheck_timeout, healthcheck_timeout: config.general.healthcheck_timeout,
ban_time: config.general.ban_time,
} }
} }
@@ -229,7 +232,7 @@ impl ConnectionPool {
match guard[shard].get(address) { match guard[shard].get(address) {
Some(timestamp) => { Some(timestamp) => {
let now = chrono::offset::Utc::now().naive_utc(); let now = chrono::offset::Utc::now().naive_utc();
if now.timestamp() - timestamp.timestamp() > BAN_TIME { if now.timestamp() - timestamp.timestamp() > self.ban_time {
// 1 minute // 1 minute
guard[shard].remove(address); guard[shard].remove(address);
false false