0g-chain/x/cdp/simulation/decoder.go
2020-04-06 18:43:43 -04:00

67 lines
2.3 KiB
Go

package simulation
import (
"bytes"
"encoding/binary"
"fmt"
"time"
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
cmn "github.com/tendermint/tendermint/libs/common"
"github.com/kava-labs/kava/x/cdp/types"
)
// DecodeStore unmarshals the KVPair's Value to the corresponding cdp type
func DecodeStore(cdc *codec.Codec, kvA, kvB cmn.KVPair) string {
switch {
case bytes.Equal(kvA.Key[:1], types.CdpIDKeyPrefix):
var cdpIDsA, cdpIDsB []uint64
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &cdpIDsA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &cdpIDsB)
return fmt.Sprintf("%v\n%v", cdpIDsA, cdpIDsB)
case bytes.Equal(kvA.Key[:1], types.CdpKeyPrefix):
var cdpA, cdpB types.CDP
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &cdpA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &cdpB)
return fmt.Sprintf("%v\n%v", cdpA, cdpB)
case bytes.Equal(kvA.Key[:1], types.CdpIDKey),
bytes.Equal(kvA.Key[:1], types.CollateralRatioIndexPrefix):
idA := binary.BigEndian.Uint64(kvA.Value)
idB := binary.BigEndian.Uint64(kvB.Value)
return fmt.Sprintf("%d\n%d", idA, idB)
case bytes.Equal(kvA.Key[:1], types.DebtDenomKey),
bytes.Equal(kvA.Key[:1], types.GovDenomKey):
var denomA, denomB string
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &denomA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &denomB)
return fmt.Sprintf("%s\n%s", denomA, denomB)
case bytes.Equal(kvA.Key[:1], types.DepositKeyPrefix):
var depositA, depositB types.Deposit
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &depositA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &depositB)
return fmt.Sprintf("%s\n%s", depositA, depositB)
case bytes.Equal(kvA.Key[:1], types.PrincipalKeyPrefix):
var totalA, totalB sdk.Int
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &totalA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &totalB)
return fmt.Sprintf("%s\n%s", totalA, totalB)
case bytes.Equal(kvA.Key[:1], types.PreviousBlockTimeKey),
bytes.Equal(kvA.Key[:1], types.PreviousDistributionTimeKey):
var timeA, timeB time.Time
cdc.MustUnmarshalBinaryLengthPrefixed(kvA.Value, &timeA)
cdc.MustUnmarshalBinaryLengthPrefixed(kvB.Value, &timeB)
return fmt.Sprintf("%s\n%s", timeA, timeB)
default:
panic(fmt.Sprintf("invalid %s key prefix %X", types.ModuleName, kvA.Key[:1]))
}
}