github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-05 10:19:50 +00:00
Code Issues Projects Releases Wiki Activity

enhance leaderelection code

Browse Source
This commit is contained in:
zhengjiajin 2018-05-11 13:29:41 +08:00
parent e59ae29fbc
commit 9fe2a5a430
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
staging/src/k8s.io/client-go/tools/leaderelection/leaderelection.go
View File

@ -119,10 +119,6 @@ type LeaderCallbacks struct {
} }
// LeaderElector is a leader election client. // LeaderElector is a leader election client.
//
// possible future methods:
// * (le *LeaderElector) IsLeader()
// * (le *LeaderElector) GetLeader()
type LeaderElector struct { type LeaderElector struct {
config LeaderElectionConfig config LeaderElectionConfig
// internal bookkeeping // internal bookkeeping
@ -239,14 +235,14 @@ func (le *LeaderElector) tryAcquireOrRenew() bool {
le.observedTime = time.Now() le.observedTime = time.Now()
} }
if le.observedTime.Add(le.config.LeaseDuration).After(now.Time) && if le.observedTime.Add(le.config.LeaseDuration).After(now.Time) &&
oldLeaderElectionRecord.HolderIdentity != le.config.Lock.Identity() { !le.IsLeader() {
glog.V(4).Infof("lock is held by %v and has not yet expired", oldLeaderElectionRecord.HolderIdentity) glog.V(4).Infof("lock is held by %v and has not yet expired", oldLeaderElectionRecord.HolderIdentity)
return false return false
} }
// 3. We're going to try to update. The leaderElectionRecord is set to it's default // 3. We're going to try to update. The leaderElectionRecord is set to it's default
// here. Let's correct it before updating. // here. Let's correct it before updating.
if oldLeaderElectionRecord.HolderIdentity == le.config.Lock.Identity() { if le.IsLeader() {
leaderElectionRecord.AcquireTime = oldLeaderElectionRecord.AcquireTime leaderElectionRecord.AcquireTime = oldLeaderElectionRecord.AcquireTime
leaderElectionRecord.LeaderTransitions = oldLeaderElectionRecord.LeaderTransitions leaderElectionRecord.LeaderTransitions = oldLeaderElectionRecord.LeaderTransitions
} else { } else {
@ -263,12 +259,12 @@ func (le *LeaderElector) tryAcquireOrRenew() bool {
return true return true
} }
func (l *LeaderElector) maybeReportTransition() { func (le *LeaderElector) maybeReportTransition() {
if l.observedRecord.HolderIdentity == l.reportedLeader { if le.observedRecord.HolderIdentity == le.reportedLeader {
return return
} }
l.reportedLeader = l.observedRecord.HolderIdentity le.reportedLeader = le.observedRecord.HolderIdentity
if l.config.Callbacks.OnNewLeader != nil { if le.config.Callbacks.OnNewLeader != nil {
go l.config.Callbacks.OnNewLeader(l.reportedLeader) go le.config.Callbacks.OnNewLeader(le.reportedLeader)
} }
} }