-
Notifications
You must be signed in to change notification settings - Fork 169
/
decoder.go
94 lines (80 loc) · 3.31 KB
/
decoder.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
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"
gogotypes "github.com/gogo/protobuf/types"
"github.com/umee-network/umee/v2/x/leverage/types"
)
// NewDecodeStore returns a decoder function closure that unmarshals the KVPair's
// Value to the corresponding leverage type.
func NewDecodeStore(cdc codec.Codec) func(kvA, kvB kv.Pair) string {
return func(kvA, kvB kv.Pair) string {
prefixA := kvA.Key[:1]
switch {
case bytes.Equal(prefixA, types.KeyPrefixRegisteredToken):
var registeredTokenA, registeredTokenB types.Token
cdc.MustUnmarshal(kvA.Value, ®isteredTokenA)
cdc.MustUnmarshal(kvB.Value, ®isteredTokenB)
return fmt.Sprintf("%v\n%v", registeredTokenA, registeredTokenB)
case bytes.Equal(prefixA, types.KeyPrefixAdjustedBorrow):
var amountA, amountB sdk.Dec
if err := amountA.Unmarshal(kvA.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
if err := amountB.Unmarshal(kvB.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
return fmt.Sprintf("%v\n%v", amountA, amountB)
case bytes.Equal(prefixA, types.KeyPrefixCollateralSetting):
return fmt.Sprintf("%v\n%v", kvA, kvB) // it is bytes: []byte{0x01}
case bytes.Equal(prefixA, types.KeyPrefixCollateralAmount):
var amountA, amountB sdk.Int
if err := amountA.Unmarshal(kvA.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
if err := amountB.Unmarshal(kvB.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
return fmt.Sprintf("%v\n%v", amountA, amountB)
case bytes.Equal(prefixA, types.KeyPrefixReserveAmount):
var amountA, amountB sdk.Int
if err := amountA.Unmarshal(kvA.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
if err := amountB.Unmarshal(kvB.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
return fmt.Sprintf("%v\n%v", amountA, amountB)
case bytes.Equal(prefixA, types.KeyPrefixLastInterestTime):
var lastInterestTimeA, lastInterestTimeB gogotypes.Int64Value
cdc.MustUnmarshal(kvA.Value, &lastInterestTimeA)
cdc.MustUnmarshal(kvA.Value, &lastInterestTimeB)
return fmt.Sprintf("%v\n%v", lastInterestTimeA, lastInterestTimeB)
case bytes.Equal(prefixA, types.KeyPrefixBadDebt):
return fmt.Sprintf("%v\n%v", kvA, kvB) // it is bytes: []byte{0x01}
case bytes.Equal(prefixA, types.KeyPrefixInterestScalar):
var scalarA, scalarB sdk.Dec
if err := scalarA.Unmarshal(kvA.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
if err := scalarB.Unmarshal(kvB.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
return fmt.Sprintf("%v\n%v", scalarA, scalarB)
case bytes.Equal(prefixA, types.KeyPrefixAdjustedTotalBorrow):
var totalA, totalB sdk.Dec
if err := totalA.Unmarshal(kvA.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
if err := totalB.Unmarshal(kvB.Value); err != nil {
panic(fmt.Sprintf("invalid unmarshal value %+v", err))
}
return fmt.Sprintf("%v\n%v", totalA, totalB)
default:
panic(fmt.Sprintf("invalid leverage key prefix %X", kvA.Key[:1]))
}
}
}