Add metric

vacp2p · Dec 14, 2022 · c36fbb2 · c36fbb2
1 parent a4999bf
commit c36fbb2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/libp2p/services/autonatservice.nim b/libp2p/services/autonatservice.nim
@@ -13,12 +13,14 @@ else:
   {.push raises: [].}
 
 import std/[options, deques, sequtils]
-import chronos
+import chronos, metrics
 import ../switch
 import ../protocols/[connectivity/autonat]
 import ../utils/heartbeat
 import ../crypto/crypto
 
+declarePublicGauge(libp2p_autonat_reachability_confidence, "autonat reachability confidence", labels = ["reachability"])
+
 type
   AutonatService* = ref object of Service
     newConnectedPeerHandler: PeerEventHandler
@@ -73,12 +75,14 @@ proc handleAnswer(self: AutonatService, ans: NetworkReachability) {.async.} =
   if confidence >= self.minConfidence:
     self.networkReachability = NetworkReachability.Reachable
     self.confidence = some(confidence)
+    libp2p_autonat_reachability_confidence.set(value = confidence, labelValues = ["Reachable"])
   else:
     let notReachableCount = toSeq(self.answerDeque.items).countIt(it == NetworkReachability.NotReachable)
     let confidence = notReachableCount / self.maxQueueSize
     if confidence >= self.minConfidence:
       self.networkReachability = NetworkReachability.NotReachable
       self.confidence = some(confidence)
+      libp2p_autonat_reachability_confidence.set(confidence, ["NotReachable"])
     else:
       self.networkReachability = NetworkReachability.Unknown
       self.confidence = none(float)

diff --git a/tests/testautonatservice.nim b/tests/testautonatservice.nim
@@ -8,7 +8,7 @@
 # those terms.
 
 import std/options
-import chronos
+import chronos, metrics
 import unittest2
 import ../libp2p/[builders,
                   switch,
@@ -60,6 +60,7 @@ suite "Autonat Service":
     await autonatStub.finished
 
     check autonatService.networkReachability() == NetworkReachability.NotReachable
+    check libp2p_autonat_reachability_confidence.value(["NotReachable"]) == 0.3
 
     await allFuturesThrowing(
       switch1.stop(), switch2.stop(), switch3.stop(), switch4.stop())
@@ -90,6 +91,7 @@ suite "Autonat Service":
     await autonatStub.finished
 
     check autonatService.networkReachability() == NetworkReachability.Reachable
+    check libp2p_autonat_reachability_confidence.value(["Reachable"]) == 0.3
 
     await allFuturesThrowing(
       switch1.stop(), switch2.stop(), switch3.stop(), switch4.stop())
@@ -130,10 +132,12 @@ suite "Autonat Service":
     await awaiter.finished
 
     check autonatService.networkReachability() == NetworkReachability.NotReachable
+    check libp2p_autonat_reachability_confidence.value(["NotReachable"]) == 0.3
 
     await autonatStub.finished
 
     check autonatService.networkReachability() == NetworkReachability.Reachable
+    check libp2p_autonat_reachability_confidence.value(["Reachable"]) == 0.3
 
     await allFuturesThrowing(switch1.stop(), switch2.stop(), switch3.stop(), switch4.stop())