region_cache: filter peers on tombstone or dropped stores (#24726) (#…

…25839)
pingcap · Jun 30, 2021 · ac6d86b · ac6d86b
1 parent e1388a2
commit ac6d86b
Show file tree

Hide file tree

Showing 3 changed files with 289 additions and 102 deletions.
diff --git a/store/mockstore/mocktikv/pd.go b/store/mockstore/mocktikv/pd.go
@@ -15,6 +15,7 @@ package mocktikv
 
 import (
 	"context"
+	"fmt"
 	"math"
 	"sync"
 	"time"
@@ -126,6 +127,13 @@ func (c *pdClient) GetStore(ctx context.Context, storeID uint64) (*metapb.Store,
 	default:
 	}
 	store := c.cluster.GetStore(storeID)
+	// It's same as PD's implementation.
+	if store == nil {
+		return nil, fmt.Errorf("invalid store ID %d, not found", storeID)
+	}
+	if store.GetState() == metapb.StoreState_Tombstone {
+		return nil, nil
+	}
 	return store, nil
 }