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

Fix client states reporting (#27)

Browse Source 
* Fix client states reporting

* clean
This commit is contained in:
Lev Kokotov
2022-02-20 12:40:09 -08:00
committed by GitHub
parent d4c1fc87ee
commit 3b795464a8
3 changed files with 70 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/client.rs
View File

@@ -192,6 +192,9 @@ impl Client {
// We expect the client to either start a transaction with regular queries
// or issue commands for our sharding and server selection protocols.
loop {
// Client idle, waiting for messages.
self.stats.client_idle(self.process_id);
// Read a complete message from the client, which normally would be
// either a `Q` (query) or `P` (prepare, extended protocol).
// We can parse it here before grabbing a server from the pool,
@@ -243,6 +246,9 @@ impl Client {
continue;
}
// Waiting for server connection.
self.stats.client_waiting(self.process_id);
// Grab a server from the pool: the client issued a regular query.
let connection = match pool.get(query_router.shard(), query_router.role()).await {
Ok(conn) => conn,
@@ -261,6 +267,9 @@ impl Client {
// Claim this server as mine for query cancellation.
server.claim(self.process_id, self.secret_key);
// Client active
self.stats.client_active(self.process_id);
// Transaction loop. Multiple queries can be issued by the client here.
// The connection belongs to the client until the transaction is over,
// or until the client disconnects if we are in session mode.
@@ -330,7 +339,6 @@ impl Client {
// If we are in session mode, we keep the server until the client disconnects.
if self.transaction_mode {
// Report this client as idle.
self.stats.client_idle();
break;
}
}
@@ -412,7 +420,6 @@ impl Client {
self.stats.transaction();
if self.transaction_mode {
self.stats.client_idle();
break;
}
}
@@ -446,7 +453,6 @@ impl Client {
self.stats.transaction();
if self.transaction_mode {
self.stats.client_idle();
break;
}
}
@@ -471,3 +477,9 @@ impl Client {
guard.remove(&(self.process_id, self.secret_key));
}
}
impl Drop for Client {
fn drop(&mut self) {
self.stats.client_disconnecting(self.process_id);
}
}