You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.3 KiB
60 lines
2.3 KiB
use diesel::connection::{Instrumentation, InstrumentationEvent};
|
|
use std::{cell::RefCell, collections::HashMap, time::Instant};
|
|
|
|
thread_local! {
|
|
static QUERY_PERF_TRACKER: RefCell<HashMap<String, Instant>> = RefCell::new(HashMap::new());
|
|
}
|
|
|
|
pub fn simple_logger() -> Option<Box<dyn Instrumentation>> {
|
|
Some(Box::new(|event: InstrumentationEvent<'_>| match event {
|
|
// TODO: Figure out where the invalid connection errors are coming from
|
|
// There seem to be some invalid errors when connecting to a SQLite database
|
|
// Until the cause of this is found and resolved, disable the Connection logging
|
|
// InstrumentationEvent::StartEstablishConnection {
|
|
// url,
|
|
// ..
|
|
// } => {
|
|
// debug!("Establishing connection: {url}")
|
|
// }
|
|
// InstrumentationEvent::FinishEstablishConnection {
|
|
// url,
|
|
// error,
|
|
// ..
|
|
// } => {
|
|
// if let Some(e) = error {
|
|
// error!("Error during establishing a connection with {url}: {e:?}")
|
|
// } else {
|
|
// debug!("Connection established: {url}")
|
|
// }
|
|
// }
|
|
InstrumentationEvent::StartQuery {
|
|
query,
|
|
..
|
|
} => {
|
|
let query_string = format!("{query:?}");
|
|
let start = Instant::now();
|
|
QUERY_PERF_TRACKER.with_borrow_mut(|map| {
|
|
map.insert(query_string, start);
|
|
});
|
|
}
|
|
InstrumentationEvent::FinishQuery {
|
|
query,
|
|
..
|
|
} => {
|
|
let query_string = format!("{query:?}");
|
|
QUERY_PERF_TRACKER.with_borrow_mut(|map| {
|
|
if let Some(start) = map.remove(&query_string) {
|
|
let duration = start.elapsed();
|
|
if duration.as_secs() >= 5 {
|
|
warn!("SLOW QUERY [{:.2}s]: {}", duration.as_secs_f32(), query_string);
|
|
} else if duration.as_secs() >= 1 {
|
|
info!("SLOW QUERY [{:.2}s]: {}", duration.as_secs_f32(), query_string);
|
|
} else {
|
|
debug!("QUERY [{:?}]: {}", duration, query_string);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
_ => {}
|
|
}))
|
|
}
|
|
|