From 60ee56a23eed36051ee82a1ac3fca2ea288fab32 Mon Sep 17 00:00:00 2001 From: Shrenuj Bansal Date: Fri, 7 Jul 2023 10:41:36 -0400 Subject: [PATCH] Add a metric to display pruning of the node's peer --- metrics/metrics.go | 5 +++++ node/modules/lp2p/pubsub.go | 1 + 2 files changed, 6 insertions(+) diff --git a/metrics/metrics.go b/metrics/metrics.go index ee7bd8695..cc2aa2ed8 100644 --- a/metrics/metrics.go +++ b/metrics/metrics.go @@ -116,6 +116,7 @@ var ( PubsubDeliverMessage = stats.Int64("pubsub/delivered", "Counter for total delivered messages", stats.UnitDimensionless) PubsubRejectMessage = stats.Int64("pubsub/rejected", "Counter for total rejected messages", stats.UnitDimensionless) PubsubDuplicateMessage = stats.Int64("pubsub/duplicate", "Counter for total duplicate messages", stats.UnitDimensionless) + PubsubPruneMessage = stats.Int64("pubsub/prune", "Counter for total prune messages", stats.UnitDimensionless) PubsubRecvRPC = stats.Int64("pubsub/recv_rpc", "Counter for total received RPCs", stats.UnitDimensionless) PubsubSendRPC = stats.Int64("pubsub/send_rpc", "Counter for total sent RPCs", stats.UnitDimensionless) PubsubDropRPC = stats.Int64("pubsub/drop_rpc", "Counter for total dropped RPCs", stats.UnitDimensionless) @@ -326,6 +327,10 @@ var ( Measure: PubsubDuplicateMessage, Aggregation: view.Count(), } + PubsubPruneMessageView = &view.View{ + Measure: PubsubPruneMessage, + Aggregation: view.Count(), + } PubsubRecvRPCView = &view.View{ Measure: PubsubRecvRPC, Aggregation: view.Count(), diff --git a/node/modules/lp2p/pubsub.go b/node/modules/lp2p/pubsub.go index 33a03f844..2b3efce6c 100644 --- a/node/modules/lp2p/pubsub.go +++ b/node/modules/lp2p/pubsub.go @@ -559,6 +559,7 @@ func (trw *tracerWrapper) Trace(evt *pubsub_pb.TraceEvent) { } case pubsub_pb.TraceEvent_PRUNE: + stats.Record(context.TODO(), metrics.PubsubPruneMessage.M(1)) if trw.traceMessage(evt.GetPrune().GetTopic()) { if trw.lp2pTracer != nil { trw.lp2pTracer.Trace(evt)