kvstore use change set api

store/memiavlstore/store.go

@@ -5,6 +5,7 @@ import (
 	"io"
 
 	"github.com/cosmos/cosmos-sdk/store/tracekv"
+	"github.com/cosmos/iavl"
 	"github.com/crypto-org-chain/cronos/memiavl"
 	"github.com/tendermint/tendermint/libs/log"
 
@@ -23,6 +24,8 @@ var (
 type Store struct {
 	tree   memiavl.Tree
 	logger log.Logger
+
+	changeSet memiavl.ChangeSet
 }
 
 func LoadStoreWithInitialVersion(dir string, logger log.Logger, initialVersion int64) (types.CommitKVStore, error) {
@@ -34,10 +37,12 @@ func LoadStoreWithInitialVersion(dir string, logger log.Logger, initialVersion i
 }
 
 func (st *Store) Commit() types.CommitID {
+	st.tree.ApplyChangeSet(&st.changeSet)
 	hash, version, err := st.tree.SaveVersion(true)
 	if err != nil {
 		panic(err)
 	}
+	st.changeSet.Pairs = st.changeSet.Pairs[:0]
 
 	return types.CommitID{
 		Version: version,
@@ -79,8 +84,13 @@ func (st *Store) CacheWrapWithTrace(w io.Writer, tc types.TraceContext) types.Ca
 	return cachekv.NewStore(tracekv.NewStore(st, w, tc))
 }
 
+// Implements types.KVStore.
+//
+// we assume Set is only called in `Commit`, so the written state is only visible after commit.
 func (st *Store) Set(key, value []byte) {
-	st.tree.Set(key, value)
+	st.changeSet.Pairs = append(st.changeSet.Pairs, &iavl.KVPair{
+		Key: key, Value: value,
+	})
 }
 
 // Implements types.KVStore.
@@ -94,8 +104,12 @@ func (st *Store) Has(key []byte) bool {
 }
 
 // Implements types.KVStore.
+//
+// we assume Delete is only called in `Commit`, so the written state is only visible after commit.
 func (st *Store) Delete(key []byte) {
-	st.tree.Remove(key)
+	st.changeSet.Pairs = append(st.changeSet.Pairs, &iavl.KVPair{
+		Key: key, Delete: true,
+	})
 }
 
 func (st *Store) Iterator(start, end []byte) types.Iterator {