58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
|
package simulation
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
||
|
|
"github.com/cosmos/cosmos-sdk/types/kv"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/group"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/group/keeper"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NewDecodeStore returns a decoder function closure that unmarshals the KVPair's
|
||
|
|
// Value to the corresponding group 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], []byte{keeper.GroupTablePrefix}):
|
||
|
|
var groupA, groupB group.GroupInfo
|
||
|
|
|
||
|
|
cdc.MustUnmarshal(kvA.Value, &groupA)
|
||
|
|
cdc.MustUnmarshal(kvB.Value, &groupB)
|
||
|
|
|
||
|
|
return fmt.Sprintf("%v\n%v", groupA, groupB)
|
||
|
|
case bytes.Equal(kvA.Key[:1], []byte{keeper.GroupMemberTablePrefix}):
|
||
|
|
var memberA, memberB group.GroupMember
|
||
|
|
|
||
|
|
cdc.MustUnmarshal(kvA.Value, &memberA)
|
||
|
|
cdc.MustUnmarshal(kvB.Value, &memberB)
|
||
|
|
|
||
|
|
return fmt.Sprintf("%v\n%v", memberA, memberB)
|
||
|
|
case bytes.Equal(kvA.Key[:1], []byte{keeper.GroupPolicyTablePrefix}):
|
||
|
|
var accA, accB group.GroupPolicyInfo
|
||
|
|
|
||
|
|
cdc.MustUnmarshal(kvA.Value, &accA)
|
||
|
|
cdc.MustUnmarshal(kvB.Value, &accB)
|
||
|
|
|
||
|
|
return fmt.Sprintf("%v\n%v", accA, accB)
|
||
|
|
case bytes.Equal(kvA.Key[:1], []byte{keeper.ProposalTablePrefix}):
|
||
|
|
var propA, propB group.Proposal
|
||
|
|
|
||
|
|
cdc.MustUnmarshal(kvA.Value, &propA)
|
||
|
|
cdc.MustUnmarshal(kvB.Value, &propB)
|
||
|
|
|
||
|
|
return fmt.Sprintf("%v\n%v", propA, propB)
|
||
|
|
case bytes.Equal(kvA.Key[:1], []byte{keeper.VoteTablePrefix}):
|
||
|
|
var voteA, voteB group.Vote
|
||
|
|
|
||
|
|
cdc.MustUnmarshal(kvA.Value, &voteA)
|
||
|
|
cdc.MustUnmarshal(kvB.Value, &voteB)
|
||
|
|
|
||
|
|
return fmt.Sprintf("%v\n%v", voteA, voteB)
|
||
|
|
default:
|
||
|
|
panic(fmt.Sprintf("invalid group key %X", kvA.Key))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|