shiroyashik/sculptor
1
0
Fork 0
mirror of https://github.com/shiroyashik/sculptor.git synced 2025-12-06 13:01:12 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

made query metrics better
Some checks failed
Push Dev / docker (push) Has been cancelled
Details

Browse source
This commit is contained in:
Shiroyasha 2025-03-01 14:41:14 +03:00
parent b02cca8608
commit 13e2a54eb2
Signed by: shiroyashik
GPG key ID: E4953D3940D7860A
3 changed files with 24 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/api/figura/websocket/types/c2s.rs
View file

@ -92,10 +92,10 @@ impl From<C2SMessage> for Vec<u8> {
impl C2SMessage {
pub fn name(&self) -> &'static str {
match self {
C2SMessage::Token(_) => "c2s>token",
C2SMessage::Ping(_, _, _) => "c2s>ping",
C2SMessage::Sub(_) => "c2s>sub",
C2SMessage::Unsub(_) => "c2s>unsub",
C2SMessage::Token(_) => "C2S;TOKEN",
C2SMessage::Ping(_, _, _) => "C2S;PING",
C2SMessage::Sub(_) => "C2S;SUB",
C2SMessage::Unsub(_) => "C2S;UNSUB",
}
}
}

12
src/api/figura/websocket/types/s2c.rs
View file

@ -88,12 +88,12 @@ impl From<S2CMessage> for Vec<u8> {
impl S2CMessage {
pub fn name(&self) -> &'static str {
match self {
S2CMessage::Auth => "s2c>auth",
S2CMessage::Ping(_, _, _, _) => "s2c>ping",
S2CMessage::Event(_) => "s2c>event",
S2CMessage::Toast(_, _, _) => "s2c>toast",
S2CMessage::Chat(_) => "s2c>chat",
S2CMessage::Notice(_) => "s2c>notice",
S2CMessage::Auth => "S2C;AUTH",
S2CMessage::Ping(_, _, _, _) => "S2C;PING",
S2CMessage::Event(_) => "S2C;EVENT",
S2CMessage::Toast(_, _, _) => "S2C;TOAST",
S2CMessage::Chat(_) => "S2C;CHAT",
S2CMessage::Notice(_) => "S2C;NOTICE",
}
}
}

18
src/metrics.rs
View file

@ -43,8 +43,10 @@ fn create_mf(name: String, help: String, field_type: MetricType, metric: Metric)
}
pub async fn track_metrics(req: Request<Body>, next: Next) -> Result<Response<Body>, StatusCode> {
let method = req.method().to_string();
let route = http_route(&req).to_string();
let start = Instant::now();
let uri = req.uri().path().to_string();
// Call the next middleware or handler
let response = next.run(req).await;
@ -52,18 +54,26 @@ pub async fn track_metrics(req: Request<Body>, next: Next) -> Result<Response<Bo
let latency = start.elapsed().as_secs_f64();
REQUESTS
.with_label_values(&[&uri, response.status().as_str()])
.with_label_values(&[&method, &route, response.status().as_str()])
.observe(latency);
Ok(response)
}
// https://github.com/davidB/tracing-opentelemetry-instrumentation-sdk/blob/main/axum-tracing-opentelemetry/src/middleware/trace_extractor.rs#L177
#[inline]
fn http_route<B>(req: &Request<B>) -> &str {
req.extensions()
.get::<axum::extract::MatchedPath>()
.map_or_else(|| "", |mp| mp.as_str())
}
pub static REQUESTS: LazyLock<prometheus::HistogramVec> = LazyLock::new(|| {
register_histogram_vec!("sculptor_requests_count", "Number of requests", &["uri", "code"], vec![0.025, 0.250, 0.500]).unwrap()
register_histogram_vec!("sculptor_requests_count", "Number of requests", &["method", "uri", "code"], vec![0.025, 0.250, 0.500]).unwrap()
});
pub static PINGS: LazyLock<prometheus::HistogramVec> = LazyLock::new(|| {
register_histogram_vec!("sculptor_pings_count", "Number of pings", &["type"], vec![0.000003, 0.00002, 0.0002]).unwrap()
register_histogram_vec!("sculptor_pings_count", "Number of pings", &["type"], vec![0.000001, 0.00001, 0.0001]).unwrap()
});
pub static PINGS_ERROR: LazyLock<prometheus::IntCounter> = LazyLock::new(|| {