refactor: boost manager and friends ready for new types

google · Nov 5, 2024 · 96c4dcb · 96c4dcb
1 parent c4562f5
commit 96c4dcb
Show file tree

Hide file tree

Showing 18 changed files with 577 additions and 283 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/config/manager/kustomization.yaml b/config/manager/kustomization.yaml
@@ -2,7 +2,8 @@ apiVersion: kustomize.config.k8s.io/v1beta1
 kind: Kustomization
 resources:
 - manager.yaml
+
 images:
 - name: controller
-  newName: ghcr.io/google/kube-startup-cpu-boost
+  newName: docker.io/library/kube-startup-cpu-boost
   newTag: dev
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",