fix: metric reset not effect

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

Author: nayihz

cmd/bbr/main.go

@@ -32,8 +32,8 @@ import (
 	metricsserver "sigs.k8s.io/controller-runtime/pkg/metrics/server"
 
 	"sigs.k8s.io/gateway-api-inference-extension/internal/runnable"
+	"sigs.k8s.io/gateway-api-inference-extension/pkg/bbr/metrics"
 	runserver "sigs.k8s.io/gateway-api-inference-extension/pkg/bbr/server"
-	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/metrics"
 	"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/logging"
 )
 

cmd/epp/main.go

@@ -132,10 +132,6 @@ func run() error {
 		return err
 	}
 
-	poolNamespacedName := types.NamespacedName{
-		Name:      *poolName,
-		Namespace: *poolNamespace,
-	}
 	metrics.Register()
 	// Register metrics handler.
 	// Metrics endpoint is enabled in 'config/default/kustomization.yaml'. The Metrics options configure the server.
@@ -147,6 +143,10 @@ func run() error {
 		FilterProvider: filters.WithAuthenticationAndAuthorization,
 	}
 
+	poolNamespacedName := types.NamespacedName{
+		Name:      *poolName,
+		Namespace: *poolNamespace,
+	}
 	mgr, err := runserver.NewDefaultManager(poolNamespacedName, cfg, metricsServerOptions)
 	if err != nil {
 		setupLog.Error(err, "Failed to create controller manager")

pkg/epp/metrics/metrics.go

@@ -203,20 +203,20 @@ func Register() {
 }
 
 // Just for integration test
-func Unregister() {
-	metrics.Registry.Unregister(requestCounter)
-	metrics.Registry.Unregister(requestErrCounter)
-	metrics.Registry.Unregister(requestLatencies)
-	metrics.Registry.Unregister(requestSizes)
-	metrics.Registry.Unregister(responseSizes)
-	metrics.Registry.Unregister(inputTokens)
-	metrics.Registry.Unregister(outputTokens)
-	metrics.Registry.Unregister(runningRequests)
-	metrics.Registry.Unregister(NormalizedTimePerOutputToken)
-
-	metrics.Registry.Unregister(inferencePoolAvgKVCache)
-	metrics.Registry.Unregister(inferencePoolAvgQueueSize)
-	metrics.Registry.Unregister(inferencePoolReadyPods)
+func Reset() {
+	requestCounter.Reset()
+	requestErrCounter.Reset()
+	requestLatencies.Reset()
+	requestSizes.Reset()
+	responseSizes.Reset()
+	inputTokens.Reset()
+	outputTokens.Reset()
+	runningRequests.Reset()
+	NormalizedTimePerOutputToken.Reset()
+	inferencePoolAvgKVCache.Reset()
+	inferencePoolAvgQueueSize.Reset()
+	inferencePoolReadyPods.Reset()
+	SchedulerPluginProcessingLatencies.Reset()
 }
 
 // RecordRequstCounter records the number of requests.

test/integration/epp/hermetic_test.go

@@ -1196,7 +1196,6 @@ func TestFullDuplexStreamed_KubeInferenceModelRequest(t *testing.T) {
 
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
-			metrics.Register()
 			client, cleanup := setUpHermeticServer(t, test.pods, true)
 			t.Cleanup(cleanup)
 			responses, err := integrationutils.StreamedRequest(t, client, test.requests, len(test.wantResponses))
@@ -1215,9 +1214,8 @@ func TestFullDuplexStreamed_KubeInferenceModelRequest(t *testing.T) {
 					}
 				}
 			}
-			metrics.Unregister()
+			metrics.Reset()
 		})
-		return
 	}
 }
 
@@ -1331,6 +1329,7 @@ func BeforeSuite() func() {
 	// Init runtime.
 	ctrl.SetLogger(logger)
 
+	metrics.Register()
 	// Register metrics handler.
 	// Metrics endpoint is enabled in 'config/default/kustomization.yaml'. The Metrics options configure the server.
 	// More info: