mirror of
https://github.com/shopspring/decimal.git
synced 2024-11-22 12:30:49 +01:00
add MsgPack Marshal And Unmarshal function
This commit is contained in:
parent
54fa471190
commit
519f778aec
1 changed files with 6 additions and 166 deletions
172
decimal_gen.go
172
decimal_gen.go
|
@ -4,28 +4,15 @@ import (
|
|||
"errors"
|
||||
)
|
||||
|
||||
// const (
|
||||
// first3 = 0xe0
|
||||
// last5 = 0x1f
|
||||
//
|
||||
// mfixstr uint8 = 0xa0
|
||||
// mstr8 uint8 = 0xd9
|
||||
// mstr16 uint8 = 0xda
|
||||
// mstr32 uint8 = 0xdb
|
||||
//
|
||||
// stringPrefixSize = 5
|
||||
// )
|
||||
|
||||
var (
|
||||
errShortBytes = errors.New("msgp: too few bytes left to read object")
|
||||
//bigEndian = binary.BigEndian
|
||||
)
|
||||
|
||||
|
||||
// MarshalMsg implements msgp.Marshaler
|
||||
func (z Decimal) MarshalMsg(b []byte) (o []byte, err error) {
|
||||
o = require(b, z.Msgsize())
|
||||
str := z.String()
|
||||
func (d Decimal) MarshalMsg(b []byte) (o []byte, err error) {
|
||||
o = require(b, d.Msgsize())
|
||||
str := d.String()
|
||||
sz := len(str)
|
||||
if sz > 30 {
|
||||
sz = 30
|
||||
|
@ -44,7 +31,7 @@ func (z Decimal) MarshalMsg(b []byte) (o []byte, err error) {
|
|||
}
|
||||
|
||||
// UnmarshalMsg implements msgp.Unmarshaler
|
||||
func (z *Decimal) UnmarshalMsg(b []byte) (o []byte, err error) {
|
||||
func (d *Decimal) UnmarshalMsg(b []byte) (o []byte, err error) {
|
||||
l := len(b)
|
||||
if l < 1 {
|
||||
return nil, errShortBytes
|
||||
|
@ -55,44 +42,16 @@ func (z *Decimal) UnmarshalMsg(b []byte) (o []byte, err error) {
|
|||
err = errShortBytes
|
||||
return
|
||||
}
|
||||
if *z, err = NewFromString(string(b[1 : sz+1])); err == nil {
|
||||
if *d, err = NewFromString(string(b[1 : sz+1])); err == nil {
|
||||
o = b[sz:]
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Msgsize returns an upper bound estimate of the number of bytes occupied by the serialized message
|
||||
func (z Decimal) Msgsize() int {
|
||||
func (d Decimal) Msgsize() int {
|
||||
return 31
|
||||
}
|
||||
//
|
||||
// // MarshalMsg implements msgp.Marshaler
|
||||
// func (z Decimal) MarshalMsg(b []byte) (o []byte, err error) {
|
||||
// o = require(b, z.Msgsize())
|
||||
// o = appendString(o, z.String())
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// // UnmarshalMsg implements msgp.Unmarshaler
|
||||
// func (z *Decimal) UnmarshalMsg(b []byte) (o []byte, err error) {
|
||||
// v, bts, err := readString(b)
|
||||
// if err != nil {
|
||||
// return
|
||||
// }
|
||||
// d, err := NewFromString(string(v))
|
||||
// if err == nil {
|
||||
// *z = d
|
||||
// o = bts
|
||||
// }
|
||||
//
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// // Msgsize returns an upper bound estimate of the number of bytes occupied by the serialized message
|
||||
// func (z Decimal) Msgsize() (s int) {
|
||||
// s = stringPrefixSize + len(z.String())
|
||||
// return
|
||||
// }
|
||||
|
||||
// Require ensures that cap(old)-len(old) >= extra.
|
||||
func require(old []byte, extra int) []byte {
|
||||
|
@ -129,122 +88,3 @@ func ensure(b []byte, sz int) ([]byte, int) {
|
|||
}
|
||||
return b[:l+sz], l
|
||||
}
|
||||
|
||||
// // AppendString appends a string as a MessagePack 'str' to the slice
|
||||
// func appendString(b []byte, s string) []byte {
|
||||
// sz := len(s)
|
||||
// var n int
|
||||
// var o []byte
|
||||
// switch {
|
||||
// case sz <= 31:
|
||||
// o, n = ensure(b, 1+sz)
|
||||
// o[n] = wfixstr(uint8(sz))
|
||||
// n++
|
||||
// case sz <= math.MaxUint8:
|
||||
// o, n = ensure(b, 2+sz)
|
||||
// prefixu8(o[n:], mstr8, uint8(sz))
|
||||
// n += 2
|
||||
// case sz <= math.MaxUint16:
|
||||
// o, n = ensure(b, 3+sz)
|
||||
// prefixu16(o[n:], mstr16, uint16(sz))
|
||||
// n += 3
|
||||
// default:
|
||||
// o, n = ensure(b, 5+sz)
|
||||
// prefixu32(o[n:], mstr32, uint32(sz))
|
||||
// n += 5
|
||||
// }
|
||||
// return o[:n+copy(o[n:], s)]
|
||||
// }
|
||||
//
|
||||
// func wfixstr(u uint8) byte {
|
||||
// return (u & last5) | mfixstr
|
||||
// }
|
||||
//
|
||||
// // write prefix and uint8
|
||||
// func prefixu8(b []byte, pre byte, sz uint8) {
|
||||
// b[0] = pre
|
||||
// b[1] = sz
|
||||
// }
|
||||
//
|
||||
// // write prefix and big-endian uint16
|
||||
// func prefixu16(b []byte, pre byte, sz uint16) {
|
||||
// b[0] = pre
|
||||
// b[1] = byte(sz >> 8)
|
||||
// b[2] = byte(sz)
|
||||
// }
|
||||
//
|
||||
// // write prefix and big-endian uint32
|
||||
// func prefixu32(b []byte, pre byte, sz uint32) {
|
||||
// b[0] = pre
|
||||
// b[1] = byte(sz >> 24)
|
||||
// b[2] = byte(sz >> 16)
|
||||
// b[3] = byte(sz >> 8)
|
||||
// b[4] = byte(sz)
|
||||
// }
|
||||
//
|
||||
// // readString reads a messagepack string field
|
||||
// // without copying. The returned []byte points
|
||||
// // to the same memory as the input slice.
|
||||
// // Possible errors:
|
||||
// // - ErrShortBytes (b not long enough)
|
||||
// func readString(b []byte) (v []byte, o []byte, err error) {
|
||||
// l := len(b)
|
||||
// if l < 1 {
|
||||
// return nil, nil, ErrShortBytes
|
||||
// }
|
||||
//
|
||||
// lead := b[0]
|
||||
// var read int
|
||||
//
|
||||
// if isfixstr(lead) {
|
||||
// read = int(rfixstr(lead))
|
||||
// b = b[1:]
|
||||
// } else {
|
||||
// switch lead {
|
||||
// case mstr8:
|
||||
// if l < 2 {
|
||||
// err = ErrShortBytes
|
||||
// return
|
||||
// }
|
||||
// read = int(b[1])
|
||||
// b = b[2:]
|
||||
//
|
||||
// case mstr16:
|
||||
// if l < 3 {
|
||||
// err = ErrShortBytes
|
||||
// return
|
||||
// }
|
||||
// read = int(bigEndian.Uint16(b[1:]))
|
||||
// b = b[3:]
|
||||
//
|
||||
// case mstr32:
|
||||
// if l < 5 {
|
||||
// err = ErrShortBytes
|
||||
// return
|
||||
// }
|
||||
// read = int(bigEndian.Uint32(b[1:]))
|
||||
// b = b[5:]
|
||||
//
|
||||
// default:
|
||||
// err = fmt.Errorf(`msgp: attempted to decode type 'str' with method for %d`, lead)
|
||||
// return
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// if len(b) < read {
|
||||
// err = ErrShortBytes
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// v = b[0:read]
|
||||
// o = b[read:]
|
||||
// return
|
||||
// }
|
||||
//
|
||||
// func isfixstr(b byte) bool {
|
||||
// return b&first3 == mfixstr
|
||||
// }
|
||||
//
|
||||
// func rfixstr(b byte) uint8 {
|
||||
// return b & last5
|
||||
// }
|
||||
|
|
Loading…
Reference in a new issue