Make QPS and burst configurable (#47)

Signed-off-by: Yi Chen <github@chenyicn.net>

Author: ChenYi015

cmd/tf-operator.v1/app/options/options.go

@@ -18,7 +18,7 @@ import (
 	"flag"
 	"time"
 
-	"k8s.io/api/core/v1"
+	v1 "k8s.io/api/core/v1"
 )
 
 const (
@@ -39,6 +39,8 @@ type ServerOption struct {
 	MonitoringPort       int
 	ResyncPeriod         time.Duration
 	EventLevel           string
+	QPS                  float64
+	Burst                int
 }
 
 // NewServerOption creates a new CMServer with a default config.
@@ -71,4 +73,7 @@ func (s *ServerOption) AddFlags(fs *flag.FlagSet) {
 	fs.DurationVar(&s.ResyncPeriod, "resync-period", DefaultResyncPeriod, "Resync interval of the tf-operator")
 
 	fs.StringVar(&s.EventLevel, "event-level", EventLevelDebug, "The tfjob event level, support info,debug.")
+
+	fs.Float64Var(&s.QPS, "qps", 10.0, "QPS for kubernetes client")
+	fs.IntVar(&s.Burst, "burst", 30, "Maximum burst for throttle")
 }

cmd/tf-operator.v1/app/server.go

@@ -20,7 +20,7 @@ import (
 	"time"
 
 	"github.com/kubeflow/tf-operator/cmd/tf-operator.v1/app/options"
-	"github.com/kubeflow/tf-operator/pkg/apis/tensorflow/v1"
+	v1 "github.com/kubeflow/tf-operator/pkg/apis/tensorflow/v1"
 	tfjobclientset "github.com/kubeflow/tf-operator/pkg/client/clientset/versioned"
 	"github.com/kubeflow/tf-operator/pkg/client/clientset/versioned/scheme"
 	tfjobinformers "github.com/kubeflow/tf-operator/pkg/client/informers/externalversions"
@@ -99,8 +99,8 @@ func Run(opt *options.ServerOption) error {
 	if err != nil {
 		log.Fatalf("Error building kubeconfig: %s", err.Error())
 	}
-	kcfg.QPS = 10
-	kcfg.Burst = 20
+	kcfg.QPS = float32(opt.QPS)
+	kcfg.Burst = opt.Burst
 
 	// Create clients.
 	kubeClientSet, leaderElectionClientSet, tfJobClientSet, kubeBatchClientSet, err := createClientSets(kcfg)