From 9c30b85180855c2cf9acf677d9eb7c5c408b9a9e Mon Sep 17 00:00:00 2001
From: Lars Gierth <larsg@systemli.org>
Date: Wed, 1 Jul 2015 21:59:08 +0200
Subject: [PATCH] swarm: fix peers_total metric

License: MIT
Signed-off-by: Lars Gierth <larsg@systemli.org>
---
 p2p/net/swarm/swarm.go     | 4 ++--
 p2p/net/swarm/swarm_net.go | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/p2p/net/swarm/swarm.go b/p2p/net/swarm/swarm.go
index 5a8afbf42..28a37d42a 100644
--- a/p2p/net/swarm/swarm.go
+++ b/p2p/net/swarm/swarm.go
@@ -289,11 +289,11 @@ func (n *ps2netNotifee) ClosedStream(s *ps.Stream) {
 type metricsNotifiee Swarm
 
 func (nn *metricsNotifiee) Connected(n inet.Network, v inet.Conn) {
-	peersTotalGauge(n.LocalPeer()).Inc()
+	peersTotalGauge(n.LocalPeer()).Set(float64(len(n.Conns())))
 }
 
 func (nn *metricsNotifiee) Disconnected(n inet.Network, v inet.Conn) {
-	peersTotalGauge(n.LocalPeer()).Dec()
+	peersTotalGauge(n.LocalPeer()).Set(float64(len(n.Conns())))
 }
 
 func (nn *metricsNotifiee) OpenedStream(n inet.Network, v inet.Stream) {}
diff --git a/p2p/net/swarm/swarm_net.go b/p2p/net/swarm/swarm_net.go
index 0628ebabf..7b6e72906 100644
--- a/p2p/net/swarm/swarm_net.go
+++ b/p2p/net/swarm/swarm_net.go
@@ -58,12 +58,12 @@ func (n *Network) LocalPeer() peer.ID {
 	return n.Swarm().LocalPeer()
 }
 
-// Peers returns the connected peers
+// Peers returns the known peer IDs from the Peerstore
 func (n *Network) Peers() []peer.ID {
 	return n.Swarm().Peers()
 }
 
-// Peers returns the connected peers
+// Peers returns the Peerstore, which tracks known peers
 func (n *Network) Peerstore() peer.Peerstore {
 	return n.Swarm().peers
 }
-- 
GitLab