【cosmos-sdk-笔记】cosmovisor 简单实现(代码:修改存储结构)

git clone https://github.com/taoyuans/mingxie.git

cd mingxie
git checkout main_v1_2

package v011 // migrations.go
import (
 "fmt"
 "mingxie/x/tvote/types"

 "github.com/cosmos/cosmos-sdk/codec"
 storetypes "github.com/cosmos/cosmos-sdk/store/types"
 sdk "github.com/cosmos/cosmos-sdk/types"
 paramtypes "github.com/cosmos/cosmos-sdk/x/params/types"
)

func UpdateParams(ctx sdk.Context, paramStore *paramtypes.Subspace) error {

 // paramStore.Set(ctx, []byte(types.VoterKey), "foobar")

 return nil
}
func MigrateStore(ctx sdk.Context, storeKey storetypes.StoreKey, cdc codec.BinaryCodec) error {
 store := ctx.KVStore(storeKey)

 return migrateValuesWithPrefix(store, cdc)
}
func migrateValuesWithPrefix(store sdk.KVStore, cdc codec.BinaryCodec) error {
 oldStoreIter := store.Iterator(nil, nil)

 for ; oldStoreIter.Valid(); oldStoreIter.Next() {
  oldKey := oldStoreIter.Key()
  oldVal := store.Get(oldKey)

  newKey, newVal := migrateValue(cdc, oldKey, oldVal)
  store.Set(newKey, newVal)
  // store.Delete(oldKey) // Delete old key, value
 }

 return nil
}
func migrateValue(cdc codec.BinaryCodec, oldKey []byte, oldVal []byte) (newKey []byte, newVal []byte) {
 var newVoter types.Voter
 err := cdc.Unmarshal(oldVal, &newVoter)
 if err != nil {
  fmt.Println(err)
 }

 newVoter.Level = uint32(newVoter.Age - newVoter.Vid)

 newKey = oldKey
 newVal = cdc.MustMarshal(&newVoter)
 return
}

package keeper // migrations.go
import (
 v2 "mingxie/x/tvote/migrations/v2"

 sdk "github.com/cosmos/cosmos-sdk/types"
)

type Migrator struct {
 keeper Keeper
}

func NewMigrator(keeper Keeper) Migrator {
 return Migrator{keeper: keeper}
}
func (m Migrator) Migrate1to2(ctx sdk.Context) error {
 if err := v2.UpdateParams(ctx, &m.keeper.paramstore); err != nil {
  return err
 }

 return v2.MigrateStore(ctx, m.keeper.storeKey, m.keeper.cdc)
}

// RegisterServices registers a gRPC query service to respond to the module-specific gRPC queries
func (am AppModule) RegisterServices(cfg module.Configurator) {
 migrator := keeper.NewMigrator(am.keeper)
 if err := cfg.RegisterMigration(types.ModuleName, 1, migrator.Migrate1to2); err != nil {
  panic(fmt.Errorf("failed to migrate %s to v011: %w", types.ModuleName, err))
 }

 types.RegisterMsgServer(cfg.MsgServer(), keeper.NewMsgServerImpl(am.keeper))
 types.RegisterQueryServer(cfg.QueryServer(), am.keeper)
}