msgpack-go

encode_value.go (4210B)

---

 package msgpack
 
 import "math"
 
 func (e *Encoder) Nil() {
 	e.grow(1)
 	e.buf = append(e.buf, 0xc0)
 }
 
 func (e *Encoder) Bool(v bool) {
 	e.grow(1)
 	if v {
 		e.buf = append(e.buf, 0xc2)
 	} else {
 		e.buf = append(e.buf, 0xc3)
 	}
 }
 
 func (e *Encoder) Int(v int) {
 	switch {
 	case v > 0xffffffff:
 		e.Uint64(uint64(v))
 	case v > 0xffff:
 		e.Uint32(uint32(v))
 	case v > 0xff:
 		e.Uint16(uint16(v))
 	case v > 0x7f:
 		e.Uint8(uint8(v))
 	case v < -0x80000000:
 		e.Int64(int64(v))
 	case v < -0x8000:
 		e.Int32(int32(v))
 	case v < -0x80:
 		e.Int16(int16(v))
 	case v < -32:
 		e.Int8(int8(v))
 	default:
 		e.grow(1)
 		e.buf = append(e.buf, byte(v))
 	}
 }
 
 func (e *Encoder) Int8(v int8) {
 	e.grow(2)
 	e.buf = append(e.buf, 0xd0, byte(v))
 }
 
 func (e *Encoder) Int16(v int16) {
 	e.grow(3)
 	e.buf = append(e.buf, 0xd1)
 	e.serialize16(uint16(v))
 }
 
 func (e *Encoder) Int32(v int32) {
 	e.grow(5)
 	e.buf = append(e.buf, 0xd2)
 	e.serialize32(uint32(v))
 }
 
 func (e *Encoder) Int64(v int64) {
 	e.grow(9)
 	e.buf = append(e.buf, 0xd3)
 	e.serialize64(uint64(v))
 }
 
 func (e *Encoder) Uint(v uint) {
 	switch {
 	case v > 0xffffffff:
 		e.Uint64(uint64(v))
 	case v > 0xffff:
 		e.Uint32(uint32(v))
 	case v > 0xff:
 		e.Uint16(uint16(v))
 	case v > 0x7f:
 		e.Uint8(uint8(v))
 	default:
 		e.grow(1)
 		e.buf = append(e.buf, byte(v))
 	}
 }
 
 func (e *Encoder) Uint8(v uint8) {
 	e.grow(2)
 	e.buf = append(e.buf, 0xcc, byte(v))
 }
 
 func (e *Encoder) Uint16(v uint16) {
 	e.grow(3)
 	e.buf = append(e.buf, 0xcd)
 	e.serialize16(v)
 }
 
 func (e *Encoder) Uint32(v uint32) {
 	e.grow(5)
 	e.buf = append(e.buf, 0xce)
 	e.serialize32(v)
 }
 
 func (e *Encoder) Uint64(v uint64) {
 	e.grow(9)
 	e.buf = append(e.buf, 0xcf)
 	e.serialize64(v)
 }
 
 func (e *Encoder) Float32(v float32) {
 	b := math.Float32bits(v)
 	e.grow(5)
 	e.buf = append(e.buf, 0xca)
 	e.serialize32(b)
 }
 
 func (e *Encoder) Float64(v float64) {
 	b := math.Float64bits(v)
 	e.grow(9)
 	e.buf = append(e.buf, 0xcb)
 	e.serialize64(b)
 }
 
 func (e *Encoder) String(s string) {
 	switch {
 	case len(s) > 0xFFFF:
 		e.grow(len(s) + 5)
 		e.buf = append(e.buf, 0xdb)
 		e.serialize32(uint32(len(s)))
 	case len(s) > 0xFF:
 		e.grow(len(s) + 3)
 		e.buf = append(e.buf, 0xda)
 		e.serialize16(uint16(len(s)))
 	case len(s) > 31:
 		e.grow(len(s) + 2)
 		e.buf = append(e.buf, 0xd9, byte(len(s)))
 	default:
 		e.grow(len(s) + 1)
 		e.buf = append(e.buf, byte(0xa0|len(s)))
 	}
 	e.buf = append(e.buf, s...)
 }
 
 func (e *Encoder) Binary(s []byte) {
 	switch {
 	case len(s) > 0xFFFF:
 		e.grow(len(s) + 5)
 		e.buf = append(e.buf, 0xc6)
 		e.serialize32(uint32(len(s)))
 	case len(s) > 0xFF:
 		e.grow(len(s) + 3)
 		e.buf = append(e.buf, 0xc5)
 		e.serialize16(uint16(len(s)))
 	default:
 		e.grow(len(s) + 2)
 		e.buf = append(e.buf, 0xc4, byte(len(s)))
 	}
 	e.buf = append(e.buf, s...)
 }
 
 func (e *Encoder) Array(v Array) {
 	n := v.Len()
 	switch {
 	case n > 0xFFFF:
 		e.grow(5)
 		e.buf = append(e.buf, 0xdd)
 		e.serialize32(uint32(n))
 	case n > 15:
 		e.grow(3)
 		e.buf = append(e.buf, 0xdc)
 		e.serialize16(uint16(n))
 	default:
 		e.grow(1)
 		e.buf = append(e.buf, byte(0x90|n))
 	}
 
 	v.EncodeMsgpackArray(e)
 }
 
 func (e *Encoder) Map(v Map) {
 	n := v.Size()
 	switch {
 	case n > 0xFFFF:
 		e.grow(5)
 		e.buf = append(e.buf, 0xdf)
 		e.serialize32(uint32(n))
 	case n > 15:
 		e.grow(3)
 		e.buf = append(e.buf, 0xde)
 		e.serialize16(uint16(n))
 	default:
 		e.grow(1)
 		e.buf = append(e.buf, byte(0x80|n))
 	}
 
 	v.EncodeMsgpackMap(e)
 }
 
 func (e *Encoder) Ext(v Extension) {
 	var (
 		typ = byte(v.Type())
 		n   = v.Size()
 	)
 	switch {
 	case n == 1:
 		e.grow(3)
 		e.buf = append(e.buf, 0xd4, typ)
 	case n == 2:
 		e.grow(4)
 		e.buf = append(e.buf, 0xd5, typ)
 	case n == 4:
 		e.grow(6)
 		e.buf = append(e.buf, 0xd6, typ)
 	case n == 8:
 		e.grow(10)
 		e.buf = append(e.buf, 0xd7, typ)
 	case n == 16:
 		e.grow(18)
 		e.buf = append(e.buf, 0xd8, typ)
 	case n > 0xFFFF:
 		e.grow(6 + n)
 		e.buf = append(e.buf, 0xc9)
 		e.serialize32(uint32(n))
 		e.buf = append(e.buf, typ)
 	case n > 0xFF:
 		e.grow(4 + n)
 		e.buf = append(e.buf, 0xc8)
 		e.serialize16(uint16(n))
 		e.buf = append(e.buf, typ)
 	default:
 		e.grow(3 + n)
 		e.buf = append(e.buf, 0xc7, byte(n), typ)
 	}
 	v.EncodeMsgpackExt(e)
 }
 
 func (e *Encoder) Raw(p []byte) {
 	e.grow(len(p))
 	e.buf = append(e.buf, p...)
 }