Skip to content

Commit 43af1d7

Browse files
vid277cdxker
authored andcommitted
feature: add sending stats for dkg
1 parent f81315e commit 43af1d7

File tree

3 files changed

+1158
-22
lines changed

3 files changed

+1158
-22
lines changed

crates/node/src/handlers/dkg/handler.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ impl<N: Network, W: Wallet> Handler<N, W> for DkgState {
3333
}
3434
Some(DkgInner::Round1Package(_)) => {
3535
dkg_round1_package_metrics!(
36-
node.network_handle.peer_name(&source_peer)
36+
node.network_handle.peer_name(&source_peer),
37+
node.network_handle.peer_name(&node.peer_id)
3738
);
3839
self.handle_round1_payload(node, source_peer, inner_dkg)?;
3940
}
@@ -44,7 +45,10 @@ impl<N: Network, W: Wallet> Handler<N, W> for DkgState {
4445
}
4546
}
4647
NetworkEvent::MessageEvent((peer, DirectMessage::Round2Package(package))) => {
47-
dkg_round2_package_metrics!(node.network_handle.peer_name(&peer));
48+
dkg_round2_package_metrics!(
49+
node.network_handle.peer_name(&peer),
50+
node.network_handle.peer_name(&node.peer_id)
51+
);
4852
self.handle_round2_payload(node, peer, package).await?;
4953
}
5054
_ => {}

crates/types/src/macros.rs

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,14 +43,22 @@ macro_rules! dkg_start_metrics {
4343

4444
#[macro_export]
4545
macro_rules! dkg_round1_package_metrics {
46-
($peer_from:expr) => {{
47-
metrics::counter!("dkg_round1_packages_received", "peer_from" => $peer_from.to_string()).increment(1);
46+
($peer_from:expr, $peer_to:expr) => {{
47+
metrics::counter!("dkg_round1_packages_received",
48+
"source" => $peer_from.to_string(),
49+
"target" => $peer_to.to_string()
50+
)
51+
.increment(1);
4852
}};
4953
}
5054

5155
#[macro_export]
5256
macro_rules! dkg_round2_package_metrics {
53-
($peer_from:expr) => {{
54-
metrics::counter!("dkg_round2_packages_received", "peer_from" => $peer_from.to_string()).increment(1);
57+
($peer_from:expr, $peer_to:expr) => {{
58+
metrics::counter!("dkg_round2_packages_received",
59+
"source" => $peer_from.to_string(),
60+
"target" => $peer_to.to_string()
61+
)
62+
.increment(1);
5563
}};
5664
}

0 commit comments

Comments
 (0)