put rpc counter inside trace

amirylm · Jul 1, 2024 · 5cfa510 · 5cfa510
1 parent cf2e37b
commit 5cfa510
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/core/pubsub.go b/core/pubsub.go
@@ -41,11 +41,6 @@ func (c *Controller) setupPubsubRouter(ctx context.Context, cfg commons.Config)
 		opts = append(opts, pubsub.WithSeenMessagesTTL(cfg.Pubsub.Overlay.SeenTtl))
 	}
 
-	opts = append(opts, pubsub.WithAppSpecificRpcInspector(func(p peer.ID, rpc *pubsub.RPC) error {
-		c.pubsubRpcCounter.Add(1)
-		return nil
-	}))
-
 	denylist := pubsub.NewMapBlacklist()
 	opts = append(opts, pubsub.WithBlacklist(denylist))
 
@@ -74,6 +69,11 @@ func (c *Controller) setupPubsubRouter(ctx context.Context, cfg commons.Config)
 		tracer := newPubsubTracer(c.lggr.Named("PubsubTracer"), cfg.Pubsub.Trace.Debug, cfg.Pubsub.Trace.Skiplist, jtracer)
 		c.psTracer = tracer.(*psTracer)
 		opts = append(opts, pubsub.WithEventTracer(tracer))
+		// TODO: config?
+		opts = append(opts, pubsub.WithAppSpecificRpcInspector(func(p peer.ID, rpc *pubsub.RPC) error {
+			c.pubsubRpcCounter.Add(1)
+			return nil
+		}))
 	}
 
 	ps, err := pubsub.NewGossipSub(ctx, c.host, opts...)