Allow override QPS and Burst via flag (#1778)

* Allow override QPS and Burst via flag. Add `--kube-api-qps` and `--kube-api-burst` command-line flags to control client QPS and Burst respectively. The default value is set to 0 to keep backward compatibility. Signed-off-by: Pierangelo Di Pilato <[email protected]> * Inline validation functions Signed-off-by: Pierangelo Di Pilato <[email protected]>
knative · Oct 7, 2020 · 1a4c99a · 1a4c99a
1 parent 4c07e18
commit 1a4c99a
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/injection/config.go b/injection/config.go
@@ -20,6 +20,7 @@ import (
  "errors"
  "flag"
  "log"
+ "math"
  "os"
  "os/user"
  "path/filepath"
@@ -37,6 +38,9 @@ func ParseAndGetRESTConfigOrDie() *rest.Config {
  "The address of the Kubernetes API server. Overrides any value in kubeconfig. Only required if out-of-cluster.")
  kubeconfig = flag.String("kubeconfig", "",
  "Path to a kubeconfig. Only required if out-of-cluster.")
+
+ burst = flag.Int("kube-api-burst", 0, "Maximum burst for throttle.")
+ qps = flag.Float64("kube-api-qps", 0, "Maximum QPS to the server from the client.")
  )
  klog.InitFlags(flag.CommandLine)
  flag.Parse()
@@ -46,6 +50,16 @@ func ParseAndGetRESTConfigOrDie() *rest.Config {
  log.Fatal("Error building kubeconfig: ", err)
  }
 
+ if *burst < 0 {
+ log.Fatal("Invalid burst value", *burst)
+ }
+ if *qps < 0 || *qps > math.MaxFloat32 {
+ log.Fatal("Invalid QPS value", *qps)
+ }
+
+ cfg.Burst = *burst
+ cfg.QPS = float32(*qps)
+
  return cfg
 }