server: Make threadiness configurable

Signed-off-by: Ce Gao <[email protected]>
kubeflow · Jun 14, 2018 · c3b785d · c3b785d
1 parent 4634c93
commit c3b785d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/tf-operator.v2/app/options/options.go b/cmd/tf-operator.v2/app/options/options.go
@@ -39,7 +39,7 @@ func (s *ServerOption) AddFlags(fs *flag.FlagSet) {
 		`The url of the Kubernetes API server,
 		 will overrides any value in kubeconfig, only required if out-of-cluster.`)
 
-	fs.IntVar(&s.Threadiness, "threadiness", 2,
+	fs.IntVar(&s.Threadiness, "threadiness", 1,
 		`How many threads to process the main logic`)
 
 	fs.BoolVar(&s.PrintVersion, "version", false, "Show version and quit")

diff --git a/cmd/tf-operator.v2/app/server.go b/cmd/tf-operator.v2/app/server.go
@@ -110,7 +110,7 @@ func Run(opt *options.ServerOption) error {
 
 	// Set leader election start function.
 	run := func(<-chan struct{}) {
-		if err := tc.Run(1, stopCh); err != nil {
+		if err := tc.Run(opt.Threadiness, stopCh); err != nil {
 			log.Errorf("Failed to run the controller: %v", err)
 		}
 	}