This is an automated cherry-pick of tikv#6636

close tikv#6635

Signed-off-by: ti-chi-bot <[email protected]>

server/election/leadership.go

@@ -16,6 +16,7 @@ package election
 
 import (
 	"context"
+	"sync"
 	"sync/atomic"
 
 	"github.com/pingcap/failpoint"
@@ -54,8 +55,14 @@ type Leadership struct {
 	leaderKey   string
 	leaderValue string
 
+<<<<<<< HEAD:server/election/leadership.go
 	keepAliveCtx        context.Context
 	keepAliceCancelFunc context.CancelFunc
+=======
+	keepAliveCtx            context.Context
+	keepAliveCancelFunc     context.CancelFunc
+	keepAliveCancelFuncLock sync.Mutex
+>>>>>>> 3bdfef1a3 (leadership: avoid potential data race (#6636)):pkg/election/leadership.go
 }
 
 // NewLeadership creates a new Leadership.
@@ -137,8 +144,15 @@ func (ls *Leadership) Keep(ctx context.Context) {
 	if ls == nil {
 		return
 	}
+<<<<<<< HEAD:server/election/leadership.go
 	ls.keepAliveCtx, ls.keepAliceCancelFunc = context.WithCancel(ctx)
 	ls.getLease().KeepAlive(ls.keepAliveCtx)
+=======
+	ls.keepAliveCancelFuncLock.Lock()
+	ls.keepAliveCtx, ls.keepAliveCancelFunc = context.WithCancel(ctx)
+	ls.keepAliveCancelFuncLock.Unlock()
+	go ls.getLease().KeepAlive(ls.keepAliveCtx)
+>>>>>>> 3bdfef1a3 (leadership: avoid potential data race (#6636)):pkg/election/leadership.go
 }
 
 // Check returns whether the leadership is still available.
@@ -230,8 +244,15 @@ func (ls *Leadership) Reset() {
 	if ls == nil || ls.getLease() == nil {
 		return
 	}
+<<<<<<< HEAD:server/election/leadership.go
 	if ls.keepAliceCancelFunc != nil {
 		ls.keepAliceCancelFunc()
+=======
+	ls.keepAliveCancelFuncLock.Lock()
+	if ls.keepAliveCancelFunc != nil {
+		ls.keepAliveCancelFunc()
+>>>>>>> 3bdfef1a3 (leadership: avoid potential data race (#6636)):pkg/election/leadership.go
 	}
+	ls.keepAliveCancelFuncLock.Unlock()
 	ls.getLease().Close()
 }