add GetObjKVStore

crypto-org-chain · Jul 22, 2024 · 45aa498 · 45aa498
1 parent be52147
commit 45aa498
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/store/rootmulti/store.go b/store/rootmulti/store.go
@@ -246,6 +246,15 @@ func (rs *Store) GetKVStore(key types.StoreKey) types.KVStore {
 	return s
 }
 
+// GetObjKVStore returns a mounted ObjKVStore for a given StoreKey.
+func (rs *Store) GetObjKVStore(key types.StoreKey) types.ObjKVStore {
+	s, ok := rs.GetStore(key).(types.ObjKVStore)
+	if !ok {
+		panic(fmt.Sprintf("store with key %v is not KVStore", key))
+	}
+	return s
+}
+
 // Implements interface MultiStore
 func (rs *Store) TracingEnabled() bool {
 	return false

diff --git a/versiondb/multistore.go b/versiondb/multistore.go
@@ -99,6 +99,11 @@ func (s *MultiStore) GetKVStore(storeKey types.StoreKey) types.KVStore {
 	return s.GetStore(storeKey).(types.KVStore)
 }
 
+// GetObjKVStore implements `MultiStore` interface
+func (s *MultiStore) GetObjKVStore(storeKey types.StoreKey) types.ObjKVStore {
+	return s.GetStore(storeKey).(types.ObjKVStore)
+}
+
 // SetTracer sets the tracer for the MultiStore that the underlying
 // stores will utilize to trace operations. A MultiStore is returned.
 func (s *MultiStore) SetTracer(w io.Writer) types.MultiStore {