add webrtc metrtics.

internal/webrtc/metrics.go

@@ -0,0 +1,115 @@
+package webrtc
+
+import (
+	"sync"
+
+	"github.com/pion/webrtc/v3"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promauto"
+	"gitlab.com/demodesk/neko/server/pkg/types"
+)
+
+type metrics struct {
+	connectionState prometheus.Gauge
+	connectionCount prometheus.Counter
+
+	bytesSent     prometheus.Gauge
+	bytesReceived prometheus.Gauge
+}
+
+type metricsCtx struct {
+	mu sync.Mutex
+
+	sessions map[string]metrics
+}
+
+func newMetrics() *metricsCtx {
+	return &metricsCtx{
+		sessions: map[string]metrics{},
+	}
+}
+
+func (m *metricsCtx) getBySession(session types.Session) metrics {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	met, ok := m.sessions[session.ID()]
+	if ok {
+		return met
+	}
+
+	met = metrics{
+		connectionState: promauto.NewGauge(prometheus.GaugeOpts{
+			Name:      "connection_state",
+			Namespace: "neko",
+			Subsystem: "webrtc",
+			Help:      "Connection state of session.",
+			ConstLabels: map[string]string{
+				"session_id": session.ID(),
+			},
+		}),
+		connectionCount: promauto.NewCounter(prometheus.CounterOpts{
+			Name:      "connection_count",
+			Namespace: "neko",
+			Subsystem: "webrtc",
+			Help:      "Connection count of a session.",
+			ConstLabels: map[string]string{
+				"session_id": session.ID(),
+			},
+		}),
+		bytesSent: promauto.NewGauge(prometheus.GaugeOpts{
+			Name:      "bytes_sent",
+			Namespace: "neko",
+			Subsystem: "webrtc",
+			Help:      "Sent bytes to a session.",
+			ConstLabels: map[string]string{
+				"session_id": session.ID(),
+			},
+		}),
+		bytesReceived: promauto.NewGauge(prometheus.GaugeOpts{
+			Name:      "bytes_received",
+			Namespace: "neko",
+			Subsystem: "webrtc",
+			Help:      "Received bytes from a session.",
+			ConstLabels: map[string]string{
+				"session_id": session.ID(),
+			},
+		}),
+	}
+
+	m.sessions[session.ID()] = met
+	return met
+}
+
+func (m *metricsCtx) NewConnection(session types.Session) {
+	met := m.getBySession(session)
+	met.connectionCount.Add(1)
+}
+
+func (m *metricsCtx) SetState(session types.Session, state webrtc.PeerConnectionState) {
+	met := m.getBySession(session)
+
+	switch state {
+	case webrtc.PeerConnectionStateNew:
+		met.connectionState.Set(0)
+	case webrtc.PeerConnectionStateConnecting:
+		met.connectionState.Set(4)
+	case webrtc.PeerConnectionStateConnected:
+		met.connectionState.Set(5)
+	case webrtc.PeerConnectionStateDisconnected:
+		met.connectionState.Set(3)
+	case webrtc.PeerConnectionStateFailed:
+		met.connectionState.Set(2)
+	case webrtc.PeerConnectionStateClosed:
+		met.connectionState.Set(1)
+	default:
+		met.connectionState.Set(-1)
+	}
+}
+
+func (m *metricsCtx) SetTransportStats(session types.Session, data webrtc.TransportStats) {
+	met := m.getBySession(session)
+
+	met.bytesSent.Set(float64(data.BytesSent))
+	met.bytesReceived.Set(float64(data.BytesReceived))
+}