github/client-go
1
0
Fork 0
mirror of https://github.com/kubernetes/client-go.git synced 2025-07-07 20:18:44 +00:00
Code Issues Projects Releases Wiki Activity

leaderelection: set timeout for tryAcquireOrRenew

Browse Source 
Kubernetes-commit: 90b287c12d921e4e08e522dbe214f5587f147956
This commit is contained in:
xuzhonghu 2018-06-20 15:44:31 +08:00 committed by Kubernetes Publisher
parent f783e1bbe3
commit 57f617c0b1
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tools/leaderelection/leaderelection.go
View File

@ -207,8 +207,20 @@ func (le *LeaderElector) renew(ctx context.Context) {
timeoutCtx, timeoutCancel := context.WithTimeout(ctx, le.config.RenewDeadline)
defer timeoutCancel()
err := wait.PollImmediateUntil(le.config.RetryPeriod, func() (bool, error) {
return le.tryAcquireOrRenew(), nil
done := make(chan bool, 1)
go func() {
defer close(done)
done <- le.tryAcquireOrRenew()
}()
select {
case <-timeoutCtx.Done():
return false, fmt.Errorf("failed to tryAcquireOrRenew %s", timeoutCtx.Err())
case result := <-done:
return result, nil
}
}, timeoutCtx.Done())
le.maybeReportTransition()
desc := le.config.Lock.Describe()
if err == nil {