address comment

Signed-off-by: Cabinfever_B <cabinfeveroier@gmail.com>
CabinfeverB · Mar 13, 2024 · 6700333 · 6700333
1 parent 6119ca3
commit 6700333
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/client/pd_service_discovery.go b/client/pd_service_discovery.go
@@ -1022,7 +1022,9 @@ func (c *pdServiceDiscovery) switchLeader(url string) (change bool, err error) {
 		c.leader.Store(leaderClient)
 		// Run callbacks
 		if c.tsoGlobalAllocLeaderUpdatedCb != nil {
-			return change, c.tsoGlobalAllocLeaderUpdatedCb(url)
+			if err := c.tsoGlobalAllocLeaderUpdatedCb(url); err != nil {
+				return change, err
+			}
 		}
 		for _, cb := range c.leaderSwitchedCbs {
 			cb()