Adding config variables to leader election process

elastic · Oct 18, 2023 · efd7411 · efd7411
1 parent defde80
commit efd7411
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/internal/pkg/composable/providers/kubernetesleaderelection/config.go b/internal/pkg/composable/providers/kubernetesleaderelection/config.go
@@ -13,10 +13,18 @@ type Config struct {
 	// Name of the leaderelection lease
 	LeaderLease string `config:"leader_lease"`
 
+	//Parameters to configure election process
+	LeaseDuration int `config:"leader_leaseduration"`
+	RenewDeadline int `config:"leader_renewdeadline"`
+	RetryPeriod   int `config:"leader_retryperiod"`
+
 	KubeClientOptions kubernetes.KubeClientOptions `config:"kube_client_options"`
 }
 
 // InitDefaults initializes the default values for the config.
 func (c *Config) InitDefaults() {
 	c.LeaderLease = "elastic-agent-cluster-leader"
+	c.LeaseDuration = 15
+	c.RenewDeadline = 10
+	c.RetryPeriod = 2
 }
diff --git a/internal/pkg/composable/providers/kubernetesleaderelection/kubernetes_leaderelection.go b/internal/pkg/composable/providers/kubernetesleaderelection/kubernetes_leaderelection.go
@@ -83,9 +83,9 @@ func (p *contextProvider) Run(ctx context.Context, comm corecomp.ContextProvider
 			},
 		},
 		ReleaseOnCancel: true,
-		LeaseDuration:   15 * time.Second,
-		RenewDeadline:   10 * time.Second,
-		RetryPeriod:     2 * time.Second,
+		LeaseDuration:   time.Duration(p.config.LeaseDuration) * time.Second,
+		RenewDeadline:   time.Duration(p.config.RenewDeadline) * time.Second,
+		RetryPeriod:     time.Duration(p.config.RetryPeriod) * time.Second,
 		Callbacks: leaderelection.LeaderCallbacks{
 			OnStartedLeading: func(ctx context.Context) {
 				p.logger.Debugf("leader election lock GAINED, id %v", id)