Merge pull request #1707 from tahsinrahman/cabpk-concurrency

✨ Expose concurrency flag for KubeadmConfig (CABPK) controller

Author: k8s-ci-robot

bootstrap/kubeadm/controllers/kubeadmconfig_controller.go

@@ -40,6 +40,7 @@ import (
 	"sigs.k8s.io/cluster-api/util/secret"
 	ctrl "sigs.k8s.io/controller-runtime"
 	"sigs.k8s.io/controller-runtime/pkg/client"
+	"sigs.k8s.io/controller-runtime/pkg/controller"
 	"sigs.k8s.io/controller-runtime/pkg/handler"
 	"sigs.k8s.io/controller-runtime/pkg/source"
 )
@@ -65,7 +66,7 @@ type KubeadmConfigReconciler struct {
 }
 
 // SetupWithManager sets up the reconciler with the Manager.
-func (r *KubeadmConfigReconciler) SetupWithManager(mgr ctrl.Manager) error {
+func (r *KubeadmConfigReconciler) SetupWithManager(mgr ctrl.Manager, option controller.Options) error {
 	if r.KubeadmInitLock == nil {
 		r.KubeadmInitLock = locking.NewControlPlaneInitMutex(ctrl.Log.WithName("init-locker"), mgr.GetClient())
 	}
@@ -87,6 +88,7 @@ func (r *KubeadmConfigReconciler) SetupWithManager(mgr ctrl.Manager) error {
 				ToRequests: handler.ToRequestsFunc(r.ClusterToKubeadmConfigs),
 			},
 		).
+		WithOptions(option).
 		Complete(r)
 }
 

main.go

@@ -63,6 +63,7 @@ func main() {
 		machineConcurrency           int
 		machineSetConcurrency        int
 		machineDeploymentConcurrency int
+		kubeadmConfigConcurrency     int
 		syncPeriod                   time.Duration
 	)
 
@@ -90,6 +91,9 @@ func main() {
 	flag.IntVar(&machineDeploymentConcurrency, "machinedeployment-concurrency", 10,
 		"Number of machine deployments to process simultaneously")
 
+	flag.IntVar(&kubeadmConfigConcurrency, "kubeadmconfig-concurrency", 10,
+		"Number of kubeadm configs to process simultaneously")
+
 	flag.DurationVar(&syncPeriod, "sync-period", 10*time.Minute,
 		"The minimum interval at which watched resources are reconciled (e.g. 15m)")
 
@@ -156,7 +160,7 @@ func main() {
 	if err = (&kubeadmcontrollers.KubeadmConfigReconciler{
 		Client: mgr.GetClient(),
 		Log:    ctrl.Log.WithName("controllers").WithName("KubeadmConfigReconciler"),
-	}).SetupWithManager(mgr); err != nil {
+	}).SetupWithManager(mgr, concurrency(kubeadmConfigConcurrency)); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "KubeadmConfig")
 		os.Exit(1)
 	}