Feature/relay integration metrics (#2376)

Extend metrics

- TransferBytesSent
- Active/idle peers
- Connection times

relay/server/peer.go

@@ -12,6 +12,7 @@ import (
 
 	"github.com/netbirdio/netbird/relay/healthcheck"
 	"github.com/netbirdio/netbird/relay/messages"
+	"github.com/netbirdio/netbird/relay/metrics"
 )
 
 const (
@@ -20,23 +21,25 @@ const (
 
 // Peer represents a peer connection
 type Peer struct {
-	log    *log.Entry
-	idS    string
-	idB    []byte
-	conn   net.Conn
-	connMu sync.RWMutex
-	store  *Store
+	metrics *metrics.Metrics
+	log     *log.Entry
+	idS     string
+	idB     []byte
+	conn    net.Conn
+	connMu  sync.RWMutex
+	store   *Store
 }
 
 // NewPeer creates a new Peer instance and prepare custom logging
-func NewPeer(id []byte, conn net.Conn, store *Store) *Peer {
+func NewPeer(metrics *metrics.Metrics, id []byte, conn net.Conn, store *Store) *Peer {
 	stringID := messages.HashIDToString(id)
 	return &Peer{
-		log:   log.WithField("peer_id", stringID),
-		idS:   stringID,
-		idB:   id,
-		conn:  conn,
-		store: store,
+		metrics: metrics,
+		log:     log.WithField("peer_id", stringID),
+		idS:     stringID,
+		idB:     id,
+		conn:    conn,
+		store:   store,
 	}
 }
 
@@ -70,6 +73,8 @@ func (p *Peer) Work() {
 		case messages.MsgTypeHealthCheck:
 			hc.OnHCResponse()
 		case messages.MsgTypeTransport:
+			p.metrics.TransferBytesRecv.Add(ctx, int64(n))
+			p.metrics.PeerActivity(p.String())
 			p.handleTransportMsg(msg)
 		case messages.MsgTypeClose:
 			p.log.Infof("peer exited gracefully")
@@ -167,8 +172,10 @@ func (p *Peer) handleTransportMsg(msg []byte) {
 		p.log.Errorf("failed to update transport message: %s", err)
 		return
 	}
-	_, err = dp.Write(msg)
+	n, err := dp.Write(msg)
 	if err != nil {
 		p.log.Errorf("failed to write transport message to: %s", dp.String())
+		return
 	}
+	p.metrics.TransferBytesSent.Add(context.Background(), int64(n))
 }

relay/server/relay.go

@@ -17,8 +17,9 @@ import (
 
 // Relay represents the relay server
 type Relay struct {
-	metrics   *metrics.Metrics
-	validator auth.Validator
+	metrics       *metrics.Metrics
+	metricsCancel context.CancelFunc
+	validator     auth.Validator
 
 	store       *Store
 	instanceURL string
@@ -43,15 +44,18 @@ type Relay struct {
 // A pointer to a Relay instance and an error. If the Relay instance is successfully created, the error is nil.
 // Otherwise, the error contains the details of what went wrong.
 func NewRelay(meter metric.Meter, exposedAddress string, tlsSupport bool, validator auth.Validator) (*Relay, error) {
-	m, err := metrics.NewMetrics(meter)
+	ctx, metricsCancel := context.WithCancel(context.Background())
+	m, err := metrics.NewMetrics(ctx, meter)
 	if err != nil {
+		metricsCancel()
 		return nil, fmt.Errorf("creating app metrics: %v", err)
 	}
 
 	r := &Relay{
-		metrics:   m,
-		validator: validator,
-		store:     NewStore(),
+		metrics:       m,
+		metricsCancel: metricsCancel,
+		validator:     validator,
+		store:         NewStore(),
 	}
 
 	if tlsSupport {
@@ -85,15 +89,15 @@ func (r *Relay) Accept(conn net.Conn) {
 		return
 	}
 
-	peer := NewPeer(peerID, conn, r.store)
+	peer := NewPeer(r.metrics, peerID, conn, r.store)
 	peer.log.Infof("peer connected from: %s", conn.RemoteAddr())
 	r.store.AddPeer(peer)
-	r.metrics.Peers.Add(context.Background(), 1)
+	r.metrics.PeerConnected(peer.String())
 	go func() {
 		peer.Work()
 		r.store.DeletePeer(peer)
 		peer.log.Debugf("relay connection closed")
-		r.metrics.Peers.Add(context.Background(), -1)
+		r.metrics.PeerDisconnected(peer.String())
 	}()
 }
 
@@ -112,6 +116,7 @@ func (r *Relay) Close(ctx context.Context) {
 		}(peer)
 	}
 	wg.Wait()
+	r.metricsCancel()
 	r.closeMu.Unlock()
 }