Expose VersionExists (#400)

Add an interface function `VersionExists` to store types

local sei integration

go.mod

@@ -37,7 +37,7 @@ require (
 	github.com/regen-network/cosmos-proto v0.3.1
 	github.com/rs/zerolog v1.30.0
 	github.com/savaki/jq v0.0.0-20161209013833-0e6baecebbf8
-	github.com/sei-protocol/sei-db v0.0.22
+	github.com/sei-protocol/sei-db v0.0.27-0.20240123064153-d6dfa112e760
 	github.com/sei-protocol/sei-tm-db v0.0.5
 	github.com/spf13/cast v1.5.0
 	github.com/spf13/cobra v1.6.1

server/mock/store.go

@@ -226,6 +226,10 @@ func (kv kvStore) ReverseSubspaceIterator(prefix []byte) sdk.Iterator {
 	panic("not implemented")
 }
 
+func (kv kvStore) VersionExists(version int64) bool {
+	panic("not implemented")
+}
+
 func NewCommitMultiStore() sdk.CommitMultiStore {
 	return multiStore{kv: make(map[sdk.StoreKey]kvStore)}
 }

store/cachekv/store.go

@@ -197,6 +197,10 @@ func (store *Store) iterator(start, end []byte, ascending bool) types.Iterator {
 	return NewCacheMergeIterator(parent, cache, ascending, store.storeKey, store.eventManager)
 }
 
+func (store *Store) VersionExists(version int64) bool {
+	return store.parent.VersionExists(version)
+}
+
 func findStartIndex(strL []string, startQ string) int {
 	// Modified binary search to find the very first element in >=startQ.
 	if len(strL) == 0 {

store/dbadapter/store.go

@@ -95,5 +95,9 @@ func (dsa Store) CacheWrapWithListeners(storeKey types.StoreKey, listeners []typ
 	return cachekv.NewStore(listenkv.NewStore(dsa, storeKey, listeners), storeKey, types.DefaultCacheSizeLimit)
 }
 
+func (dsa Store) VersionExists(version int64) bool {
+	panic("no versioning for dbadater")
+}
+
 // dbm.DB implements KVStore so we can CacheKVStore it.
 var _ types.KVStore = Store{}

store/gaskv/store.go

@@ -126,6 +126,10 @@ func (gs *Store) iterator(start, end []byte, ascending bool) types.Iterator {
 	return gi
 }
 
+func (gs *Store) VersionExists(version int64) bool {
+	return gs.parent.VersionExists(version)
+}
+
 type gasIterator struct {
 	gasMeter  types.GasMeter
 	gasConfig types.GasConfig

store/listenkv/store.go

@@ -81,6 +81,10 @@ func (s *Store) iterator(start, end []byte, ascending bool) types.Iterator {
 	return newTraceIterator(parent, s.listeners)
 }
 
+func (s *Store) VersionExists(version int64) bool {
+	return s.parent.VersionExists(version)
+}
+
 type listenIterator struct {
 	parent    types.Iterator
 	listeners []types.WriteListener

store/prefix/store.go

@@ -124,6 +124,10 @@ func (s Store) ReverseIterator(start, end []byte) types.Iterator {
 	return newPrefixIterator(s.prefix, start, end, iter)
 }
 
+func (s Store) VersionExists(version int64) bool {
+	return s.parent.VersionExists(version)
+}
+
 var _ types.Iterator = (*prefixIterator)(nil)
 
 type prefixIterator struct {

store/tracekv/store.go

@@ -182,6 +182,10 @@ func (tkv *Store) CacheWrapWithListeners(_ types.StoreKey, _ []types.WriteListen
 	panic("cannot CacheWrapWithListeners a TraceKVStore")
 }
 
+func (tkv *Store) VersionExists(version int64) bool {
+	return tkv.parent.VersionExists(version)
+}
+
 // writeOperation writes a KVStore operation to the underlying io.Writer as
 // JSON-encoded data where the key/value pair is base64 encoded.
 func writeOperation(w io.Writer, op operation, tc types.TraceContext, key, value []byte) {

store/types/store.go

@@ -253,6 +253,8 @@ type KVStore interface {
 	ReverseIterator(start, end []byte) Iterator
 
 	GetWorkingHash() ([]byte, error)
+
+	VersionExists(version int64) bool
 }
 
 // Iterator is an alias db's Iterator for convenience.

storev2/commitment/store.go

@@ -177,3 +177,8 @@ func (st *Store) Query(req abci.RequestQuery) (res abci.ResponseQuery) {
 
 	return res
 }
+
+func (st *Store) VersionExists(version int64) bool {
+	// one version per SC tree
+	return version == st.tree.Version()
+}

storev2/state/store.go

@@ -1,10 +1,11 @@
 package state
 
 import (
-	"cosmossdk.io/errors"
 	"fmt"
 	"io"
 
+	"cosmossdk.io/errors"
+
 	"github.com/cosmos/cosmos-sdk/store/cachekv"
 	"github.com/cosmos/cosmos-sdk/store/listenkv"
 	"github.com/cosmos/cosmos-sdk/store/tracekv"
@@ -125,3 +126,11 @@ func (st *Store) Query(req abci.RequestQuery) (res abci.ResponseQuery) {
 
 	return res
 }
+
+func (st *Store) VersionExists(version int64) bool {
+	earliest, err := st.store.GetEarliestVersion()
+	if err != nil {
+		panic(err)
+	}
+	return version >= earliest
+}

x/bank/keeper/keeper.go

@@ -51,6 +51,8 @@ type Keeper interface {
 	DelegateCoins(ctx sdk.Context, delegatorAddr, moduleAccAddr sdk.AccAddress, amt sdk.Coins) error
 	UndelegateCoins(ctx sdk.Context, moduleAccAddr, delegatorAddr sdk.AccAddress, amt sdk.Coins) error
 
+	GetStoreKey() sdk.StoreKey
+
 	types.QueryServer
 }
 
@@ -677,6 +679,10 @@ func (k BaseKeeper) trackUndelegation(ctx sdk.Context, addr sdk.AccAddress, amt
 	return nil
 }
 
+func (k BaseKeeper) GetStoreKey() sdk.StoreKey {
+	return k.storeKey
+}
+
 // IterateTotalSupply iterates over the total supply calling the given cb (callback) function
 // with the balance of each coin.
 // The iteration stops if the callback returns true.