support new option for WithAllowRouterServiceHandle

Signed-off-by: 童剑 <1045931706@qq.com>

Author: bufferflies

go.mod

@@ -1,6 +1,10 @@
 module github.com/tikv/client-go/v2
 
-go 1.23.0
+go 1.23.12
+
+toolchain go1.24.1
+
+replace github.com/tikv/pd/client => github.com/bufferflies/pd/client v0.0.0-20251113073927-030b4e3ceaca
 
 require (
 	github.com/VividCortex/ewma v1.2.0

go.sum

@@ -6,6 +6,8 @@ github.com/VividCortex/ewma v1.2.0/go.mod h1:nz4BbCtbLyFDeC9SUHbtcT5644juEuWfUAU
 github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=
 github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
 github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
+github.com/bufferflies/pd/client v0.0.0-20251113073927-030b4e3ceaca h1:X2vn4hnHMNxcAV4fFEE+3eVYvjp9F9Ll3LWciYaRGNM=
+github.com/bufferflies/pd/client v0.0.0-20251113073927-030b4e3ceaca/go.mod h1:3Zt8kRLZHAEGaU65Eg3+8MrlRFH6gpSJ00WdAXn8zTY=
 github.com/cakturk/go-netstat v0.0.0-20200220111822-e5b49efee7a5 h1:BjkPE3785EwPhhyuFkbINB+2a1xATwk8SNDWnJiD41g=
 github.com/cakturk/go-netstat v0.0.0-20200220111822-e5b49efee7a5/go.mod h1:jtAfVaU/2cu1+wdSRPWE2c1N2qeAA3K4RH9pYgqwets=
 github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=

integration_tests/go.mod

@@ -2,6 +2,8 @@ module integration_tests
 
 go 1.24.1
 
+replace github.com/tikv/pd/client => github.com/bufferflies/pd/client v0.0.0-20251113073927-030b4e3ceaca
+
 require (
 	github.com/google/uuid v1.6.0
 	github.com/ninedraft/israce v0.0.3

integration_tests/go.sum

@@ -884,6 +884,7 @@ github.com/blacktear23/go-proxyprotocol v1.0.6 h1:eTt6UMpEnq59NjON49b3Cay8Dm0sCs
 github.com/blacktear23/go-proxyprotocol v1.0.6/go.mod h1:FSCbgnRZrQXazBLL5snfBbrcFSMtcmUDhSRb9OfFA1o=
 github.com/boombuler/barcode v1.0.0/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8=
 github.com/boombuler/barcode v1.0.1/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8=
+github.com/bufferflies/pd/client v0.0.0-20251113073927-030b4e3ceaca/go.mod h1:3Zt8kRLZHAEGaU65Eg3+8MrlRFH6gpSJ00WdAXn8zTY=
 github.com/cakturk/go-netstat v0.0.0-20200220111822-e5b49efee7a5 h1:BjkPE3785EwPhhyuFkbINB+2a1xATwk8SNDWnJiD41g=
 github.com/cakturk/go-netstat v0.0.0-20200220111822-e5b49efee7a5/go.mod h1:jtAfVaU/2cu1+wdSRPWE2c1N2qeAA3K4RH9pYgqwets=
 github.com/carlmjohnson/flagext v0.21.0 h1:/c4uK3ie786Z7caXLcIMvePNSSiH3bQVGDvmGLMme60=

internal/locate/region_cache.go

@@ -1638,7 +1638,7 @@ func (c *RegionCache) findRegionByKey(bo *retry.Backoffer, key []byte, isEndKey
 	if r == nil || expired {
 		// load region when it is not exists or expired.
 		observeLoadRegion(tag, r, expired, 0)
-		lr, err := c.loadRegion(bo, key, isEndKey, opt.WithAllowFollowerHandle())
+		lr, err := c.loadRegion(bo, key, isEndKey, opt.WithAllowFollowerHandle(), opt.WithAllowRouterServiceHandle())
 		if err != nil {
 			// no region data, return error if failure.
 			return nil, err
@@ -2447,6 +2447,7 @@ func (c *RegionCache) batchScanRegions(bo *retry.Backoffer, keyRanges []router.K
 		start := time.Now()
 		pdOpts := []opt.GetRegionOption{
 			opt.WithAllowFollowerHandle(),
+			opt.WithAllowRouterServiceHandle(),
 			opt.WithOutputMustContainAllKeyRange(),
 		}
 		if batchOpt.needBuckets {

internal/mockstore/mocktikv/pd.go

@@ -593,6 +593,21 @@ type gcStatesClient struct {
 	keyspaceID uint32
 }
 
+func (c gcStatesClient) SetGlobalGCBarrier(ctx context.Context, barrierID string, barrierTS uint64, ttl time.Duration) (*pdgc.GlobalGCBarrierInfo, error) {
+	//TODO implement me
+	return nil, errors.New("unimplemented")
+}
+
+func (c gcStatesClient) GetAllKeyspacesGCStates(ctx context.Context) (pdgc.ClusterGCStates, error) {
+	//TODO implement me
+	return pdgc.ClusterGCStates{}, errors.New("unimplemented")
+}
+
+// DeleteGlobalGCBarrier deletes a global GC barrier.
+func (c gcStatesClient) DeleteGlobalGCBarrier(_ context.Context, _ string) (*pdgc.GlobalGCBarrierInfo, error) {
+	return nil, errors.New("unimplemented")
+}
+
 func (c gcStatesClient) SetGCBarrier(ctx context.Context, barrierID string, barrierTS uint64, ttl time.Duration) (*pdgc.GCBarrierInfo, error) {
 	if c.keyspaceID != constants.NullKeyspaceID {
 		panic("unimplemented")