From 9e03c9033937e03cc7c0ec2b2dd230c6f1e5e306 Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 15:17:14 +0800
Subject: [PATCH 1/6] *: support setting endKey for ScanRange

Signed-off-by: disksing <i@disksing.com>
---
 client/client.go                     |  5 +++--
 go.mod                               |  2 ++
 go.sum                               |  4 ++--
 pkg/mock/mockcluster/mockcluster.go  |  4 ++--
 server/api/region.go                 |  4 ++--
 server/api/router.go                 |  2 +-
 server/cluster.go                    | 19 +++++--------------
 server/coordinator.go                |  2 +-
 server/core/basic_cluster.go         | 17 +++++------------
 server/core/region.go                | 25 ++++++++-----------------
 server/grpc_service.go               |  2 +-
 server/schedule/range_cluster.go     | 24 ++----------------------
 server/schedulers/adjacent_region.go |  2 +-
 server/statistics/region.go          |  6 ++----
 tools/pd-ctl/pdctl/command/global.go |  2 +-
 15 files changed, 38 insertions(+), 82 deletions(-)

diff --git a/client/client.go b/client/client.go
index bdacf6c0909..543c037f433 100644
--- a/client/client.go
+++ b/client/client.go
@@ -56,7 +56,7 @@ type Client interface {
 	// Limit limits the maximum number of regions returned.
 	// If a region has no leader, corresponding leader will be placed by a peer
 	// with empty value (PeerID is 0).
-	ScanRegions(ctx context.Context, key []byte, limit int) ([]*metapb.Region, []*metapb.Peer, error)
+	ScanRegions(ctx context.Context, key, endKey []byte, limit int) ([]*metapb.Region, []*metapb.Peer, error)
 	// GetStore gets a store from PD by store id.
 	// The store may expire later. Caller is responsible for caching and taking care
 	// of store change.
@@ -689,7 +689,7 @@ func (c *client) GetRegionByID(ctx context.Context, regionID uint64) (*metapb.Re
 	return resp.GetRegion(), resp.GetLeader(), nil
 }
 
-func (c *client) ScanRegions(ctx context.Context, key []byte, limit int) ([]*metapb.Region, []*metapb.Peer, error) {
+func (c *client) ScanRegions(ctx context.Context, key, endKey []byte, limit int) ([]*metapb.Region, []*metapb.Peer, error) {
 	if span := opentracing.SpanFromContext(ctx); span != nil {
 		span = opentracing.StartSpan("pdclient.ScanRegions", opentracing.ChildOf(span.Context()))
 		defer span.Finish()
@@ -701,6 +701,7 @@ func (c *client) ScanRegions(ctx context.Context, key []byte, limit int) ([]*met
 		Header:   c.requestHeader(),
 		StartKey: key,
 		Limit:    int32(limit),
+		EndKey:   endKey,
 	})
 	cancel()
 	if err != nil {
diff --git a/go.mod b/go.mod
index 9cdc8fd34e1..8de627b290b 100644
--- a/go.mod
+++ b/go.mod
@@ -47,3 +47,5 @@ require (
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.0.0
 )
+
+replace github.com/pingcap/kvproto => github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a
diff --git a/go.sum b/go.sum
index 398ffcab422..028c659f6a9 100644
--- a/go.sum
+++ b/go.sum
@@ -21,6 +21,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
+github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a h1:YV5474lam4SKIMro+Rjyi5qwro6VK9anvZTv8eAkY10=
+github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a/go.mod h1:QMdbTAXCHzzygQzqcG9uVUgU2fKeSN1GmfMiykdSzzY=
 github.com/dustin/go-humanize v0.0.0-20171111073723-bb3d318650d4/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
 github.com/dustin/go-humanize v0.0.0-20180421182945-02af3965c54e h1:Fw7ZmgiklsLh5EQWyHh1sumKSCG1+yjEctIpGKib87s=
 github.com/dustin/go-humanize v0.0.0-20180421182945-02af3965c54e/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
@@ -108,8 +110,6 @@ github.com/pingcap/errors v0.11.0 h1:DCJQB8jrHbQ1VVlMFIrbj2ApScNNotVmkSNplu2yUt4
 github.com/pingcap/errors v0.11.0/go.mod h1:Oi8TUi2kEtXXLMJk9l1cGmz20kV3TaQ0usTwv5KuLY8=
 github.com/pingcap/failpoint v0.0.0-20190512135322-30cc7431d99c h1:hvQd3aOLKLF7xvRV6DzvPkKY4QXzfVbjU1BhW0d9yL8=
 github.com/pingcap/failpoint v0.0.0-20190512135322-30cc7431d99c/go.mod h1:DNS3Qg7bEDhU6EXNHF+XSv/PGznQaMJ5FWvctpm6pQI=
-github.com/pingcap/kvproto v0.0.0-20190516013202-4cf58ad90b6c h1:pY/MQQ5UajEHfSnQS8rFAM9gw9bBKzqBl414cdfhpRQ=
-github.com/pingcap/kvproto v0.0.0-20190516013202-4cf58ad90b6c/go.mod h1:QMdbTAXCHzzygQzqcG9uVUgU2fKeSN1GmfMiykdSzzY=
 github.com/pingcap/log v0.0.0-20190715063458-479153f07ebd h1:hWDol43WY5PGhsh3+8794bFHY1bPrmu6bTalpssCrGg=
 github.com/pingcap/log v0.0.0-20190715063458-479153f07ebd/go.mod h1:WpHUKhNZ18v116SvGrmjkA9CBhYmuUTKL+p8JC9ANEw=
 github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
diff --git a/pkg/mock/mockcluster/mockcluster.go b/pkg/mock/mockcluster/mockcluster.go
index cb83ee9f870..464c326dd4a 100644
--- a/pkg/mock/mockcluster/mockcluster.go
+++ b/pkg/mock/mockcluster/mockcluster.go
@@ -55,8 +55,8 @@ func (mc *Cluster) allocID() (uint64, error) {
 }
 
 // ScanRegions scans region with start key, until number greater than limit.
-func (mc *Cluster) ScanRegions(startKey []byte, limit int) []*core.RegionInfo {
-	return mc.Regions.ScanRange(startKey, limit)
+func (mc *Cluster) ScanRegions(startKey, endKey []byte, limit int) []*core.RegionInfo {
+	return mc.Regions.ScanRange(startKey, endKey, limit)
 }
 
 // LoadRegion puts region info without leader
diff --git a/server/api/region.go b/server/api/region.go
index ef67fff5511..3b3e83243ad 100644
--- a/server/api/region.go
+++ b/server/api/region.go
@@ -149,7 +149,7 @@ func (h *regionsHandler) GetAll(w http.ResponseWriter, r *http.Request) {
 	h.rd.JSON(w, http.StatusOK, regionsInfo)
 }
 
-func (h *regionsHandler) ScanRegionsByKey(w http.ResponseWriter, r *http.Request) {
+func (h *regionsHandler) ScanRegions(w http.ResponseWriter, r *http.Request) {
 	cluster := h.svr.GetRaftCluster()
 	if cluster == nil {
 		h.rd.JSON(w, http.StatusInternalServerError, server.ErrNotBootstrapped.Error())
@@ -170,7 +170,7 @@ func (h *regionsHandler) ScanRegionsByKey(w http.ResponseWriter, r *http.Request
 	if limit > maxRegionLimit {
 		limit = maxRegionLimit
 	}
-	regions := cluster.ScanRegionsByKey([]byte(startKey), limit)
+	regions := cluster.ScanRegions([]byte(startKey), nil, limit)
 	regionsInfo := convertToAPIRegions(regions)
 	h.rd.JSON(w, http.StatusOK, regionsInfo)
 }
diff --git a/server/api/router.go b/server/api/router.go
index 1d4c17686f9..bc0ed0bbf7a 100644
--- a/server/api/router.go
+++ b/server/api/router.go
@@ -90,7 +90,7 @@ func createRouter(prefix string, svr *server.Server) *mux.Router {
 
 	regionsHandler := newRegionsHandler(svr, rd)
 	router.HandleFunc("/api/v1/regions", regionsHandler.GetAll).Methods("GET")
-	router.HandleFunc("/api/v1/regions/key", regionsHandler.ScanRegionsByKey).Methods("GET")
+	router.HandleFunc("/api/v1/regions/key", regionsHandler.ScanRegions).Methods("GET")
 	router.HandleFunc("/api/v1/regions/store/{id}", regionsHandler.GetStoreRegions).Methods("GET")
 	router.HandleFunc("/api/v1/regions/writeflow", regionsHandler.GetTopWriteFlow).Methods("GET")
 	router.HandleFunc("/api/v1/regions/readflow", regionsHandler.GetTopReadFlow).Methods("GET")
diff --git a/server/cluster.go b/server/cluster.go
index 2687f6cdc1a..8059a8d48ba 100644
--- a/server/cluster.go
+++ b/server/cluster.go
@@ -557,19 +557,10 @@ func (c *RaftCluster) GetRegionInfoByKey(regionKey []byte) *core.RegionInfo {
 	return c.core.SearchRegion(regionKey)
 }
 
-// ScanRegionsByKey scans region with start key, until number greater than limit.
-func (c *RaftCluster) ScanRegionsByKey(startKey []byte, limit int) []*core.RegionInfo {
-	return c.core.ScanRange(startKey, limit)
-}
-
-// ScanRegions scans region with start key, until number greater than limit.
-func (c *RaftCluster) ScanRegions(startKey []byte, limit int) []*core.RegionInfo {
-	return c.core.ScanRange(startKey, limit)
-}
-
-// ScanRangeWithEndKey scans regions intersecting [start key, end key).
-func (c *RaftCluster) ScanRangeWithEndKey(startKey, endKey []byte) []*core.RegionInfo {
-	return c.core.ScanRangeWithEndKey(startKey, endKey)
+// ScanRegions scans region with start key, until the region contains endKey, or
+// total number greater than limit.
+func (c *RaftCluster) ScanRegions(startKey, endKey []byte, limit int) []*core.RegionInfo {
+	return c.core.ScanRange(startKey, endKey, limit)
 }
 
 // GetRegionByID gets region and leader peer by regionID from cluster.
@@ -660,7 +651,7 @@ func (c *RaftCluster) GetAverageRegionSize() int64 {
 func (c *RaftCluster) GetRegionStats(startKey, endKey []byte) *statistics.RegionStats {
 	c.RLock()
 	defer c.RUnlock()
-	return statistics.GetRegionStats(c.core.ScanRangeWithEndKey, startKey, endKey)
+	return statistics.GetRegionStats(c.core.ScanRange(startKey, endKey, -1))
 }
 
 // GetStoresStats returns stores' statistics from cluster.
diff --git a/server/coordinator.go b/server/coordinator.go
index fe9b828484c..fda994c8216 100644
--- a/server/coordinator.go
+++ b/server/coordinator.go
@@ -108,7 +108,7 @@ func (c *coordinator) patrolRegions() {
 			return
 		}
 
-		regions := c.cluster.ScanRegions(key, patrolScanRegionLimit)
+		regions := c.cluster.ScanRegions(key, nil, patrolScanRegionLimit)
 		if len(regions) == 0 {
 			// Resets the scan key.
 			key = nil
diff --git a/server/core/basic_cluster.go b/server/core/basic_cluster.go
index fd2f00300f9..e765e4de018 100644
--- a/server/core/basic_cluster.go
+++ b/server/core/basic_cluster.go
@@ -284,19 +284,12 @@ func (bc *BasicCluster) SearchPrevRegion(regionKey []byte) *RegionInfo {
 	return bc.Regions.SearchPrevRegion(regionKey)
 }
 
-// ScanRange scans from the first region containing or behind start key,
-// until number greater than limit.
-func (bc *BasicCluster) ScanRange(startKey []byte, limit int) []*RegionInfo {
+// ScanRange scans regions intersecting [start key, end key), returns at most
+// `limit` regions. limit <= 0 means no limit.
+func (bc *BasicCluster) ScanRange(startKey, endKey []byte, limit int) []*RegionInfo {
 	bc.RLock()
 	defer bc.RUnlock()
-	return bc.Regions.ScanRange(startKey, limit)
-}
-
-// ScanRangeWithEndKey scans regions intersecting [start key, end key).
-func (bc *BasicCluster) ScanRangeWithEndKey(startKey, endKey []byte) []*RegionInfo {
-	bc.RLock()
-	defer bc.RUnlock()
-	return bc.Regions.ScanRangeWithEndKey(startKey, endKey)
+	return bc.Regions.ScanRange(startKey, endKey, limit)
 }
 
 // GetOverlaps returns the regions which are overlapped with the specified region range.
@@ -322,7 +315,7 @@ type RegionSetInformer interface {
 	GetStoreRegionCount(storeID uint64) int
 	GetRegion(id uint64) *RegionInfo
 	GetAdjacentRegions(region *RegionInfo) (*RegionInfo, *RegionInfo)
-	ScanRegions(startKey []byte, limit int) []*RegionInfo
+	ScanRegions(startKey, endKey []byte, limit int) []*RegionInfo
 }
 
 // StoreSetInformer provides access to a shared informer of stores.
diff --git a/server/core/region.go b/server/core/region.go
index 75620f73501..2331b5da047 100644
--- a/server/core/region.go
+++ b/server/core/region.go
@@ -709,30 +709,21 @@ func (r *RegionsInfo) GetFollower(storeID uint64, regionID uint64) *RegionInfo {
 	return r.followers[storeID].Get(regionID)
 }
 
-// ScanRange scans from the first region containing or behind start key,
-// until number greater than limit.
-func (r *RegionsInfo) ScanRange(startKey []byte, limit int) []*RegionInfo {
-	res := make([]*RegionInfo, 0, limit)
-	r.tree.scanRange(startKey, func(metaRegion *metapb.Region) bool {
-		res = append(res, r.GetRegion(metaRegion.GetId()))
-		return len(res) < limit
-	})
-	return res
-}
-
-// ScanRangeWithEndKey scans regions intersecting [start key, end key).
-func (r *RegionsInfo) ScanRangeWithEndKey(startKey, endKey []byte) []*RegionInfo {
-	var regions []*RegionInfo
+// ScanRange scans regions intersecting [start key, end key), returns at most
+// `limit` regions. limit <= 0 means no limit.
+func (r *RegionsInfo) ScanRange(startKey, endKey []byte, limit int) []*RegionInfo {
+	var res []*RegionInfo
 	r.tree.scanRange(startKey, func(meta *metapb.Region) bool {
 		if len(endKey) > 0 && bytes.Compare(meta.StartKey, endKey) >= 0 {
 			return false
 		}
-		if region := r.GetRegion(meta.GetId()); region != nil {
-			regions = append(regions, region)
+		if limit > 0 && len(res) >= limit {
+			return false
 		}
+		res = append(res, r.GetRegion(meta.GetId()))
 		return true
 	})
-	return regions
+	return res
 }
 
 // ScanRangeWithIterator scans from the first region containing or behind start key,
diff --git a/server/grpc_service.go b/server/grpc_service.go
index 1dbe53e1b5c..602033bc1bc 100644
--- a/server/grpc_service.go
+++ b/server/grpc_service.go
@@ -458,7 +458,7 @@ func (s *Server) ScanRegions(ctx context.Context, request *pdpb.ScanRegionsReque
 	if cluster == nil {
 		return &pdpb.ScanRegionsResponse{Header: s.notBootstrappedHeader()}, nil
 	}
-	regions := cluster.ScanRegionsByKey(request.GetStartKey(), int(request.GetLimit()))
+	regions := cluster.ScanRegions(request.GetStartKey(), request.GetEndKey(), int(request.GetLimit()))
 	resp := &pdpb.ScanRegionsResponse{Header: s.header()}
 	for _, r := range regions {
 		leader := r.GetLeader()
diff --git a/server/schedule/range_cluster.go b/server/schedule/range_cluster.go
index 162cdaf2e21..30d2256027c 100644
--- a/server/schedule/range_cluster.go
+++ b/server/schedule/range_cluster.go
@@ -14,8 +14,6 @@
 package schedule
 
 import (
-	"bytes"
-
 	"github.com/gogo/protobuf/proto"
 	"github.com/pingcap/kvproto/pkg/pdpb"
 	"github.com/pingcap/pd/server/core"
@@ -34,26 +32,8 @@ const scanLimit = 128
 // The cluster can only know the regions within [startKey, endKey].
 func GenRangeCluster(cluster Cluster, startKey, endKey []byte) *RangeCluster {
 	regions := core.NewRegionsInfo()
-	scanKey := startKey
-	loopEnd := false
-	for !loopEnd {
-		collect := cluster.ScanRegions(scanKey, scanLimit)
-		if len(collect) == 0 {
-			break
-		}
-		for _, r := range collect {
-			if bytes.Compare(r.GetStartKey(), endKey) < 0 {
-				regions.SetRegion(r)
-			} else {
-				loopEnd = true
-				break
-			}
-			if string(r.GetEndKey()) == "" {
-				loopEnd = true
-				break
-			}
-			scanKey = r.GetEndKey()
-		}
+	for _, r := range cluster.ScanRegions(startKey, endKey, -1) {
+		regions.AddRegion(r)
 	}
 	return &RangeCluster{
 		Cluster: cluster,
diff --git a/server/schedulers/adjacent_region.go b/server/schedulers/adjacent_region.go
index 4dd011923a3..277a94b2500 100644
--- a/server/schedulers/adjacent_region.go
+++ b/server/schedulers/adjacent_region.go
@@ -157,7 +157,7 @@ func (l *balanceAdjacentRegionScheduler) Schedule(cluster schedule.Cluster) []*o
 	}
 
 	l.cacheRegions.clear()
-	regions := cluster.ScanRegions(l.lastKey, scanLimit)
+	regions := cluster.ScanRegions(l.lastKey, nil, scanLimit)
 	// scan to the end
 	if len(regions) <= 1 {
 		schedulerStatus.WithLabelValues(l.GetName(), "adjacent_count").Set(float64(l.adjacentRegionsCount))
diff --git a/server/statistics/region.go b/server/statistics/region.go
index 3b57166b8d6..d27af1167ca 100644
--- a/server/statistics/region.go
+++ b/server/statistics/region.go
@@ -123,11 +123,9 @@ func (s *RegionStats) Observe(r *core.RegionInfo) {
 	}
 }
 
-// GetRegionStats scans regions that intersect range [startKey, endKey) and sums up
-// their statistics.
-func GetRegionStats(f func(startKey, endKey []byte) []*core.RegionInfo, startKey, endKey []byte) *RegionStats {
+// GetRegionStats sums regions' statistics.
+func GetRegionStats(regions []*core.RegionInfo) *RegionStats {
 	stats := newRegionStats()
-	regions := f(startKey, endKey)
 	for _, region := range regions {
 		stats.Observe(region)
 	}
diff --git a/tools/pd-ctl/pdctl/command/global.go b/tools/pd-ctl/pdctl/command/global.go
index 733455258fe..01c74d16026 100644
--- a/tools/pd-ctl/pdctl/command/global.go
+++ b/tools/pd-ctl/pdctl/command/global.go
@@ -139,7 +139,7 @@ func tryURLs(cmd *cobra.Command, f DoFunc) {
 		var u *url.URL
 		u, err = url.Parse(endpoint)
 		if err != nil {
-			cmd.Println("address format is wrong, should like 'http://127.0.0.1:2379' or '127.0.0.1:2379'")
+			cmd.Println("address format is wrong, should like 'http://127.0.0.1:2379' or '127.0.0.1:2379'", err)
 			os.Exit(1)
 		}
 		// tolerate some schemes that will be used by users, the TiKV SDK

From f21a264f7b36227954a0eb4917da4568b2eb6d23 Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 15:22:31 +0800
Subject: [PATCH 2/6] fix test

Signed-off-by: disksing <i@disksing.com>
---
 client/client_test.go | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/client/client_test.go b/client/client_test.go
index 4bdb43972c7..4fa492bb6fe 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -287,8 +287,8 @@ func (s *testClientSuite) TestScanRegions(c *C) {
 	region3 := core.NewRegionInfo(regions[3], nil)
 	s.srv.GetRaftCluster().HandleRegionHeartbeat(region3)
 
-	check := func(start []byte, limit int, expect []*metapb.Region) {
-		scanRegions, leaders, err := s.client.ScanRegions(context.Background(), start, limit)
+	check := func(start, end []byte, limit int, expect []*metapb.Region) {
+		scanRegions, leaders, err := s.client.ScanRegions(context.Background(), start, end, limit)
 		c.Assert(err, IsNil)
 		c.Assert(scanRegions, HasLen, len(expect))
 		c.Assert(leaders, HasLen, len(expect))
@@ -305,9 +305,11 @@ func (s *testClientSuite) TestScanRegions(c *C) {
 		}
 	}
 
-	check([]byte{0}, 10, regions)
-	check([]byte{1}, 5, regions[1:6])
-	check([]byte{100}, 1, nil)
+	check([]byte{0}, nil, 10, regions)
+	check([]byte{1}, nil, 5, regions[1:6])
+	check([]byte{100}, nil, 1, nil)
+	check([]byte{1}, []byte{6}, 0, regions[1:6])
+	check([]byte{1}, []byte{6}, 2, regions[1:3])
 }
 
 func (s *testClientSuite) TestGetRegionByID(c *C) {

From 204f7992a2828f1472922b9874f5bf7f53d46e7c Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 15:27:44 +0800
Subject: [PATCH 3/6] minor fix

Signed-off-by: disksing <i@disksing.com>
---
 tools/pd-ctl/pdctl/command/global.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/pd-ctl/pdctl/command/global.go b/tools/pd-ctl/pdctl/command/global.go
index 01c74d16026..733455258fe 100644
--- a/tools/pd-ctl/pdctl/command/global.go
+++ b/tools/pd-ctl/pdctl/command/global.go
@@ -139,7 +139,7 @@ func tryURLs(cmd *cobra.Command, f DoFunc) {
 		var u *url.URL
 		u, err = url.Parse(endpoint)
 		if err != nil {
-			cmd.Println("address format is wrong, should like 'http://127.0.0.1:2379' or '127.0.0.1:2379'", err)
+			cmd.Println("address format is wrong, should like 'http://127.0.0.1:2379' or '127.0.0.1:2379'")
 			os.Exit(1)
 		}
 		// tolerate some schemes that will be used by users, the TiKV SDK

From c441dcb73596bbf926780ac27fe5a4db137d1596 Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 15:29:58 +0800
Subject: [PATCH 4/6] minor fix

Signed-off-by: disksing <i@disksing.com>
---
 client/client_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client/client_test.go b/client/client_test.go
index 4fa492bb6fe..8569186ecb3 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -279,7 +279,7 @@ func (s *testClientSuite) TestScanRegions(c *C) {
 
 	// Wait for region heartbeats.
 	testutil.WaitUntil(c, func(c *C) bool {
-		scanRegions, _, err := s.client.ScanRegions(context.Background(), []byte{0}, 10)
+		scanRegions, _, err := s.client.ScanRegions(context.Background(), []byte{0}, nil, 10)
 		return err == nil && len(scanRegions) == 10
 	})
 

From aa46d5b249a4290175933502bd345a08f60d37c3 Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 17:09:39 +0800
Subject: [PATCH 5/6] update kvproto

Signed-off-by: disksing <i@disksing.com>
---
 go.mod | 4 +---
 go.sum | 4 ++--
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/go.mod b/go.mod
index 8de627b290b..22e7f98a70f 100644
--- a/go.mod
+++ b/go.mod
@@ -29,7 +29,7 @@ require (
 	github.com/pingcap/check v0.0.0-20190102082844-67f458068fc8
 	github.com/pingcap/errcode v0.0.0-20180921232412-a1a7271709d9
 	github.com/pingcap/failpoint v0.0.0-20190512135322-30cc7431d99c
-	github.com/pingcap/kvproto v0.0.0-20190516013202-4cf58ad90b6c
+	github.com/pingcap/kvproto v0.0.0-20190822090350-11ea838aedf7
 	github.com/pingcap/log v0.0.0-20190715063458-479153f07ebd
 	github.com/pkg/errors v0.8.1
 	github.com/prometheus/client_golang v0.8.0
@@ -47,5 +47,3 @@ require (
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
 	gopkg.in/natefinch/lumberjack.v2 v2.0.0
 )
-
-replace github.com/pingcap/kvproto => github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a
diff --git a/go.sum b/go.sum
index 028c659f6a9..311bbb5b65b 100644
--- a/go.sum
+++ b/go.sum
@@ -21,8 +21,6 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM=
 github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
-github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a h1:YV5474lam4SKIMro+Rjyi5qwro6VK9anvZTv8eAkY10=
-github.com/disksing/kvproto v0.0.0-20190822071505-1c5b11a4e77a/go.mod h1:QMdbTAXCHzzygQzqcG9uVUgU2fKeSN1GmfMiykdSzzY=
 github.com/dustin/go-humanize v0.0.0-20171111073723-bb3d318650d4/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
 github.com/dustin/go-humanize v0.0.0-20180421182945-02af3965c54e h1:Fw7ZmgiklsLh5EQWyHh1sumKSCG1+yjEctIpGKib87s=
 github.com/dustin/go-humanize v0.0.0-20180421182945-02af3965c54e/go.mod h1:HtrtbFcZ19U5GC7JDqmcUSB87Iq5E25KnS6fMYU6eOk=
@@ -110,6 +108,8 @@ github.com/pingcap/errors v0.11.0 h1:DCJQB8jrHbQ1VVlMFIrbj2ApScNNotVmkSNplu2yUt4
 github.com/pingcap/errors v0.11.0/go.mod h1:Oi8TUi2kEtXXLMJk9l1cGmz20kV3TaQ0usTwv5KuLY8=
 github.com/pingcap/failpoint v0.0.0-20190512135322-30cc7431d99c h1:hvQd3aOLKLF7xvRV6DzvPkKY4QXzfVbjU1BhW0d9yL8=
 github.com/pingcap/failpoint v0.0.0-20190512135322-30cc7431d99c/go.mod h1:DNS3Qg7bEDhU6EXNHF+XSv/PGznQaMJ5FWvctpm6pQI=
+github.com/pingcap/kvproto v0.0.0-20190822090350-11ea838aedf7 h1:tt24R4cv6GlvnmvkHNC1OrS/ETvXxbJkJ1Nrk4prtWo=
+github.com/pingcap/kvproto v0.0.0-20190822090350-11ea838aedf7/go.mod h1:QMdbTAXCHzzygQzqcG9uVUgU2fKeSN1GmfMiykdSzzY=
 github.com/pingcap/log v0.0.0-20190715063458-479153f07ebd h1:hWDol43WY5PGhsh3+8794bFHY1bPrmu6bTalpssCrGg=
 github.com/pingcap/log v0.0.0-20190715063458-479153f07ebd/go.mod h1:WpHUKhNZ18v116SvGrmjkA9CBhYmuUTKL+p8JC9ANEw=
 github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=

From 313b820df327f297a7600a7d5feaf8bf44992980 Mon Sep 17 00:00:00 2001
From: disksing <i@disksing.com>
Date: Thu, 22 Aug 2019 17:15:47 +0800
Subject: [PATCH 6/6] address comment

Signed-off-by: disksing <i@disksing.com>
---
 client/client.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/client/client.go b/client/client.go
index 543c037f433..e7a9d0d8a00 100644
--- a/client/client.go
+++ b/client/client.go
@@ -700,8 +700,8 @@ func (c *client) ScanRegions(ctx context.Context, key, endKey []byte, limit int)
 	resp, err := c.leaderClient().ScanRegions(ctx, &pdpb.ScanRegionsRequest{
 		Header:   c.requestHeader(),
 		StartKey: key,
-		Limit:    int32(limit),
 		EndKey:   endKey,
+		Limit:    int32(limit),
 	})
 	cancel()
 	if err != nil {