Lorenz Herzberger 76ec1fbb7d
feat: add migration for activated ta counter
Signed-off-by: Lorenz Herzberger <lorenzherzberger@gmail.com>
2024-07-22 10:58:40 +02:00

30 lines
717 B
Go

package v3
import (
"encoding/binary"
"github.com/cosmos/cosmos-sdk/codec"
"github.com/cosmos/cosmos-sdk/store/prefix"
storetypes "github.com/cosmos/cosmos-sdk/store/types"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/planetmint/planetmint-go/x/machine/types"
)
func MigrateStore(ctx sdk.Context, storeKey storetypes.StoreKey, cdc codec.BinaryCodec) error {
store := prefix.NewStore(ctx.KVStore(storeKey), types.KeyPrefix(types.ActivatedTACounterPrefix))
count := uint64(0)
iterator := store.Iterator(nil, nil)
defer iterator.Close()
for ; iterator.Valid(); iterator.Next() {
count++
}
bz := make([]byte, 8)
binary.BigEndian.PutUint64(bz, count)
store.Set([]byte{1}, bz)
return nil
}