protocols/gossipsub: Add mesh metrics

Cargo.toml

@@ -39,7 +39,7 @@ dns-tokio = ["libp2p-dns", "libp2p-dns/tokio"]
 floodsub = ["libp2p-floodsub"]
 identify = ["libp2p-identify", "libp2p-metrics/identify"]
 kad = ["libp2p-kad", "libp2p-metrics/kad"]
-gossipsub = ["libp2p-gossipsub"]
+gossipsub = ["libp2p-gossipsub", "libp2p-metrics/gossipsub"]
 metrics = ["libp2p-metrics"]
 mdns = ["libp2p-mdns"]
 mplex = ["libp2p-mplex"]

misc/metrics/Cargo.toml

@@ -10,12 +10,14 @@ keywords = ["peer-to-peer", "libp2p", "networking"]
 categories = ["network-programming", "asynchronous"]
 
 [features]
+gossipsub = ["libp2p-gossipsub"]
 identify = ["libp2p-identify"]
 kad = ["libp2p-kad"]
 ping = ["libp2p-ping"]
 
 [dependencies]
 libp2p-core=  { version = "0.30.0", path = "../../core" }
+libp2p-gossipsub =  { version = "0.33.1", path = "../../protocols/gossipsub", optional = true }
 libp2p-identify =  { version = "0.31.0", path = "../../protocols/identify", optional = true }
 libp2p-kad =  { version = "0.33.0", path = "../../protocols/kad", optional = true }
 libp2p-ping =  { version = "0.31.0", path = "../../protocols/ping", optional = true }

misc/metrics/src/gossipsub.rs

@@ -0,0 +1,52 @@
+// Copyright 2021 Protocol Labs.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and associated documentation files (the "Software"),
+// to deal in the Software without restriction, including without limitation
+// the rights to use, copy, modify, merge, publish, distribute, sublicense,
+// and/or sell copies of the Software, and to permit persons to whom the
+// Software is furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+// DEALINGS IN THE SOFTWARE.
+
+use open_metrics_client::metrics::counter::Counter;
+use open_metrics_client::registry::Registry;
+
+pub struct Metrics {
+    messages: Counter,
+}
+
+impl Metrics {
+    pub fn new(registry: &mut Registry) -> Self {
+        let sub_registry = registry.sub_registry_with_prefix("gossipsub");
+
+        let messages = Counter::default();
+        sub_registry.register(
+            "messages",
+            "Number of messages received",
+            Box::new(messages.clone()),
+        );
+
+        Self { messages }
+    }
+}
+
+impl super::Recorder<libp2p_gossipsub::GossipsubEvent> for super::Metrics {
+    fn record(&self, event: &libp2p_gossipsub::GossipsubEvent) {
+        match event {
+            libp2p_gossipsub::GossipsubEvent::Message { .. } => {
+                self.gossipsub.messages.inc();
+            }
+            _ => {}
+        }
+    }
+}

misc/metrics/src/lib.rs

@@ -25,6 +25,8 @@
 //!
 //! See `examples` directory for more.
 
+#[cfg(feature = "gossipsub")]
+mod gossipsub;
 #[cfg(feature = "identify")]
 mod identify;
 #[cfg(feature = "kad")]
@@ -37,6 +39,8 @@ use open_metrics_client::registry::Registry;
 
 /// Set of Swarm and protocol metrics derived from emitted events.
 pub struct Metrics {
+    #[cfg(feature = "gossipsub")]
+    gossipsub: gossipsub::Metrics,
     #[cfg(feature = "identify")]
     identify: identify::Metrics,
     #[cfg(feature = "kad")]
@@ -58,6 +62,8 @@ impl Metrics {
     pub fn new(registry: &mut Registry) -> Self {
         let sub_registry = registry.sub_registry_with_prefix("libp2p");
         Self {
+            #[cfg(feature = "gossipsub")]
+            gossipsub: gossipsub::Metrics::new(sub_registry),
             #[cfg(feature = "identify")]
             identify: identify::Metrics::new(sub_registry),
             #[cfg(feature = "kad")]

protocols/gossipsub/Cargo.toml

@@ -29,6 +29,8 @@ regex = "1.4.0"
 futures-timer = "3.0.2"
 pin-project = "1.0.8"
 instant = "0.1.11"
+# Metrics dependencies
+open-metrics-client = "0.12"
 
 [dev-dependencies]
 async-std = "1.6.3"