refactor: boost manager and friends ready for new types

google · Dec 20, 2024 · 64d0b33 · 64d0b33
1 parent 4ece136
commit 64d0b33
Show file tree

Hide file tree

Showing 17 changed files with 547 additions and 258 deletions.
diff --git a/cmd/main.go b/cmd/main.go
@@ -161,7 +161,7 @@ func setupControllers(mgr ctrl.Manager, boostMgr boost.Manager, cfg *config.Conf
 		Log:     ctrl.Log.WithName("boost-reconciler"),
 		Manager: boostMgr,
 	}
-	boostMgr.SetStartupCPUBoostReconciler(boostCtrl)
+	boostMgr.SetBoostReconciler(boost.RegularBoostTypeName, boostCtrl)
 	if err := boostCtrl.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "StartupCPUBoost")
 		os.Exit(1)

diff --git a/internal/boost/boost_suite_test.go b/internal/boost/boost_suite_test.go
@@ -19,6 +19,7 @@ import (
 	"time"
 
 	autoscaling "github.com/google/kube-startup-cpu-boost/api/v1alpha1"
+	cpuboost "github.com/google/kube-startup-cpu-boost/internal/boost"
 	bpod "github.com/google/kube-startup-cpu-boost/internal/boost/pod"
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
@@ -56,6 +57,7 @@ var _ = BeforeSuite(func() {
 
 	annotTemplate = &bpod.BoostPodAnnotation{
 		BoostTimestamp: time.Now(),
+		BoostType:      cpuboost.RegularBoostTypeName,
 		InitCPURequests: map[string]string{
 			"container-one": "500m",
 			"continer-two":  "500m",