Add metrics to Libp2p networking processes

common/common_metrics.go

@@ -0,0 +1,23 @@
+package common
+
+import (
+	"github.com/dominant-strategies/go-quai/metrics_config"
+	"github.com/prometheus/client_golang/prometheus"
+)
+
+var (
+	messageMetrics *prometheus.CounterVec
+	peerMetrics    *prometheus.GaugeVec
+)
+
+func init() {
+	registerMetrics()
+}
+
+func registerMetrics() {
+	messageMetrics = metrics_config.NewCounterVec("MessageCounters", "Counters to track messages sent over the P2P layer")
+	messageMetrics.WithLabelValues("sent")
+	messageMetrics.WithLabelValues("received")
+
+	peerMetrics = metrics_config.NewGaugeVec("PeerGauges", "Track the number of peers connected to this node")
+}

common/stream_services.go

@@ -22,6 +22,9 @@ func ReadMessageFromStream(stream network.Stream) ([]byte, error) {
 	if _, err := io.ReadFull(stream, lenBytes); err != nil {
 		return nil, errors.Wrap(err, "failed to read message length")
 	}
+	if messageMetrics != nil {
+		messageMetrics.WithLabelValues("received").Inc()
+	}
 	msgLen := binary.BigEndian.Uint32(lenBytes)
 
 	// Now read the message
@@ -49,6 +52,9 @@ func WriteMessageToStream(stream network.Stream, msg []byte) error {
 	if _, err := stream.Write(lenBytes); err != nil {
 		return errors.Wrap(err, "failed to write message length to stream")
 	}
+	if messageMetrics != nil {
+		messageMetrics.WithLabelValues("sent").Inc()
+	}
 
 	// Then write the message itself
 	_, err := stream.Write(msg)