diff --git a/internal/mockstore/mocktikv/pd.go b/internal/mockstore/mocktikv/pd.go index 769f2c89..e5cec547 100644 --- a/internal/mockstore/mocktikv/pd.go +++ b/internal/mockstore/mocktikv/pd.go @@ -41,6 +41,7 @@ import ( "sync" "time" + "github.com/pingcap/kvproto/pkg/keyspacepb" "github.com/pingcap/kvproto/pkg/metapb" "github.com/pingcap/kvproto/pkg/pdpb" "github.com/pkg/errors" @@ -251,3 +252,11 @@ func (c *pdClient) GetLeaderAddr() string { return "mockpd" } func (c *pdClient) UpdateOption(option pd.DynamicOption, value interface{}) error { return nil } + +func (c *pdClient) LoadKeyspace(ctx context.Context, name string) (*keyspacepb.KeyspaceMeta, error) { + return nil, nil +} + +func (c *pdClient) WatchKeyspaces(ctx context.Context) (chan []*keyspacepb.KeyspaceMeta, error) { + return nil, nil +}