yume/yumechi-no-kuni-proxy-worker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add histogram metrics

Browse source 
Signed-off-by: eternal-flame-AD <yume@yumechi.jp>
This commit is contained in:
ゆめ 2024-11-23 12:30:31 -06:00
parent 366504eec7
commit cb01d445f1
No known key found for this signature in database
1 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/main.rs
View file

@ -6,9 +6,11 @@ use axum::{
};
use clap::Parser;
use futures::TryFutureExt;
use prometheus::core::{GenericCounterVec, GenericGauge};
#[cfg(feature = "metrics")]
use prometheus::{core::AtomicU64, Opts};
use prometheus::{
core::{AtomicU64, GenericCounterVec, GenericGauge},
Opts,
};
use tokio::sync::mpsc;
use tower_service::Service;
use yumechi_no_kuni_proxy_worker::{
@ -107,16 +109,28 @@ fn main() {
)
.expect("Failed to create prometheus counter");
let histo_request_timing = prometheus::HistogramVec::new(
prometheus::HistogramOpts::new("misskey_media_proxy_request_timing", "Request timing")
.buckets(vec![0.01, 0.02, 0.05, 0.1, 0.25, 0.5, 1.0, 2.5, 5.0, 10.0]),
&["status", "postprocess"],
)
.expect("Failed to create prometheus histogram");
reg.register(Box::new(counter_requests_served.clone()))
.expect("Failed to register prometheus counter");
reg.register(Box::new(counter_requests_received.clone()))
.expect("Failed to register prometheus counter");
reg.register(Box::new(histo_request_timing.clone()))
.expect("Failed to register prometheus histogram");
router = router
.layer(middleware::from_fn(move |req, mut next: Next| {
counter_requests_received.with_label_values(&[]).inc();
let counter_requests_served = counter_requests_served.clone();
let histo_request_timing = histo_request_timing.clone();
let begin = std::time::Instant::now();
next.call(req)
.map_ok(move |res| {
let status = match res.status().as_u16() {
@ -150,6 +164,12 @@ fn main() {
counter_requests_served
.with_label_values(&[&status, post_process])
.inc();
let elapsed = begin.elapsed();
histo_request_timing
.with_label_values(&[&status, post_process])
.observe(elapsed.as_secs_f64());
res
})
.map_err(|e: Infallible| e)