Split the keys to search

Signed-off-by: JmPotato <[email protected]>
tikv · Jan 17, 2025 · de256ce · de256ce
1 parent 108b0a3
commit de256ce
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/server/grpc_service.go b/server/grpc_service.go
@@ -1532,14 +1532,12 @@ func (s *GrpcServer) GetRegionByID(ctx context.Context, request *pdpb.GetRegionB
 
 // QueryRegion provides a stream processing of the region query.
 func (s *GrpcServer) QueryRegion(stream pdpb.PD_QueryRegionServer) error {
-	if s.GetServiceMiddlewarePersistOptions().IsGRPCRateLimitEnabled() {
-		fName := currentFunction()
-		limiter := s.GetGRPCRateLimiter()
-		if done, err := limiter.Allow(fName); err == nil {
-			defer done()
-		} else {
-			return err
-		}
+	done, err := s.rateLimitCheck()
+	if err != nil {
+		return err
+	}
+	if done != nil {
+		defer done()
 	}
 
 	for {