Merge pull request #103 from Labelbox/issue-99

ISSUE-99 - Implement suggestion

Author: xvzf

controllers/composable_controller.go

@@ -69,15 +69,20 @@ type ComposableReconciler struct {
 	Resolver   sdk.ResolveObject
 }
 
+type ReconcilerOptions struct {
+	QueriesPerSecond float32
+}
+
 // ManagerSettableReconciler - a Reconciler that can be added to a Manager
 type ManagerSettableReconciler interface {
 	reconcile.Reconciler
 	SetupWithManager(mgr ctrl.Manager) error
 }
 
 // NewReconciler ...
-func NewReconciler(mgr ctrl.Manager) ManagerSettableReconciler {
+func NewReconciler(mgr ctrl.Manager, opts ReconcilerOptions) ManagerSettableReconciler {
 	cfg := mgr.GetConfig()
+	cfg.QPS = opts.QueriesPerSecond
 	return &ComposableReconciler{
 		Client: mgr.GetClient(),
 		Scheme: mgr.GetScheme(),
@@ -96,9 +101,9 @@ func (r *ComposableReconciler) setController(controller controller.Controller) {
 	r.Controller = controller
 }
 
-//+kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables,verbs=get;list;watch;create;update;patch;delete
-//+kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables/status,verbs=get;update;patch
-//+kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables/finalizers,verbs=update
+// +kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables,verbs=get;list;watch;create;update;patch;delete
+// +kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables/status,verbs=get;update;patch
+// +kubebuilder:rbac:groups=ibmcloud.ibm.com,resources=composables/finalizers,verbs=update
 func (r *ComposableReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
 	logger := log.FromContext(ctx).WithValues("composable", req.NamespacedName)
 

main.go

@@ -57,6 +57,7 @@ func main() {
 	var developmentMode bool
 	var probeAddr string
 	var syncPeriod time.Duration
+	var queriesPerSecond float32
 	flag.StringVar(&metricsAddr, "metrics-bind-address", ":8080", "The address the metric endpoint binds to.")
 	flag.StringVar(&probeAddr, "health-probe-bind-address", ":8081", "The address the probe endpoint binds to.")
 	flag.BoolVar(&enableLeaderElection, "leader-elect", false,
@@ -65,6 +66,7 @@ func main() {
 	flag.BoolVar(&developmentMode, "development-mode", false, "Enable development mode")
 	flag.DurationVar(&syncPeriod, "sync-period", 60*time.Second, "Sync period")
 	flag.Int("max-concurrent-reconciles", 1, "Maximum number of concurrent reconciles for controllers.")
+	flag.Float32Var(&queriesPerSecond, "queries-per-second", 300.0, "Maximum number of queries per second made by the reconciler client.")
 	viper.BindPFlag("max-concurrent-reconciles", flag.Lookup("max-concurrent-reconciles"))
 	flag.Parse()
 
@@ -88,7 +90,9 @@ func main() {
 		os.Exit(1)
 	}
 
-	reconciler := controllers.NewReconciler(mgr)
+	reconciler := controllers.NewReconciler(mgr, controllers.ReconcilerOptions{
+		QueriesPerSecond: queriesPerSecond,
+	})
 	if err = reconciler.SetupWithManager(mgr); err != nil {
 		setupLog.Error(err, "unable to create controller", "controller", "Composable")
 		os.Exit(1)