Sim refactor 1: move store decoders to modules

simapp/app.go

@@ -110,6 +110,7 @@ func NewSimApp(
 ) *SimApp {
 
 	cdc := MakeCodec()
+	storeDecoderRegistry := make(sdk.StoreDecoderRegistry)
 
 	bApp := bam.NewBaseApp(appName, logger, db, auth.DefaultTxDecoder(cdc), baseAppOptions...)
 	bApp.SetCommitMultiStoreTracer(traceStore)
@@ -194,6 +195,7 @@ func NewSimApp(
 
 	app.mm.RegisterInvariants(&app.crisisKeeper)
 	app.mm.RegisterRoutes(app.Router(), app.QueryRouter())
+	app.mm.RegisterStoreDecoders(storeDecoderRegistry)
 
 	// initialize stores
 	app.MountKVStores(keys)

simapp/utils.go

@@ -2,16 +2,13 @@
 package simapp
 
 import (
-	"bytes"
-	"encoding/binary"
 	"encoding/json"
 	"fmt"
 	"io"
 	"io/ioutil"
 	"math/rand"
 	"time"
 
-	"github.com/tendermint/tendermint/crypto"
 	"github.com/tendermint/tendermint/crypto/secp256k1"
 	cmn "github.com/tendermint/tendermint/libs/common"
 	"github.com/tendermint/tendermint/libs/log"
@@ -22,15 +19,22 @@ import (
 	"github.com/cosmos/cosmos-sdk/codec"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/cosmos/cosmos-sdk/x/auth"
+	authsim "github.com/cosmos/cosmos-sdk/x/auth/simulation"
 	"github.com/cosmos/cosmos-sdk/x/bank"
 	"github.com/cosmos/cosmos-sdk/x/distribution"
+	distributionsim "github.com/cosmos/cosmos-sdk/x/distribution/simulation"
 	"github.com/cosmos/cosmos-sdk/x/genaccounts"
 	"github.com/cosmos/cosmos-sdk/x/gov"
+	govsim "github.com/cosmos/cosmos-sdk/x/gov/simulation"
 	"github.com/cosmos/cosmos-sdk/x/mint"
+	mintsim "github.com/cosmos/cosmos-sdk/x/mint/simulation"
 	"github.com/cosmos/cosmos-sdk/x/simulation"
 	"github.com/cosmos/cosmos-sdk/x/slashing"
+	slashingsim "github.com/cosmos/cosmos-sdk/x/slashing/simulation"
 	"github.com/cosmos/cosmos-sdk/x/staking"
+	stakingsim "github.com/cosmos/cosmos-sdk/x/staking/simulation"
 	"github.com/cosmos/cosmos-sdk/x/supply"
+	supplysim "github.com/cosmos/cosmos-sdk/x/supply/simulation"
 )
 
 // List of available flags for the simulator
@@ -500,7 +504,11 @@ func GenStakingGenesisState(
 
 // GetSimulationLog unmarshals the KVPair's Value to the corresponding type based on the
 // each's module store key and the prefix bytes of the KVPair's key.
-func GetSimulationLog(storeName string, cdcA, cdcB *codec.Codec, kvs []cmn.KVPair) (log string) {
+func GetSimulationLog(sdr sdk.StoreDecoderRegistry, storeName string, cdcA, cdcB *codec.Codec, kvs []cmn.KVPair) (log string) {
+	if kvs%2 != 0 {
+		panic("KVPairs are not multiple of 2. There should be one for each app store")
+	}
+
 	var kvA, kvB cmn.KVPair
 	for i := 0; i < len(kvs); i += 2 {
 		kvA = kvs[i]
@@ -511,231 +519,13 @@ func GetSimulationLog(storeName string, cdcA, cdcB *codec.Codec, kvs []cmn.KVPai
 			continue
 		}
 
-		switch storeName {
-		case auth.StoreKey:
-			log += DecodeAccountStore(cdcA, cdcB, kvA, kvB)
-		case mint.StoreKey:
-			log += DecodeMintStore(cdcA, cdcB, kvA, kvB)
-		case staking.StoreKey:
-			log += DecodeStakingStore(cdcA, cdcB, kvA, kvB)
-		case slashing.StoreKey:
-			log += DecodeSlashingStore(cdcA, cdcB, kvA, kvB)
-		case gov.StoreKey:
-			log += DecodeGovStore(cdcA, cdcB, kvA, kvB)
-		case distribution.StoreKey:
-			log += DecodeDistributionStore(cdcA, cdcB, kvA, kvB)
-		case supply.StoreKey:
-			log += DecodeSupplyStore(cdcA, cdcB, kvA, kvB)
-		default:
+		decoder, ok := sdr[storeName]
+		if ok {
+			log += decoder(cdcA, cdcB, kvA, kvB)
+		} else {
 			log += fmt.Sprintf("store A %X => %X\nstore B %X => %X\n", kvA.Key, kvA.Value, kvB.Key, kvB.Value)
 		}
 	}
 
 	return
 }
-
-// DecodeAccountStore unmarshals the KVPair's Value to the corresponding auth type
-func DecodeAccountStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], auth.AddressStoreKeyPrefix):
-		var accA, accB auth.Account
-		cdcA.MustUnmarshalBinaryBare(kvA.Value, &accA)
-		cdcB.MustUnmarshalBinaryBare(kvB.Value, &accB)
-		return fmt.Sprintf("%v\n%v", accA, accB)
-	case bytes.Equal(kvA.Key, auth.GlobalAccountNumberKey):
-		var globalAccNumberA, globalAccNumberB uint64
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &globalAccNumberA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &globalAccNumberB)
-		return fmt.Sprintf("GlobalAccNumberA: %d\nGlobalAccNumberB: %d", globalAccNumberA, globalAccNumberB)
-	default:
-		panic(fmt.Sprintf("invalid account key %X", kvA.Key))
-	}
-}
-
-// DecodeMintStore unmarshals the KVPair's Value to the corresponding mint type
-func DecodeMintStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key, mint.MinterKey):
-		var minterA, minterB mint.Minter
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &minterA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &minterB)
-		return fmt.Sprintf("%v\n%v", minterA, minterB)
-	default:
-		panic(fmt.Sprintf("invalid mint key %X", kvA.Key))
-	}
-}
-
-// DecodeDistributionStore unmarshals the KVPair's Value to the corresponding distribution type
-func DecodeDistributionStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], distribution.FeePoolKey):
-		var feePoolA, feePoolB distribution.FeePool
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &feePoolA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &feePoolB)
-		return fmt.Sprintf("%v\n%v", feePoolA, feePoolB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.ProposerKey):
-		return fmt.Sprintf("%v\n%v", sdk.ConsAddress(kvA.Value), sdk.ConsAddress(kvB.Value))
-
-	case bytes.Equal(kvA.Key[:1], distribution.ValidatorOutstandingRewardsPrefix):
-		var rewardsA, rewardsB distribution.ValidatorOutstandingRewards
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &rewardsA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &rewardsB)
-		return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.DelegatorWithdrawAddrPrefix):
-		return fmt.Sprintf("%v\n%v", sdk.AccAddress(kvA.Value), sdk.AccAddress(kvB.Value))
-
-	case bytes.Equal(kvA.Key[:1], distribution.DelegatorStartingInfoPrefix):
-		var infoA, infoB distribution.DelegatorStartingInfo
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &infoA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &infoB)
-		return fmt.Sprintf("%v\n%v", infoA, infoB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.ValidatorHistoricalRewardsPrefix):
-		var rewardsA, rewardsB distribution.ValidatorHistoricalRewards
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &rewardsA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &rewardsB)
-		return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.ValidatorCurrentRewardsPrefix):
-		var rewardsA, rewardsB distribution.ValidatorCurrentRewards
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &rewardsA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &rewardsB)
-		return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.ValidatorAccumulatedCommissionPrefix):
-		var commissionA, commissionB distribution.ValidatorAccumulatedCommission
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &commissionA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &commissionB)
-		return fmt.Sprintf("%v\n%v", commissionA, commissionB)
-
-	case bytes.Equal(kvA.Key[:1], distribution.ValidatorSlashEventPrefix):
-		var eventA, eventB distribution.ValidatorSlashEvent
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &eventA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &eventB)
-		return fmt.Sprintf("%v\n%v", eventA, eventB)
-
-	default:
-		panic(fmt.Sprintf("invalid distribution key prefix %X", kvA.Key[:1]))
-	}
-}
-
-// DecodeStakingStore unmarshals the KVPair's Value to the corresponding staking type
-func DecodeStakingStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], staking.LastTotalPowerKey):
-		var powerA, powerB sdk.Int
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &powerA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &powerB)
-		return fmt.Sprintf("%v\n%v", powerA, powerB)
-
-	case bytes.Equal(kvA.Key[:1], staking.ValidatorsKey):
-		var validatorA, validatorB staking.Validator
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &validatorA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &validatorB)
-		return fmt.Sprintf("%v\n%v", validatorA, validatorB)
-
-	case bytes.Equal(kvA.Key[:1], staking.LastValidatorPowerKey),
-		bytes.Equal(kvA.Key[:1], staking.ValidatorsByConsAddrKey),
-		bytes.Equal(kvA.Key[:1], staking.ValidatorsByPowerIndexKey):
-		return fmt.Sprintf("%v\n%v", sdk.ValAddress(kvA.Value), sdk.ValAddress(kvB.Value))
-
-	case bytes.Equal(kvA.Key[:1], staking.DelegationKey):
-		var delegationA, delegationB staking.Delegation
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &delegationA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &delegationB)
-		return fmt.Sprintf("%v\n%v", delegationA, delegationB)
-
-	case bytes.Equal(kvA.Key[:1], staking.UnbondingDelegationKey),
-		bytes.Equal(kvA.Key[:1], staking.UnbondingDelegationByValIndexKey):
-		var ubdA, ubdB staking.UnbondingDelegation
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &ubdA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &ubdB)
-		return fmt.Sprintf("%v\n%v", ubdA, ubdB)
-
-	case bytes.Equal(kvA.Key[:1], staking.RedelegationKey),
-		bytes.Equal(kvA.Key[:1], staking.RedelegationByValSrcIndexKey):
-		var redA, redB staking.Redelegation
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &redA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &redB)
-		return fmt.Sprintf("%v\n%v", redA, redB)
-
-	default:
-		panic(fmt.Sprintf("invalid staking key prefix %X", kvA.Key[:1]))
-	}
-}
-
-// DecodeSlashingStore unmarshals the KVPair's Value to the corresponding slashing type
-func DecodeSlashingStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], slashing.ValidatorSigningInfoKey):
-		var infoA, infoB slashing.ValidatorSigningInfo
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &infoA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &infoB)
-		return fmt.Sprintf("%v\n%v", infoA, infoB)
-
-	case bytes.Equal(kvA.Key[:1], slashing.ValidatorMissedBlockBitArrayKey):
-		var missedA, missedB bool
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &missedA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &missedB)
-		return fmt.Sprintf("missedA: %v\nmissedB: %v", missedA, missedB)
-
-	case bytes.Equal(kvA.Key[:1], slashing.AddrPubkeyRelationKey):
-		var pubKeyA, pubKeyB crypto.PubKey
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &pubKeyA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &pubKeyB)
-		bechPKA := sdk.MustBech32ifyAccPub(pubKeyA)
-		bechPKB := sdk.MustBech32ifyAccPub(pubKeyB)
-		return fmt.Sprintf("PubKeyA: %s\nPubKeyB: %s", bechPKA, bechPKB)
-
-	default:
-		panic(fmt.Sprintf("invalid slashing key prefix %X", kvA.Key[:1]))
-	}
-}
-
-// DecodeGovStore unmarshals the KVPair's Value to the corresponding gov type
-func DecodeGovStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], gov.ProposalsKeyPrefix):
-		var proposalA, proposalB gov.Proposal
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &proposalA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &proposalB)
-		return fmt.Sprintf("%v\n%v", proposalA, proposalB)
-
-	case bytes.Equal(kvA.Key[:1], gov.ActiveProposalQueuePrefix),
-		bytes.Equal(kvA.Key[:1], gov.InactiveProposalQueuePrefix),
-		bytes.Equal(kvA.Key[:1], gov.ProposalIDKey):
-		proposalIDA := binary.LittleEndian.Uint64(kvA.Value)
-		proposalIDB := binary.LittleEndian.Uint64(kvB.Value)
-		return fmt.Sprintf("proposalIDA: %d\nProposalIDB: %d", proposalIDA, proposalIDB)
-
-	case bytes.Equal(kvA.Key[:1], gov.DepositsKeyPrefix):
-		var depositA, depositB gov.Deposit
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &depositA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &depositB)
-		return fmt.Sprintf("%v\n%v", depositA, depositB)
-
-	case bytes.Equal(kvA.Key[:1], gov.VotesKeyPrefix):
-		var voteA, voteB gov.Vote
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &voteA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &voteB)
-		return fmt.Sprintf("%v\n%v", voteA, voteB)
-
-	default:
-		panic(fmt.Sprintf("invalid governance key prefix %X", kvA.Key[:1]))
-	}
-}
-
-// DecodeSupplyStore unmarshals the KVPair's Value to the corresponding supply type
-func DecodeSupplyStore(cdcA, cdcB *codec.Codec, kvA, kvB cmn.KVPair) string {
-	switch {
-	case bytes.Equal(kvA.Key[:1], supply.SupplyKey):
-		var supplyA, supplyB supply.Supply
-		cdcA.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &supplyA)
-		cdcB.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &supplyB)
-		return fmt.Sprintf("%v\n%v", supplyB, supplyB)
-	default:
-		panic(fmt.Sprintf("invalid supply key %X", kvA.Key))
-	}
-}