Skip to content

Commit

Permalink
feat(types/collections): add LegacyDec collection value (#21693)
Browse files Browse the repository at this point in the history
Co-authored-by: Luca Graziotti <luca@noble.xyz>
  • Loading branch information
johnletey and g-luca authored Sep 13, 2024
1 parent d54f034 commit 3bc707a
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 @@ -37,6 +37,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 @@ -57,8 +60,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 @@ -246,6 +250,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 3bc707a

Please sign in to comment.