Refine the kv.Storage interface

Signed-off-by: JmPotato <[email protected]>
tikv · Nov 7, 2023 · acf7555 · acf7555
1 parent 635aa6e
commit acf7555
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tikv/interface.go b/tikv/interface.go
@@ -41,6 +41,7 @@ import (
 	"github.com/tikv/client-go/v2/oracle"
 	"github.com/tikv/client-go/v2/tikvrpc"
 	"github.com/tikv/client-go/v2/txnkv/txnlock"
+	pdhttp "github.com/tikv/pd/client/http"
 )
 
 // Storage represent the kv.Storage runs on TiKV.
@@ -69,6 +70,9 @@ type Storage interface {
 	// GetTiKVClient gets the TiKV client.
 	GetTiKVClient() Client
 
+	// GetPDHTTPClient gets the PD HTTP client.
+	GetPDHTTPClient() pdhttp.Client
+
 	// Closed returns the closed channel.
 	Closed() <-chan struct{}
 

diff --git a/tikv/kv.go b/tikv/kv.go
@@ -148,6 +148,8 @@ type KVStore struct {
 	gP     Pool
 }
 
+var _ Storage = (*KVStore)(nil)
+
 // Go run the function in a separate goroutine.
 func (s *KVStore) Go(f func()) error {
 	return s.gP.Run(f)