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

Add server login stat; refactor for better naming (#34)

Browse Source
This commit is contained in:
Lev Kokotov
2022-02-21 17:28:50 -08:00
committed by GitHub
parent f805b43a08
commit f21a3d8d8c
2 changed files with 95 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/pool.rs
View File

@@ -351,7 +351,12 @@ impl ManageConnection for ServerPool {
self.address, self.user.name
);
Server::startup(
// Put a temporary process_id into the stats
// for server login.
let process_id = rand::random::<i32>();
self.stats.server_login(process_id);
match Server::startup(
&self.address,
&self.user,
&self.database,
@@ -359,6 +364,18 @@ impl ManageConnection for ServerPool {
self.stats.clone(),
)
.await
{
Ok(conn) => {
// Remove the temporary process_id from the stats.
self.stats.server_disconnecting(process_id);
Ok(conn)
}
Err(err) => {
// Remove the temporary process_id from the stats.
self.stats.server_disconnecting(process_id);
Err(err)
}
}
}
/// Determines if the connection is still connected to the database.