address the comment

Signed-off-by: Ryan Leung <[email protected]>
tikv · Oct 17, 2022 · 3f5e02d · 3f5e02d
1 parent 8e18727
commit 3f5e02d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/server/core/basic_cluster.go b/server/core/basic_cluster.go
@@ -225,13 +225,14 @@ func (bc *BasicCluster) GetStoreRegions(storeID uint64) []*RegionInfo {
 // GetLeaderStoreByRegionID returns the leader store of the given region.
 func (bc *BasicCluster) GetLeaderStoreByRegionID(regionID uint64) *StoreInfo {
 	bc.Regions.mu.RLock()
-	bc.Stores.mu.RLock()
-	defer bc.Stores.mu.RUnlock()
-	defer bc.Regions.mu.RUnlock()
 	region := bc.Regions.GetRegion(regionID)
 	if region == nil || region.GetLeader() == nil {
 		return nil
 	}
+	bc.Regions.mu.RUnlock()
+
+	bc.Stores.mu.RLock()
+	defer bc.Stores.mu.RUnlock()
 	return bc.Stores.GetStore(region.GetLeader().GetStoreId())
 }