Split clients for rate limit (#83)

* split clients for rate limit --------- Co-authored-by: Mangirdas <[email protected]>
castai · Sep 8, 2023 · 4778713 · 4778713
1 parent 03864b5
commit 4778713
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -109,16 +109,25 @@ func run(
 	if err != nil {
 		return err
 	}
+	restConfigLeader := rest.CopyConfig(restconfig)
+	restConfigDynamic := rest.CopyConfig(restconfig)
+
 	restconfig.RateLimiter = flowcontrol.NewTokenBucketRateLimiter(float32(cfg.KubeClient.QPS), cfg.KubeClient.Burst)
+	restConfigLeader.RateLimiter = flowcontrol.NewTokenBucketRateLimiter(float32(cfg.KubeClient.QPS), cfg.KubeClient.Burst)
+	restConfigDynamic.RateLimiter = flowcontrol.NewTokenBucketRateLimiter(float32(cfg.KubeClient.QPS), cfg.KubeClient.Burst)
 
 	helmClient := helm.NewClient(logger, helm.NewChartLoader(), restconfig)
 
 	clientset, err := kubernetes.NewForConfig(restconfig)
 	if err != nil {
 		return err
 	}
+	clientSetLeader, err := kubernetes.NewForConfig(restConfigLeader)
+	if err != nil {
+		return err
+	}
 
-	dynamicClient, err := dynamic.NewForConfig(restconfig)
+	dynamicClient, err := dynamic.NewForConfig(restConfigDynamic)
 	if err != nil {
 		return err
 	}
@@ -190,7 +199,7 @@ func run(
 
 	if cfg.LeaderElection.Enabled {
 		// Run actions service with leader election. Blocks.
-		return runWithLeaderElection(ctx, log, clientset, leaderHealthCheck, cfg.LeaderElection, svc.Run)
+		return runWithLeaderElection(ctx, log, clientSetLeader, leaderHealthCheck, cfg.LeaderElection, svc.Run)
 	}
 
 	// Run action service. Blocks.