pezkuwichain/pezkuwi-telemetry
1
0
Fork 0
mirror of https://github.com/pezkuwichain/pezkuwi-telemetry.git synced 2026-06-13 19:41:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose metrics in a format that prometheus understands

Browse Source
This commit is contained in:
James Wilson
2021-08-11 15:58:48 +01:00
parent 4f7b2c8ec5
commit 92da674d4d
4 changed files with 141 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/telemetry_core/src/main.rs
+27 -4
View File
@@ -178,7 +178,11 @@ async fn start_server(num_aggregators: usize, opts: Opts) -> anyhow::Result<()>
let _ = ws_send.close().await;
},
))
}
},
// Return metrics in a prometheus-friendly text based format:
(&Method::GET, "/metrics") => {
Ok(return_prometheus_metrics(aggregator).await)
},
// 404 for anything else:
_ => Ok(Response::builder()
.status(404)
@@ -427,9 +431,7 @@ where
Some(msgs) => msgs,
None => break,
};
if _feed_id == 1 {
println!("FEED 1 message len: {}", msgs.len());
}
// There is only one message type at the mo; bytes to send
// to the websocket. collect them all up to dispatch in one shot.
let all_msg_bytes = msgs.into_iter().map(|msg| match msg {
@@ -481,3 +483,24 @@ if _feed_id == 1 {
// loop ended; give socket back to parent:
(tx_to_aggregator, ws_send)
}
async fn return_prometheus_metrics(aggregator: AggregatorSet) -> Response<hyper::Body> {
let metrics = aggregator.latest_metrics();
let mut s = String::new();
for (idx, m) in metrics.iter().enumerate() {
s.push_str(&format!("telemetry_connected_feeds{{aggregator=\"{}\"}} {} {}\n", idx, m.connected_feeds, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_connected_nodes{{aggregator=\"{}\"}} {} {}\n", idx, m.connected_nodes, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_connected_shards{{aggregator=\"{}\"}} {} {}\n", idx, m.connected_shards, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_chains_subscribed_to{{aggregator=\"{}\"}} {} {}\n", idx, m.chains_subscribed_to, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_subscribed_feeds{{aggregator=\"{}\"}} {} {}\n", idx, m.subscribed_feeds, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_subscribed_finality_feeds{{aggregator=\"{}\"}} {} {}\n", idx, m.subscribed_finality_feeds, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_total_messages_to_feeds{{aggregator=\"{}\"}} {} {}\n", idx, m.total_messages_to_feeds, m.timestamp_unix_ms));
s.push_str(&format!("telemetry_total_messages_to_aggregator{{aggregator=\"{}\"}} {} {}\n\n", idx, m.total_messages_to_aggregator, m.timestamp_unix_ms));
}
Response::builder()
.header(http::header::CONTENT_TYPE, "text/plain; version=0.0.4")
.body(s.into())
.unwrap()
}