ghostream/internal/monitoring/monitoring.go

// Package monitoring serves Prometheus monitoring endpoints
package monitoring

import (
	"log"
	"net/http"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

// Options holds web package configuration
type Options struct {
	Enabled       bool
	ListenAddress string
}

var (
	// WebViewerServed is the total amount of viewer page served
	WebViewerServed = promauto.NewCounter(prometheus.CounterOpts{
		Name: "ghostream_web_viewer_served_total",
		Help: "The total amount of viewer served",
	})

	// WebSessions is the total amount of WebRTC session exchange
	WebSessions = promauto.NewCounter(prometheus.CounterOpts{
		Name: "ghostream_web_sessions_total",
		Help: "The total amount of WebRTC sessions exchanged",
	})

	// WebRTCConnectedSessions is the total amount of WebRTC session exchange
	WebRTCConnectedSessions = promauto.NewGauge(prometheus.GaugeOpts{
		Name: "ghostream_webrtc_connected_sessions",
		Help: "The current amount of opened WebRTC sessions",
	})
)

// Serve monitoring server that expose prometheus metrics
func Serve(cfg *Options) {
	if !cfg.Enabled {
		// Monitoring is not enabled, ignore
		return
	}

	mux := http.NewServeMux()
	mux.Handle("/metrics", promhttp.Handler())
	log.Printf("Monitoring HTTP server listening on %s", cfg.ListenAddress)
	log.Fatal(http.ListenAndServe(cfg.ListenAddress, mux))
}