Report Query times (#166)

* Report avg and total query timing

* Report query times

* fmt
This commit is contained in:
Mostafa Abdelraouf
2022-09-15 01:21:45 -05:00
committed by GitHub
parent 4ae1bc8d32
commit f7a951745c
3 changed files with 39 additions and 3 deletions

View File

@@ -2,6 +2,7 @@
use bytes::{Buf, BufMut, BytesMut};
use log::{debug, error, info, trace};
use std::collections::HashMap;
use std::time::Instant;
use tokio::io::{split, AsyncReadExt, BufReader, ReadHalf, WriteHalf};
use tokio::net::TcpStream;
use tokio::sync::broadcast::Receiver;
@@ -994,6 +995,7 @@ where
self.send_server_message(server, message, &address, &pool)
.await?;
let query_start = Instant::now();
// Read all data the server has to offer, which can be multiple messages
// buffered in 8196 bytes chunks.
loop {
@@ -1013,7 +1015,11 @@ where
}
// Report query executed statistics.
self.stats.query(self.process_id, server.server_id());
self.stats.query(
self.process_id,
server.server_id(),
Instant::now().duration_since(query_start).as_millis(),
);
Ok(())
}