Make leader election lease duration configurable through manager options

Author: vikaschoudhary16

pkg/manager/internal.go

@@ -40,6 +40,13 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/webhook"
 )
 
+const (
+	// Values taken from: https://github.com/kubernetes/apiserver/blob/master/pkg/apis/config/v1alpha1/defaults.go
+	defaultLeaseDuration = 15
+	defaultRenewDeadline = 10
+	defaultRetryPeriod   = 2
+)
+
 var log = logf.RuntimeLog.WithName("manager")
 
 type controllerManager struct {
@@ -101,6 +108,16 @@ type controllerManager struct {
 	host string
 
 	webhookServer *webhook.Server
+
+	// leaseDurationSeconds is the duration that non-leader candidates will
+	// wait to force acquire leadership.
+	leaseDurationSeconds *time.Duration
+	// renewDeadlineSeconds is the duration that the acting master will retry
+	// refreshing leadership before giving up.
+	renewDeadlineSeconds *time.Duration
+	// retryPeriodSeconds is the duration the LeaderElector clients should wait
+	// between tries of actions.
+	retryPeriodSeconds *time.Duration
 }
 
 // Add sets dependencies on i, and adds it to the list of runnables to start.
@@ -286,13 +303,28 @@ func (cm *controllerManager) start() {
 }
 
 func (cm *controllerManager) startLeaderElection() (err error) {
+
+	var leaseDuration, renewDeadline, retryPeriod time.Duration
+	if cm.leaseDurationSeconds == nil {
+		leaseDuration = defaultLeaseDuration
+	} else {
+		leaseDuration = *cm.leaseDurationSeconds
+	}
+	if cm.renewDeadlineSeconds == nil {
+		renewDeadline = defaultRenewDeadline
+	} else {
+		renewDeadline = *cm.renewDeadlineSeconds
+	}
+	if cm.retryPeriodSeconds == nil {
+		retryPeriod = defaultRetryPeriod
+	} else {
+		retryPeriod = *cm.retryPeriodSeconds
+	}
 	l, err := leaderelection.NewLeaderElector(leaderelection.LeaderElectionConfig{
-		Lock: cm.resourceLock,
-		// Values taken from: https://github.com/kubernetes/apiserver/blob/master/pkg/apis/config/v1alpha1/defaults.go
-		// TODO(joelspeed): These timings should be configurable
-		LeaseDuration: 15 * time.Second,
-		RenewDeadline: 10 * time.Second,
-		RetryPeriod:   2 * time.Second,
+		Lock:          cm.resourceLock,
+		LeaseDuration: leaseDuration * time.Second,
+		RenewDeadline: renewDeadline * time.Second,
+		RetryPeriod:   retryPeriod * time.Second,
 		Callbacks: leaderelection.LeaderCallbacks{
 			OnStartedLeading: func(_ context.Context) {
 				cm.start()

pkg/manager/manager.go

@@ -113,6 +113,17 @@ type Options struct {
 	// will use for holding the leader lock.
 	LeaderElectionID string
 
+	// LeaseDuration is the duration that non-leader candidates will
+	// wait to force acquire leadership. This is measured against time of
+	// last observed ack.
+	LeaseDurationSeconds *time.Duration
+	// RenewDeadline is the duration that the acting master will retry
+	// refreshing leadership before giving up.
+	RenewDeadlineSeconds *time.Duration
+	// RetryPeriod is the duration the LeaderElector clients should wait
+	// between tries of actions.
+	RetryPeriodSeconds *time.Duration
+
 	// Namespace if specified restricts the manager's cache to watch objects in
 	// the desired namespace Defaults to all namespaces
 	//
@@ -232,21 +243,24 @@ func New(config *rest.Config, options Options) (Manager, error) {
 	stop := make(chan struct{})
 
 	return &controllerManager{
-		config:           config,
-		scheme:           options.Scheme,
-		errChan:          make(chan error),
-		cache:            cache,
-		fieldIndexes:     cache,
-		client:           writeObj,
-		apiReader:        apiReader,
-		recorderProvider: recorderProvider,
-		resourceLock:     resourceLock,
-		mapper:           mapper,
-		metricsListener:  metricsListener,
-		internalStop:     stop,
-		internalStopper:  stop,
-		port:             options.Port,
-		host:             options.Host,
+		config:               config,
+		scheme:               options.Scheme,
+		errChan:              make(chan error),
+		cache:                cache,
+		fieldIndexes:         cache,
+		client:               writeObj,
+		apiReader:            apiReader,
+		recorderProvider:     recorderProvider,
+		resourceLock:         resourceLock,
+		mapper:               mapper,
+		metricsListener:      metricsListener,
+		internalStop:         stop,
+		internalStopper:      stop,
+		port:                 options.Port,
+		host:                 options.Host,
+		leaseDurationSeconds: options.LeaseDurationSeconds,
+		renewDeadlineSeconds: options.RenewDeadlineSeconds,
+		retryPeriodSeconds:   options.RetryPeriodSeconds,
 	}, nil
 }