address comment

Signed-off-by: husharp <[email protected]>

client/base_client.go

@@ -140,7 +140,7 @@ func (c *baseClient) memberLoop() {
 		case <-c.checkLeaderCh:
 		case <-ticker.C:
 		case <-ctx.Done():
-			log.Info("[pd] exit member loop")
+			log.Info("[pd] exit member loop due to context canceled")
 			return
 		}
 		failpoint.Inject("skipUpdateMember", func() {

client/retry/backoff.go

@@ -34,21 +34,19 @@ func WithBackoff(
 	fn func() error,
 	bo *BackOffer,
 ) error {
-	err := fn()
-	if err != nil {
+	if err := fn(); err != nil {
 		select {
 		case <-ctx.Done():
-			return err
 		case <-time.After(bo.NextBackoff()):
 			failpoint.Inject("backOffExecute", func() {
 				testBackOffExecuteFlag = true
 			})
 		}
-	} else {
-		bo.ResetBackoff()
-		return nil
+		return err
 	}
-	return err
+	// Reset backoff when fn() succeed.
+	bo.ResetBackoff()
+	return nil
 }
 
 // InitialBackOffer make the initial state for retrying.