- move to the latest version of grpc-prometheus

Signed-off-by: arpechenin <[email protected]>

Author: arpechenin

backend/src/apiserver/main.go

@@ -27,8 +27,7 @@ import (
 	"strings"
 	"sync"
 
-	grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
-	grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus"
+	grpc_prometheus "github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus"
 
 	"github.com/fsnotify/fsnotify"
 	"github.com/golang/glog"
@@ -212,16 +211,23 @@ func startRpcServer(resourceManager *resource.ResourceManager) {
 		glog.Fatalf("Failed to start RPC server: %v", err)
 	}
 
-	grpc_prometheus.EnableHandlingTimeHistogram(
-		grpc_prometheus.WithHistogramBuckets([]float64{
-			0.01, 0.03, 0.1, 0.3, 1, 3, 10, 15, 30, 60, 120, 300, //10 ms -> 5 min
-		}),
+	metrics := grpc_prometheus.NewServerMetrics(
+		grpc_prometheus.WithServerHandlingTimeHistogram(
+			grpc_prometheus.WithHistogramBuckets([]float64{
+				0.01, 0.03, 0.1, 0.3, 1, 3, 10, 15, 30, 60, 120, 300,
+			}),
+		),
 	)
-	m := grpc_middleware.WithUnaryServerChain(
-		apiServerInterceptor,
-		grpc_prometheus.UnaryServerInterceptor,
+
+	s := grpc.NewServer(
+		grpc.ChainUnaryInterceptor(
+			apiServerInterceptor,
+			metrics.UnaryServerInterceptor(),
+		),
+		grpc.MaxRecvMsgSize(math.MaxInt32),
 	)
-	s := grpc.NewServer(m, grpc.MaxRecvMsgSize(math.MaxInt32))
+
+	metrics.InitializeMetrics(s)
 
 	sharedExperimentServer := server.NewExperimentServer(resourceManager, &server.ExperimentServerOptions{CollectMetrics: *collectMetricsFlag})
 	sharedPipelineServer := server.NewPipelineServer(

go.mod

@@ -25,6 +25,7 @@ require (
 	github.com/google/go-cmp v0.6.0
 	github.com/google/uuid v1.6.0
 	github.com/gorilla/mux v1.8.0
+	github.com/grpc-ecosystem/go-grpc-middleware/providers/prometheus v1.1.0
 	github.com/grpc-ecosystem/go-grpc-middleware/v2 v2.3.2
 	github.com/grpc-ecosystem/grpc-gateway v1.16.0
 	github.com/jackc/pgx/v5 v5.5.4