cherry pick tikv#2830 to release-4.0

Signed-off-by: ti-srebot <ti-srebot@pingcap.com>

client/base_client.go

@@ -48,8 +48,7 @@ type baseClient struct {
 	security SecurityOption
 
 	gRPCDialOptions []grpc.DialOption
-
-	timeout time.Duration
+	timeout         time.Duration
 }
 
 // SecurityOption records options about tls

client/client.go

@@ -41,6 +41,8 @@ type Region struct {
 type Client interface {
 	// GetClusterID gets the cluster ID from PD.
 	GetClusterID(ctx context.Context) uint64
+	// GetMemberInfo gets the members Info from PD
+	GetMemberInfo(ctx context.Context) ([]*pdpb.Member, error)
 	// GetLeaderAddr returns current leader's address. It returns "" before
 	// syncing leader from server.
 	GetLeaderAddr() string
@@ -219,6 +221,24 @@ func (c *client) checkStreamTimeout(loopCtx context.Context, cancel context.Canc
 	}
 }
 
+func (c *client) GetMemberInfo(ctx context.Context) ([]*pdpb.Member, error) {
+	start := time.Now()
+	defer func() { cmdDurationGetMemberInfo.Observe(time.Since(start).Seconds()) }()
+
+	ctx, cancel := context.WithTimeout(ctx, c.timeout)
+	resp, err := c.leaderClient().GetMembers(ctx, &pdpb.GetMembersRequest{
+		Header: c.requestHeader(),
+	})
+	cancel()
+	if err != nil {
+		cmdFailDurationGetMemberInfo.Observe(time.Since(start).Seconds())
+		c.ScheduleCheckLeader()
+		return nil, errors.WithStack(err)
+	}
+	members := resp.GetMembers()
+	return members, nil
+}
+
 func (c *client) tsLoop() {
 	defer c.wg.Done()
 

client/metrics.go

@@ -59,6 +59,7 @@ var (
 	cmdDurationTSO                      = cmdDuration.WithLabelValues("tso")
 	cmdDurationTSOAsyncWait             = cmdDuration.WithLabelValues("tso_async_wait")
 	cmdDurationGetRegion                = cmdDuration.WithLabelValues("get_region")
+	cmdDurationGetMemberInfo            = cmdDuration.WithLabelValues("get_member_info")
 	cmdDurationGetPrevRegion            = cmdDuration.WithLabelValues("get_prev_region")
 	cmdDurationGetRegionByID            = cmdDuration.WithLabelValues("get_region_byid")
 	cmdDurationScanRegions              = cmdDuration.WithLabelValues("scan_regions")
@@ -71,6 +72,7 @@ var (
 
 	cmdFailDurationGetRegion                  = cmdFailedDuration.WithLabelValues("get_region")
 	cmdFailDurationTSO                        = cmdFailedDuration.WithLabelValues("tso")
+	cmdFailDurationGetMemberInfo              = cmdFailedDuration.WithLabelValues("get_member_info")
 	cmdFailDurationGetPrevRegion              = cmdFailedDuration.WithLabelValues("get_prev_region")
 	cmdFailedDurationGetRegionByID            = cmdFailedDuration.WithLabelValues("get_region_byid")
 	cmdFailedDurationScanRegions              = cmdFailedDuration.WithLabelValues("scan_regions")