fix reset ts

Signed-off-by: Ryan Leung <[email protected]>
tikv · Apr 13, 2023 · 5122707 · 5122707
1 parent 4bd23b2
commit 5122707
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/tso/global_allocator.go b/pkg/tso/global_allocator.go
@@ -543,6 +543,11 @@ func (gta *GlobalTSOAllocator) campaignLeader() {
 	}()
 
 	gta.member.EnableLeader()
+	defer resetLeaderOnce.Do(func() {
+		cancel()
+		gta.member.ResetLeader()
+	})
+
 	// TODO: if enable-local-tso is true, check the cluster dc-location after the primary is elected
 	// go gta.tsoAllocatorManager.ClusterDCLocationChecker()
 	log.Info("tso primary is ready to serve", zap.String("tso-primary-name", gta.member.Name()))