From fe828d87d8a1eb4c018400fe525e127c2c0401c3 Mon Sep 17 00:00:00 2001
From: Dmitriy Ryajov <dryajov@gmail.com>
Date: Tue, 16 Jun 2020 22:14:02 -0600
Subject: [PATCH] count published messages (#224)

---
 libp2p/protocols/pubsub/floodsub.nim  | 4 ++++
 libp2p/protocols/pubsub/gossipsub.nim | 2 ++
 libp2p/protocols/pubsub/pubsub.nim    | 1 +
 libp2p/switch.nim                     | 1 +
 4 files changed, 8 insertions(+)

diff --git a/libp2p/protocols/pubsub/floodsub.nim b/libp2p/protocols/pubsub/floodsub.nim
index 517bb9817..4e6055cb0 100644
--- a/libp2p/protocols/pubsub/floodsub.nim
+++ b/libp2p/protocols/pubsub/floodsub.nim
@@ -92,6 +92,7 @@ method rpcHandler*(f: FloodSub,
         for p in toSendPeers:
           if p in f.peers and f.peers[p].id != peer.id:
             sent.add(f.peers[p].send(@[RPCMsg(messages: m.messages)]))
+
         # wait for all the futures now
         sent = await allFinished(sent)
         checkFutures(sent)
@@ -133,10 +134,13 @@ method publish*(f: FloodSub,
   for p in f.floodsub[topic]:
     trace "publishing message", name = topic, peer = p, data = data.shortLog
     sent.add(f.peers[p].send(@[RPCMsg(messages: @[msg])]))
+
   # wait for all the futures now
   sent = await allFinished(sent)
   checkFutures(sent)
 
+  libp2p_pubsub_messages_published.inc(labelValues = [topic])
+
 method unsubscribe*(f: FloodSub,
                     topics: seq[TopicPair]) {.async.} =
   await procCall PubSub(f).unsubscribe(topics)
diff --git a/libp2p/protocols/pubsub/gossipsub.nim b/libp2p/protocols/pubsub/gossipsub.nim
index e68d50d3b..e3c8eb44e 100644
--- a/libp2p/protocols/pubsub/gossipsub.nim
+++ b/libp2p/protocols/pubsub/gossipsub.nim
@@ -480,6 +480,8 @@ method publish*(g: GossipSub,
       sent.add(g.peers[p].send(@[RPCMsg(messages: @[msg])]))
     checkFutures(await allFinished(sent))
 
+    libp2p_pubsub_messages_published.inc(labelValues = [topic])
+
 method start*(g: GossipSub) {.async.} =
   ## start pubsub
   ## start long running/repeating procedures
diff --git a/libp2p/protocols/pubsub/pubsub.nim b/libp2p/protocols/pubsub/pubsub.nim
index 0667af9df..03fb96025 100644
--- a/libp2p/protocols/pubsub/pubsub.nim
+++ b/libp2p/protocols/pubsub/pubsub.nim
@@ -26,6 +26,7 @@ declareGauge(libp2p_pubsub_peers, "pubsub peer instances")
 declareGauge(libp2p_pubsub_topics, "pubsub subscribed topics")
 declareCounter(libp2p_pubsub_validation_success, "pubsub successfully validated messages")
 declareCounter(libp2p_pubsub_validation_failure, "pubsub failed validated messages")
+declarePublicCounter(libp2p_pubsub_messages_published, "published messages", labels = ["topic"])
 declareGauge(libp2p_pubsub_peers_per_topic, "pubsub peers per topic", labels = ["topic"])
 
 type
diff --git a/libp2p/switch.nim b/libp2p/switch.nim
index 131987015..63025d28c 100644
--- a/libp2p/switch.nim
+++ b/libp2p/switch.nim
@@ -12,6 +12,7 @@ import chronos, chronicles, metrics
 import connection,
        transports/transport,
        multistream,
+       multiaddress,
        protocols/protocol,
        protocols/secure/secure,
        protocols/secure/plaintext, # for plain text