Exapose the rateLimiter

virtual-kubelet · Jun 30, 2020 · a553fa2 · a553fa2
1 parent 2d648e1
commit a553fa2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/internal/commands/root/root.go b/internal/commands/root/root.go
@@ -196,6 +196,7 @@ func runRootCommandWithProviderAndClient(ctx context.Context, pInit provider.Ini
 		SecretInformer:    secretInformer,
 		ConfigMapInformer: configMapInformer,
 		ServiceInformer:   serviceInformer,
+		RateLimiter:       c.RateLimiter,
 	})
 	if err != nil {
 		return errors.Wrap(err, "error setting up pod controller")

diff --git a/opts/opts.go b/opts/opts.go
@@ -23,6 +23,7 @@ import (
 	"github.com/mitchellh/go-homedir"
 	"github.com/pkg/errors"
 	corev1 "k8s.io/api/core/v1"
+	"k8s.io/client-go/util/workqueue"
 )
 
 // Defaults for root command options
@@ -93,6 +94,8 @@ type Opts struct {
 	KubeAPIQPS int32
 	// KubeAPIBurst is the burst to allow while talking with kubernetes apiserver
 	KubeAPIBurst int32
+	// RateLimiter defines the rate limit of work queue
+	RateLimiter workqueue.RateLimiter
 
 	Version string
 }
@@ -147,6 +150,7 @@ func setDefaults(o *Opts) {
 	o.KubeClusterDomain = DefaultKubeClusterDomain
 	o.StreamIdleTimeout = DefaultStreamIdleTimeout
 	o.StreamCreationTimeout = DefaultStreamCreationTimeout
+	o.RateLimiter = workqueue.DefaultControllerRateLimiter()
 }
 
 func getEnv(key, defaultValue string) string {