ghostream/internal/monitoring/monitoring.go

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 {
	ListenAddress string
}

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

// ServeHTTP server that expose prometheus metrics
func ServeHTTP(cfg *Options) {
	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))
}