From b9da313d68a6a705e440b55c7251b72da1b62fd2 Mon Sep 17 00:00:00 2001 From: JmPotato Date: Fri, 3 Jul 2026 14:04:47 +0800 Subject: [PATCH] mockstore: implement LoadKeyspaceByID in PD mocks Signed-off-by: JmPotato --- internal/mockstore/mocktikv/pd.go | 7 ++++++- txnkv/transaction/test_util.go | 8 +++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/internal/mockstore/mocktikv/pd.go b/internal/mockstore/mocktikv/pd.go index e0192a3c13..65e6b60e0e 100644 --- a/internal/mockstore/mocktikv/pd.go +++ b/internal/mockstore/mocktikv/pd.go @@ -53,6 +53,7 @@ import ( "github.com/tikv/client-go/v2/util" pd "github.com/tikv/pd/client" pdgc "github.com/tikv/pd/client/clients/gc" + "github.com/tikv/pd/client/clients/metastorage" "github.com/tikv/pd/client/clients/router" "github.com/tikv/pd/client/clients/tso" "github.com/tikv/pd/client/constants" @@ -430,6 +431,10 @@ func (c *pdClient) LoadKeyspace(ctx context.Context, name string) (*keyspacepb.K return nil, nil } +func (c *pdClient) LoadKeyspaceByID(ctx context.Context, id uint32) (*keyspacepb.KeyspaceMeta, error) { + return nil, nil +} + func (c *pdClient) WatchKeyspaces(ctx context.Context) (chan []*keyspacepb.KeyspaceMeta, error) { return nil, nil } @@ -486,7 +491,7 @@ func (c *pdClient) GetLocalTSWithinKeyspaceAsync(ctx context.Context, dcLocation return nil } -func (c *pdClient) Watch(ctx context.Context, key []byte, opts ...opt.MetaStorageOption) (chan []*meta_storagepb.Event, error) { +func (c *pdClient) Watch(ctx context.Context, key []byte, opts ...opt.MetaStorageOption) (chan *metastorage.WatchResponse, error) { return nil, nil } diff --git a/txnkv/transaction/test_util.go b/txnkv/transaction/test_util.go index 27dd19827f..5c1b6b14bf 100644 --- a/txnkv/transaction/test_util.go +++ b/txnkv/transaction/test_util.go @@ -34,6 +34,7 @@ import ( "github.com/tikv/client-go/v2/util/async" pd "github.com/tikv/pd/client" "github.com/tikv/pd/client/clients/gc" + "github.com/tikv/pd/client/clients/metastorage" "github.com/tikv/pd/client/clients/router" "github.com/tikv/pd/client/clients/tso" "github.com/tikv/pd/client/opt" @@ -254,6 +255,11 @@ func (u *unimplementedPDClient) LoadKeyspace(ctx context.Context, name string) ( panic("unimplemented") } +// LoadKeyspaceByID implements pd.Client. +func (u *unimplementedPDClient) LoadKeyspaceByID(ctx context.Context, id uint32) (*keyspacepb.KeyspaceMeta, error) { + panic("unimplemented") +} + // LoadResourceGroups implements pd.Client. func (u *unimplementedPDClient) LoadResourceGroups(ctx context.Context) ([]*resource_manager.ResourceGroup, int64, error) { panic("unimplemented") @@ -335,7 +341,7 @@ func (u *unimplementedPDClient) UpdateServiceSafePointV2(ctx context.Context, ke } // Watch implements pd.Client. -func (u *unimplementedPDClient) Watch(ctx context.Context, key []byte, opts ...opt.MetaStorageOption) (chan []*meta_storagepb.Event, error) { +func (u *unimplementedPDClient) Watch(ctx context.Context, key []byte, opts ...opt.MetaStorageOption) (chan *metastorage.WatchResponse, error) { panic("unimplemented") }