allow users define controller workers

Signed-off-by: mingzhou.swx <[email protected]>
openkruise · Aug 23, 2022 · c8baf73 · c8baf73
1 parent 65b75a6
commit c8baf73
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/pkg/controller/batchrelease/batchrelease_controller.go b/pkg/controller/batchrelease/batchrelease_controller.go
@@ -52,6 +52,8 @@ var (
 )
 
 func init() {
+	flag.IntVar(&concurrentReconciles, "batchrelease-workers", 3, "Max concurrent workers for batchRelease controller.")
+
 	watchedWorkload = sync.Map{}
 	watchedWorkload.LoadOrStore(util.ControllerKindDep.String(), struct{}{})
 	watchedWorkload.LoadOrStore(util.ControllerKindSts.String(), struct{}{})

diff --git a/pkg/controller/rollout/rollout_controller.go b/pkg/controller/rollout/rollout_controller.go
@@ -18,6 +18,7 @@ package rollout
 
 import (
 	"context"
+	"flag"
 	"sync"
 	"time"
 
@@ -42,6 +43,8 @@ var (
 )
 
 func init() {
+	flag.IntVar(&concurrentReconciles, "rollout-workers", 3, "Max concurrent workers for rollout controller.")
+
 	watchedWorkload = sync.Map{}
 	watchedWorkload.LoadOrStore(util.ControllerKindDep.String(), struct{}{})
 	watchedWorkload.LoadOrStore(util.ControllerKindSts.String(), struct{}{})