prometheus metrics caoyingjunz#231

cmd/localstorage-controller/main.go

@@ -20,7 +20,6 @@ import (
 	"context"
 	"flag"
 	"github.com/caoyingjunz/csi-driver-localstorage/pkg/metrics"
-	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"net"
 	"net/http"
 	"os"
@@ -172,8 +171,7 @@ func main() {
 
 	if *metricsPort > 0 {
 		mux := http.NewServeMux()
-		mux.Handle(*metricsPath, promhttp.Handler())
-
+		metrics.InstallHandler(mux, *metricsPath)
 		go wait.Until(func() {
 			err := metrics.TimingAcquisition(ctx, *interval)
 			if err != nil {

pkg/metrics/metric.go

@@ -4,11 +4,13 @@ import (
 	"context"
 	"github.com/caoyingjunz/csi-driver-localstorage/pkg/client/clientset/versioned"
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
 	meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/util/runtime"
 	"k8s.io/client-go/tools/clientcmd"
 	"k8s.io/client-go/util/homedir"
 	"k8s.io/klog/v2"
+	"net/http"
 	"path/filepath"
 	"time"
 )
@@ -35,6 +37,11 @@ func init() {
 	prometheus.MustRegister(VolumesTotalGauge, VolumeSizeGauge)
 }
 
+// InstallHandler registers the prometheus handler
+func InstallHandler(mux *http.ServeMux, path string) {
+	mux.Handle(path, promhttp.Handler())
+}
+
 // RegisterVolumeSize registers the size of localstorage volumes
 func RegisterVolumeSize(volId, volName, volPath string, size float64) {
 	VolumeSizeGauge.WithLabelValues(volId, volName, volPath).Set(size)