address comments

Signed-off-by: Ryan Leung <rleungx@gmail.com>
tikv · Oct 25, 2023 · 7e28af7 · 7e28af7
1 parent c209d34
commit 7e28af7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pkg/schedule/scatter/region_scatterer.go b/pkg/schedule/scatter/region_scatterer.go
@@ -189,8 +189,11 @@ func (r *RegionScatterer) ScatterRegionsByID(regionsID []uint64, group string, r
 		return 0, nil, errEmptyRegion
 	}
 	if len(regionsID) == 1 {
-		scatterSkipNoRegionCounter.Inc()
-		return 0, nil, errRegionNotFound
+		region := r.cluster.GetRegion(regionsID[0])
+		if region == nil {
+			scatterSkipNoRegionCounter.Inc()
+			return 0, nil, errRegionNotFound
+		}
 	}
 	failures := make(map[uint64]error, len(regionsID))
 	regions := make([]*core.RegionInfo, 0, len(regionsID))