docker/pgcat
1
0
Fork 0
mirror of https://github.com/postgresml/pgcat.git synced 2026-03-27 10:46:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Reduce traffic on the stats channel (#69)

Browse Source
This commit is contained in:
Lev Kokotov
2022-05-17 13:05:25 -07:00
committed by GitHub
parent 54699222f8
commit fe32b5ef17
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pool.rs
View File

@@ -209,23 +209,22 @@ impl ConnectionPool {
let index = self.round_robin % addresses.len(); let index = self.round_robin % addresses.len();
let address = &addresses[index]; let address = &addresses[index];
self.stats.client_waiting(process_id, address.id);
// Make sure you're getting a primary or a replica // Make sure you're getting a primary or a replica
// as per request. If no specific role is requested, the first // as per request. If no specific role is requested, the first
// available will be chosen. // available will be chosen.
if address.role != role { if address.role != role {
self.stats.client_disconnecting(process_id, address.id);
continue; continue;
} }
allowed_attempts -= 1; allowed_attempts -= 1;
if self.is_banned(address, shard, role) { if self.is_banned(address, shard, role) {
self.stats.client_disconnecting(process_id, address.id);
continue; continue;
} }
// Indicate we're waiting on a server connection from a pool.
self.stats.client_waiting(process_id, address.id);
// Check if we can connect // Check if we can connect
let mut conn = match self.databases[shard][index].get().await { let mut conn = match self.databases[shard][index].get().await {
Ok(conn) => conn, Ok(conn) => conn,