97 lines
3.0 KiB
Go
97 lines
3.0 KiB
Go
package simulation
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"cosmossdk.io/core/codec"
|
|
"cosmossdk.io/x/distribution/types"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/kv"
|
|
)
|
|
|
|
// NewDecodeStore returns a decoder function closure that unmarshals the KVPair's
|
|
// Value to the corresponding distribution type.
|
|
func NewDecodeStore(cdc codec.Codec) func(kvA, kvB kv.Pair) string {
|
|
return func(kvA, kvB kv.Pair) string {
|
|
switch {
|
|
case bytes.Equal(kvA.Key[:1], types.FeePoolKey):
|
|
var feePoolA, feePoolB types.FeePool
|
|
if err := cdc.Unmarshal(kvA.Value, &feePoolA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &feePoolB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", feePoolA, feePoolB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.ValidatorOutstandingRewardsPrefix):
|
|
var rewardsA, rewardsB types.ValidatorOutstandingRewards
|
|
if err := cdc.Unmarshal(kvA.Value, &rewardsA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &rewardsB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.DelegatorWithdrawAddrPrefix):
|
|
return fmt.Sprintf("%v\n%v", sdk.AccAddress(kvA.Value), sdk.AccAddress(kvB.Value))
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.DelegatorStartingInfoPrefix):
|
|
var infoA, infoB types.DelegatorStartingInfo
|
|
if err := cdc.Unmarshal(kvA.Value, &infoA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &infoB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", infoA, infoB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.ValidatorHistoricalRewardsPrefix):
|
|
var rewardsA, rewardsB types.ValidatorHistoricalRewards
|
|
if err := cdc.Unmarshal(kvA.Value, &rewardsA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &rewardsB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.ValidatorCurrentRewardsPrefix):
|
|
var rewardsA, rewardsB types.ValidatorCurrentRewards
|
|
if err := cdc.Unmarshal(kvA.Value, &rewardsA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &rewardsB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", rewardsA, rewardsB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.ValidatorAccumulatedCommissionPrefix):
|
|
var commissionA, commissionB types.ValidatorAccumulatedCommission
|
|
if err := cdc.Unmarshal(kvA.Value, &commissionA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &commissionB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", commissionA, commissionB)
|
|
|
|
case bytes.Equal(kvA.Key[:1], types.ValidatorSlashEventPrefix):
|
|
var eventA, eventB types.ValidatorSlashEvent
|
|
if err := cdc.Unmarshal(kvA.Value, &eventA); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := cdc.Unmarshal(kvB.Value, &eventB); err != nil {
|
|
panic(err)
|
|
}
|
|
return fmt.Sprintf("%v\n%v", eventA, eventB)
|
|
|
|
default:
|
|
panic(fmt.Sprintf("invalid distribution key prefix %X", kvA.Key[:1]))
|
|
}
|
|
}
|
|
}
|