34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package simulation
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/kv"
|
|
"github.com/cosmos/cosmos-sdk/x/capability/types"
|
|
)
|
|
|
|
// NewDecodeStore returns a decoder function closure that unmarshals the KVPair's
|
|
// Value to the corresponding capability type.
|
|
func NewDecodeStore(cdc codec.Marshaler) func(kvA, kvB kv.Pair) string {
|
|
return func(kvA, kvB kv.Pair) string {
|
|
switch {
|
|
case bytes.Equal(kvA.Key, types.KeyIndex):
|
|
idxA := sdk.BigEndianToUint64(kvA.Value)
|
|
idxB := sdk.BigEndianToUint64(kvB.Value)
|
|
return fmt.Sprintf("Index A: %d\nIndex B: %d\n", idxA, idxB)
|
|
|
|
case bytes.HasPrefix(kvA.Key, types.KeyPrefixIndexCapability):
|
|
var capOwnersA, capOwnersB types.CapabilityOwners
|
|
cdc.MustUnmarshalBinaryBare(kvA.Value, &capOwnersA)
|
|
cdc.MustUnmarshalBinaryBare(kvB.Value, &capOwnersB)
|
|
return fmt.Sprintf("CapabilityOwners A: %v\nCapabilityOwners B: %v\n", capOwnersA, capOwnersB)
|
|
|
|
default:
|
|
panic(fmt.Sprintf("invalid %s key prefix %X (%s)", types.ModuleName, kvA.Key, string(kvA.Key)))
|
|
}
|
|
}
|
|
}
|