Add configurable workers for each controller, defaults to 10

Signed-off-by: Dilip Tadepalli <[email protected]>

cmd/kcp/help.go

@@ -19,6 +19,8 @@ package main
 import (
  "fmt"
  "io"
+ "strconv"
+ "strings"
 
  "github.com/spf13/cobra"
  "github.com/spf13/pflag"
@@ -67,3 +69,31 @@ func printMostImportantFlags(w io.Writer, fss cliflag.NamedFlagSets, cols int, v
 
  cliflag.PrintSections(w, filteredFFS, cols)
 }
+
+func parseControllerSettings(flagString string) map[string]int {
+ settingsMap := make(map[string]int)
+
+ // Split the flag string based on commas
+ settings := strings.Split(flagString, ",")
+
+ // Iterate over each setting
+ for _, setting := range settings {
+ // Split the setting into key and value based on '='
+ parts := strings.Split(setting, "=")
+ if len(parts) == 2 {
+ key := strings.TrimSpace(parts[0])
+ value := strings.TrimSpace(parts[1])
+ num, err := strconv.Atoi(value)
+ if err != nil {
+ fmt.Println("Error:", err)
+ continue
+ }
+ settingsMap[key] = num
+ } else {
+ // Handle invalid format
+ fmt.Printf("Invalid format for setting: %s\n", setting)
+ }
+ }
+
+ return settingsMap
+}

cmd/kcp/kcp.go

@@ -68,6 +68,7 @@ func main() {
 
  // manually extract root directory from flags first as it influence all other flags
  rootDir := ".kcp"
+ controllerThreadsStr := ""
  for i, f := range os.Args {
  if f == "--root-directory" {
  if i < len(os.Args)-1 {
@@ -76,6 +77,10 @@ func main() {
  } else if strings.HasPrefix(f, "--root-directory=") {
  rootDir = strings.TrimPrefix(f, "--root-directory=")
  }
+
+ if strings.HasPrefix(f, "--controller-threads=") {
+ controllerThreadsStr = strings.TrimPrefix(f, "--controller-threads=")
+ }
  }
 
  serverOptions := options.NewOptions(rootDir)
@@ -136,7 +141,8 @@ func main() {
  }
  }
 
- s, err := server.NewServer(completedConfig)
+ controllerThreads := parseControllerSettings(controllerThreadsStr)
+ s, err := server.NewServer(completedConfig, server.WithControllerThreads(controllerThreads))
  if err != nil {
  return err
  }