Register all GO process metrics (#1010)

Author: biglittlebigben

pkg/handler/handler.go

@@ -19,6 +19,8 @@ import (
 	"path"
 
 	"github.com/frostbyte73/core"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/collectors"
 	"google.golang.org/grpc"
 
 	"github.com/livekit/egress/pkg/config"
@@ -44,6 +46,10 @@ type Handler struct {
 }
 
 func NewHandler(conf *config.PipelineConfig, bus psrpc.MessageBus) (*Handler, error) {
+	// Register all GO process metrics
+	prometheus.Unregister(prometheus.NewGoCollector())
+	prometheus.MustRegister(collectors.NewGoCollector(collectors.WithGoCollectorRuntimeMetrics(collectors.MetricsAll)))
+
 	ipcClient, err := ipc.NewServiceClient(path.Join(config.TmpDir, conf.NodeID))
 	if err != nil {
 		return nil, err

pkg/service/metrics.go

@@ -21,6 +21,7 @@ import (
 
 	"github.com/linkdata/deadlock"
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/collectors"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	dto "github.com/prometheus/client_model/go"
 	"github.com/prometheus/common/expfmt"
@@ -38,6 +39,9 @@ type MetricsService struct {
 }
 
 func NewMetricsService(pm *ProcessManager) *MetricsService {
+	prometheus.Unregister(prometheus.NewGoCollector())
+	prometheus.MustRegister(collectors.NewGoCollector(collectors.WithGoCollectorRuntimeMetrics(collectors.MetricsAll)))
+
 	return &MetricsService{
 		pm: pm,
 	}