Merge pull request #10 from cwdsuzhou/March/expose_qps

Feature: expose kube api qps and burst

Author: cpuguy83

internal/commands/root/flag.go

@@ -43,6 +43,11 @@ func installFlags(flags *pflag.FlagSet, c *opts.Opts) {
 	flags.DurationVar(&c.InformerResyncPeriod, "full-resync-period", c.InformerResyncPeriod, "how often to perform a full resync of pods between kubernetes and the provider")
 	flags.DurationVar(&c.StartupTimeout, "startup-timeout", c.StartupTimeout, "How long to wait for the virtual-kubelet to start")
 
+	flags.Int32Var(&c.KubeAPIQPS, "kube-api-qps", c.KubeAPIQPS,
+		"kubeAPIQPS is the QPS to use while talking with kubernetes apiserver")
+	flags.Int32Var(&c.KubeAPIBurst, "kube-api-burst", c.KubeAPIBurst,
+		"kubeAPIBurst is the burst to allow while talking with kubernetes apiserver")
+
 	flagset := flag.NewFlagSet("klog", flag.PanicOnError)
 	klog.InitFlags(flagset)
 	flagset.VisitAll(func(f *flag.Flag) {

internal/commands/root/root.go

@@ -66,7 +66,7 @@ func runRootCommand(ctx context.Context, s *provider.Store, c *opts.Opts) error
 		return errors.Errorf("provider %q not found", c.Provider)
 	}
 
-	client, err := newClient(c.KubeConfigPath)
+	client, err := newClient(c.KubeConfigPath, c.KubeAPIQPS, c.KubeAPIBurst)
 	if err != nil {
 		return err
 	}
@@ -246,7 +246,7 @@ func waitFor(ctx context.Context, time time.Duration, ready <-chan struct{}) err
 	}
 }
 
-func newClient(configPath string) (*kubernetes.Clientset, error) {
+func newClient(configPath string, qps, burst int32) (*kubernetes.Clientset, error) {
 	var config *rest.Config
 
 	// Check if the kubeConfig file exists.
@@ -264,6 +264,14 @@ func newClient(configPath string) (*kubernetes.Clientset, error) {
 		}
 	}
 
+	if qps != 0 {
+		config.QPS = float32(qps)
+	}
+
+	if burst != 0 {
+		config.Burst = int(burst)
+	}
+
 	if masterURI := os.Getenv("MASTER_URI"); masterURI != "" {
 		config.Host = masterURI
 	}

opts/opts.go

@@ -82,6 +82,11 @@ type Opts struct {
 	// Startup Timeout is how long to wait for the kubelet to start
 	StartupTimeout time.Duration
 
+	// KubeAPIQPS is the QPS to use while talking with kubernetes apiserver
+	KubeAPIQPS int32
+	// KubeAPIBurst is the burst to allow while talking with kubernetes apiserver
+	KubeAPIBurst int32
+
 	Version string
 }