refactor(newSecretBasedPromClientController): inline setupPrometheusProvider into newSecretBasedPromClientController

Author: ingvagabund

pkg/descheduler/descheduler.go

@@ -130,17 +130,26 @@ func newInClusterPromClientController(prometheusClient promapi.Client, prometheu
 	}
 }
 
-func newSecretBasedPromClientController(prometheusClient promapi.Client, prometheusConfig *api.Prometheus) (*secretBasedPromClientController, error) {
+func newSecretBasedPromClientController(prometheusClient promapi.Client, prometheusConfig *api.Prometheus, namespacedSharedInformerFactory informers.SharedInformerFactory) (*secretBasedPromClientController, error) {
 	if prometheusConfig == nil || prometheusConfig.AuthToken == nil || prometheusConfig.AuthToken.SecretReference == nil {
 		return nil, fmt.Errorf("prometheus metrics source configuration is missing authentication token secret")
 	}
+
+	if namespacedSharedInformerFactory == nil {
+		return nil, fmt.Errorf("namespacedSharedInformerFactory not configured")
+	}
+
 	ctrl := &secretBasedPromClientController{
 		promClient:             prometheusClient,
 		queue:                  workqueue.NewRateLimitingQueueWithConfig(workqueue.DefaultControllerRateLimiter(), workqueue.RateLimitingQueueConfig{Name: "descheduler"}),
 		prometheusConfig:       prometheusConfig,
 		createPrometheusClient: client.CreatePrometheusClient,
 	}
 
+	authTokenSecret := ctrl.prometheusConfig.AuthToken.SecretReference
+	namespacedSharedInformerFactory.Core().V1().Secrets().Informer().AddEventHandler(ctrl.eventHandler())
+	ctrl.namespacedSecretsLister = namespacedSharedInformerFactory.Core().V1().Secrets().Lister().Secrets(authTokenSecret.Namespace)
+
 	return ctrl, nil
 }
 
@@ -181,20 +190,6 @@ func metricsProviderListToMap(providersList []api.MetricsProvider) map[api.Metri
 	return providersMap
 }
 
-// setupPrometheusProvider sets up the prometheus provider on the descheduler if configured
-func setupPrometheusProvider(ctrl *secretBasedPromClientController, namespacedSharedInformerFactory informers.SharedInformerFactory) error {
-	if ctrl == nil {
-		return nil
-	}
-	if namespacedSharedInformerFactory == nil {
-		return fmt.Errorf("namespacedSharedInformerFactory not configured")
-	}
-	authTokenSecret := ctrl.prometheusConfig.AuthToken.SecretReference
-	namespacedSharedInformerFactory.Core().V1().Secrets().Informer().AddEventHandler(ctrl.eventHandler())
-	ctrl.namespacedSecretsLister = namespacedSharedInformerFactory.Core().V1().Secrets().Lister().Secrets(authTokenSecret.Namespace)
-	return nil
-}
-
 func getPrometheusConfig(providersList []api.MetricsProvider) *api.Prometheus {
 	prometheusProvider := metricsProviderListToMap(providersList)[api.PrometheusMetrics]
 	if prometheusProvider == nil {
@@ -207,7 +202,7 @@ func configureSecretPromClientReconciler(prometheusConfig *api.Prometheus) bool
 	return prometheusConfig.URL != "" && prometheusConfig.AuthToken != nil
 }
 
-func newDescheduler(ctx context.Context, rs *options.DeschedulerServer, deschedulerPolicy *api.DeschedulerPolicy, evictionPolicyGroupVersion string, eventRecorder events.EventRecorder, client clientset.Interface, sharedInformerFactory informers.SharedInformerFactory, kubeClientSandbox *kubeClientSandbox) (*descheduler, error) {
+func newDescheduler(ctx context.Context, rs *options.DeschedulerServer, deschedulerPolicy *api.DeschedulerPolicy, evictionPolicyGroupVersion string, eventRecorder events.EventRecorder, client clientset.Interface, sharedInformerFactory, namespacedSharedInformerFactory informers.SharedInformerFactory, kubeClientSandbox *kubeClientSandbox) (*descheduler, error) {
 	podInformer := sharedInformerFactory.Core().V1().Pods().Informer()
 	// Temporarily register the PVC because it is used by the DefaultEvictor plugin during
 	// the descheduling cycle, where informer registration is ignored.
@@ -253,7 +248,7 @@ func newDescheduler(ctx context.Context, rs *options.DeschedulerServer, deschedu
 	if prometheusConfig != nil {
 		if configureSecretPromClientReconciler(prometheusConfig) {
 			// Secret-based mode
-			ctrl, err := newSecretBasedPromClientController(rs.PrometheusClient, prometheusConfig)
+			ctrl, err := newSecretBasedPromClientController(rs.PrometheusClient, prometheusConfig, namespacedSharedInformerFactory)
 			if err != nil {
 				return nil, err
 			}
@@ -615,16 +610,11 @@ func bootstrapDescheduler(
 	eventRecorder events.EventRecorder,
 ) (*descheduler, runFncType, error) {
 	// Always create descheduler with real client/factory first to register all informers
-	descheduler, err := newDescheduler(ctx, rs, deschedulerPolicy, evictionPolicyGroupVersion, eventRecorder, rs.Client, sharedInformerFactory, nil)
+	descheduler, err := newDescheduler(ctx, rs, deschedulerPolicy, evictionPolicyGroupVersion, eventRecorder, rs.Client, sharedInformerFactory, namespacedSharedInformerFactory, nil)
 	if err != nil {
 		return nil, nil, fmt.Errorf("failed to create new descheduler: %v", err)
 	}
 
-	// Setup Prometheus provider
-	if err := setupPrometheusProvider(descheduler.secretBasedPromClientCtrl, namespacedSharedInformerFactory); err != nil {
-		return nil, nil, fmt.Errorf("failed to setup Prometheus provider: %v", err)
-	}
-
 	// If in dry run mode, replace the descheduler with one using fake client/factory
 	if rs.DryRun {
 		// Create sandbox with resources to mirror from real client
@@ -638,15 +628,10 @@ func bootstrapDescheduler(
 		// TODO(ingvagabund): drop the previous queue
 		// TODO(ingvagabund): stop the previous pod evictor
 		// Replace descheduler with one using fake client/factory
-		descheduler, err = newDescheduler(ctx, rs, deschedulerPolicy, evictionPolicyGroupVersion, eventRecorder, kubeClientSandbox.fakeClient(), kubeClientSandbox.fakeSharedInformerFactory(), kubeClientSandbox)
+		descheduler, err = newDescheduler(ctx, rs, deschedulerPolicy, evictionPolicyGroupVersion, eventRecorder, kubeClientSandbox.fakeClient(), kubeClientSandbox.fakeSharedInformerFactory(), namespacedSharedInformerFactory, kubeClientSandbox)
 		if err != nil {
 			return nil, nil, fmt.Errorf("failed to create dry run descheduler: %v", err)
 		}
-
-		// Setup Prometheus provider (with the real shared informer factory as the secret is only read)
-		if err := setupPrometheusProvider(descheduler.secretBasedPromClientCtrl, namespacedSharedInformerFactory); err != nil {
-			return nil, nil, fmt.Errorf("failed to setup Prometheus provider for the dry run descheduler: %v", err)
-		}
 	}
 
 	// init is responsible for starting all informer factories, metrics providers

pkg/descheduler/descheduler_test.go

@@ -1347,7 +1347,7 @@ func TestEvictedPodRestorationInDryRun(t *testing.T) {
 	defer eventBroadcaster.Shutdown()
 
 	// Always create descheduler with real client/factory first to register all informers
-	descheduler, err := newDescheduler(ctxCancel, rs, internalDeschedulerPolicy, "v1", eventRecorder, rs.Client, sharedInformerFactory, nil)
+	descheduler, err := newDescheduler(ctxCancel, rs, internalDeschedulerPolicy, "v1", eventRecorder, rs.Client, sharedInformerFactory, nil, nil)
 	if err != nil {
 		t.Fatalf("Unable to create descheduler instance: %v", err)
 	}
@@ -1362,7 +1362,7 @@ func TestEvictedPodRestorationInDryRun(t *testing.T) {
 	}
 
 	// Replace descheduler with one using fake client/factory
-	descheduler, err = newDescheduler(ctxCancel, rs, internalDeschedulerPolicy, "v1", eventRecorder, kubeClientSandbox.fakeClient(), kubeClientSandbox.fakeSharedInformerFactory(), kubeClientSandbox)
+	descheduler, err = newDescheduler(ctxCancel, rs, internalDeschedulerPolicy, "v1", eventRecorder, kubeClientSandbox.fakeClient(), kubeClientSandbox.fakeSharedInformerFactory(), nil, kubeClientSandbox)
 	if err != nil {
 		t.Fatalf("Unable to create dry run descheduler instance: %v", err)
 	}
@@ -1839,15 +1839,10 @@ func setupPromClientControllerTest(ctx context.Context, t *testing.T, objects []
 		Prometheus: prometheusConfig,
 	}})
 
-	ctrl, err := newSecretBasedPromClientController(nil, prometheusConfig)
+	ctrl, err := newSecretBasedPromClientController(nil, prometheusConfig, namespacedInformerFactory)
 	if err != nil {
 		return nil, err
 	}
-	if setNamespacedSharedInformerFactory {
-		if err := setupPrometheusProvider(ctrl, namespacedInformerFactory); err != nil {
-			t.Fatal(err)
-		}
-	}
 
 	namespacedInformerFactory.Start(ctx.Done())
 	namespacedInformerFactory.WaitForCacheSync(ctx.Done())