YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ws: prometheus: store ip version (v4/v6) for connections and messages

Browse source
This commit is contained in:
Joakim Frostegård 2023-01-17 21:06:34 +01:00
parent 8b7c3c481c
commit 32253a37df
1 changed files with 49 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

63
aquatic_ws/src/workers/socket.rs
View file

@ -153,7 +153,11 @@ pub async fn run_socket_worker(
let task_handle = spawn_local_into(enclose!((config, access_list, control_message_senders, in_message_senders, connection_slab, opt_tls_config) async move {
#[cfg(feature = "metrics")]
::metrics::increment_gauge!("aquatic_active_connections", 1.0);
::metrics::increment_gauge!(
"aquatic_active_connections",
1.0,
"ip_version" => ip_version_to_metrics_str(ip_version)
);
if let Err(err) = run_connection(
config.clone(),
@ -177,7 +181,11 @@ pub async fn run_socket_worker(
// Clean up after closed connection
#[cfg(feature = "metrics")]
::metrics::decrement_gauge!("aquatic_active_connections", 1.0);
::metrics::decrement_gauge!(
"aquatic_active_connections",
1.0,
"ip_version" => ip_version_to_metrics_str(ip_version)
);
// Remove reference in separate statement to avoid
// multiple RefCell borrows
@ -425,6 +433,7 @@ async fn run_stream_agnostic_connection<
pending_scrape_slab,
connection_id,
server_start_instant,
ip_version,
};
let result = writer.run_out_message_loop().await;
@ -508,7 +517,11 @@ impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionReader<S> {
match in_message {
InMessage::AnnounceRequest(announce_request) => {
#[cfg(feature = "metrics")]
::metrics::increment_counter!("aquatic_requests_total", "type" => "announce");
::metrics::increment_counter!(
"aquatic_requests_total",
"type" => "announce",
"ip_version" => ip_version_to_metrics_str(self.ip_version)
);
let info_hash = announce_request.info_hash;
@ -581,7 +594,11 @@ impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionReader<S> {
}
InMessage::ScrapeRequest(ScrapeRequest { info_hashes, .. }) => {
#[cfg(feature = "metrics")]
::metrics::increment_counter!("aquatic_requests_total", "type" => "scrape");
::metrics::increment_counter!(
"aquatic_requests_total",
"type" => "scrape",
"ip_version" => ip_version_to_metrics_str(self.ip_version)
);
let info_hashes = if let Some(info_hashes) = info_hashes {
info_hashes
@ -663,7 +680,11 @@ impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionReader<S> {
});
#[cfg(feature = "metrics")]
::metrics::increment_counter!("aquatic_responses_total", "type" => "error");
::metrics::increment_counter!(
"aquatic_requests_total",
"type" => "error",
"ip_version" => ip_version_to_metrics_str(self.ip_version)
);
result
}
@ -686,6 +707,7 @@ struct ConnectionWriter<S> {
pending_scrape_slab: Rc<RefCell<Slab<PendingScrapeResponse>>>,
server_start_instant: ServerStartInstant,
connection_id: ConnectionId,
ip_version: IpVersion,
}
impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionWriter<S> {
@ -749,16 +771,21 @@ impl<S: futures::AsyncRead + futures::AsyncWrite + Unpin> ConnectionWriter<S> {
match result {
Ok(Ok(())) => {
#[cfg(feature = "metrics")]
let out_message_type = match &out_message {
OutMessage::Offer(_) => "offer",
OutMessage::Answer(_) => "offer_answer",
OutMessage::AnnounceResponse(_) => "announce",
OutMessage::ScrapeResponse(_) => "scrape",
OutMessage::ErrorResponse(_) => "error",
};
{
let out_message_type = match &out_message {
OutMessage::Offer(_) => "offer",
OutMessage::Answer(_) => "offer_answer",
OutMessage::AnnounceResponse(_) => "announce",
OutMessage::ScrapeResponse(_) => "scrape",
OutMessage::ErrorResponse(_) => "error",
};
#[cfg(feature = "metrics")]
::metrics::increment_counter!("aquatic_responses_total", "type" => out_message_type);
::metrics::increment_counter!(
"aquatic_responses_total",
"type" => out_message_type,
"ip_version" => ip_version_to_metrics_str(self.ip_version),
);
}
self.connection_slab
.borrow_mut()
@ -839,3 +866,11 @@ fn create_tcp_listener(
Ok(unsafe { TcpListener::from_raw_fd(socket.into_raw_fd()) })
}
#[cfg(feature = "metrics")]
fn ip_version_to_metrics_str(ip_version: IpVersion) -> &'static str {
match ip_version {
IpVersion::V4 => "4",
IpVersion::V6 => "6",
}
}