use rusqlite::{Connection, params}; use smalldog::{Request, Response, ffi}; use time::{Duration, OffsetDateTime}; #[unsafe(no_mangle)] extern "C" fn cgi_handle(req: *const ffi::ModuleRequest) -> *const ffi::ModuleResponse { let mut response = Response::new(); let mut body = String::new(); let request = Request::from_mod_request(req); let db = if let Some(path) = request.header("CORGI_STATS_DB") { Connection::open(path).unwrap() } else { return make_error(500, "could not open stats database"); }; let now = OffsetDateTime::now_utc(); let fifteen_ago = now - Duration::minutes(15); let query = "SELECT count(requests.id) AS request_count, agents.agent FROM requests \ INNER JOIN agents ON requests.agent_id = agents.id \ WHERE requests.timestamp > ?1 \ GROUP BY requests.agent_id;"; let mut prepared = db.prepare(query).unwrap(); let mut agents: Vec<(usize, String)> = prepared .query_map(params![fifteen_ago], |row| Ok((row.get(0)?, row.get(1)?))) .unwrap() .map(|r| r.unwrap()) .collect(); agents.sort_by(|a, b| a.0.cmp(&b.0).reverse()); let highest_five_query = "SELECT count(requests.id) AS request_count, agents.agent FROM requests \ INNER JOIN agents ON requests.agent_id = agents.id \ GROUP BY requests.agent_id \ ORDER BY request_count DESC LIMIT 10;"; let mut prepared = db.prepare(highest_five_query).unwrap(); let highest_five: Vec<(usize, String)> = prepared .query_map(params![], |row| Ok((row.get(0)?, row.get(1)?))) .unwrap() .map(|r| r.unwrap()) .collect(); let sum_highest_five = highest_five.iter().fold(0, |acc, (count, _)| acc + count); body.push_str("
In the last fifteen minutes:");
body.push_str("total | req/m | agent\n");
for (count, agent) in &agents {
body.push_str(&format!(
"{count:<5} | {:<5.1} | {agent}\n",
*count as f32 / 15.0
));
}
body.push_str("
Highest ten all time requesters:");
body.push_str(" total | %of10 | agent\n");
for (count, agent) in highest_five {
body.push_str(&format!(
" {count:<7} | {:<5.1} | {agent}\n",
(count as f32 / sum_highest_five as f32) * 100.0
));
}
body.push_str("