Skip to content

Commit

Permalink
feat(types/collections): add LegacyDec collection value (backport #…
Browse files Browse the repository at this point in the history
  • Loading branch information
mergify[bot] authored Sep 13, 2024
1 parent 45b18a1 commit 64beb97
Showing 1 changed file with 42 additions and 2 deletions.
44 changes: 42 additions & 2 deletions types/collections.go
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,9 @@ var (
// UintValue represents a collections.ValueCodec to work with Uint.
UintValue collcodec.ValueCodec[math.Uint] = uintValueCodec{}

// LegacyDecValue represents a collections.ValueCodec to work with LegacyDec.
LegacyDecValue collcodec.ValueCodec[math.LegacyDec] = legacyDecValueCodec{}

// TimeKey represents a collections.KeyCodec to work with time.Time
// Deprecated: exists only for state compatibility reasons, should not
// be used for new storage keys using time. Please use the time KeyCodec
Expand All @@ -56,8 +59,9 @@ var (
)

const (
Int string = "math.Int"
Uint string = "math.Uint"
Int string = "math.Int"
Uint string = "math.Uint"
LegacyDec string = "math.LegacyDec"
)

type addressUnion interface {
Expand Down Expand Up @@ -245,6 +249,42 @@ func (i uintValueCodec) ValueType() string {
return Uint
}

type legacyDecValueCodec struct{}

func (i legacyDecValueCodec) Encode(value math.LegacyDec) ([]byte, error) {
return value.Marshal()
}

func (i legacyDecValueCodec) Decode(b []byte) (math.LegacyDec, error) {
v := new(math.LegacyDec)
err := v.Unmarshal(b)
if err != nil {
return math.LegacyDec{}, err
}
return *v, nil
}

func (i legacyDecValueCodec) EncodeJSON(value math.LegacyDec) ([]byte, error) {
return value.MarshalJSON()
}

func (i legacyDecValueCodec) DecodeJSON(b []byte) (math.LegacyDec, error) {
v := new(math.LegacyDec)
err := v.UnmarshalJSON(b)
if err != nil {
return math.LegacyDec{}, err
}
return *v, nil
}

func (i legacyDecValueCodec) Stringify(value math.LegacyDec) string {
return value.String()
}

func (i legacyDecValueCodec) ValueType() string {
return LegacyDec
}

type timeKeyCodec struct{}

func (timeKeyCodec) Encode(buffer []byte, key time.Time) (int, error) {
Expand Down

0 comments on commit 64beb97

Please sign in to comment.