1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
use std::ffi::CStr;
use rusqlite::{Connection, params};
use smalldog::{ModuleRequest, ModuleResponse, Request, Response};
use time::{Duration, OffsetDateTime};
#[unsafe(no_mangle)]
extern "C" fn cgi_handle(req: *const ModuleRequest) -> *const ModuleResponse {
let mut response = Response::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());
response.push_str("<p>In the last fifteen minutes:<br/><code><pre>");
response.push_str("total | req/m | agent\n");
for (count, agent) in &agents {
response.push_str(&format!(
"{count:<5} | {:<5.1} | {agent}\n",
*count as f32 / 15.0
));
}
response.push_str("</pre></code></p>");
response.into_mod_response(200)
}
fn make_error<S: AsRef<str>>(code: u16, msg: S) -> *const ModuleResponse {
unsafe {
smalldog::HEADERS[0][0] = c"Content-Length".as_ptr();
smalldog::HEADERS[0][1] = c"text/html".as_ptr();
}
let mut response = Response::new();
response.push_str(msg.as_ref());
response.into_mod_response(code)
}
#[unsafe(no_mangle)]
extern "C" fn cgi_cleanup(response: *const ModuleResponse) {
Response::cleanup(response);
}
|
