54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package msgpack
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/hashicorp/go-msgpack/codec"
|
|
)
|
|
|
|
var bufPool = sync.Pool{
|
|
New: func() interface{} { return new(bytes.Buffer) },
|
|
}
|
|
|
|
var rdrPool = sync.Pool{
|
|
New: func() interface{} { return new(bytes.Reader) },
|
|
}
|
|
|
|
func NewEncoder(w io.Writer) *codec.Encoder {
|
|
return codec.NewEncoder(w, &codec.MsgpackHandle{})
|
|
}
|
|
|
|
func NewDecoder(r io.Reader) *codec.Decoder {
|
|
return codec.NewDecoder(r, &codec.MsgpackHandle{})
|
|
}
|
|
|
|
type msgPackEncoder struct{}
|
|
|
|
var MsgPack = msgPackEncoder{}
|
|
|
|
func (msgPackEncoder) Marshal(v interface{}) ([]byte, error) {
|
|
buf := bufPool.Get().(*bytes.Buffer)
|
|
buf.Reset()
|
|
if err := NewEncoder(buf).Encode(v); err != nil {
|
|
bufPool.Put(buf)
|
|
return nil, err
|
|
}
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
func (msgPackEncoder) Unmarshal(b []byte, v interface{}) error {
|
|
rdr := rdrPool.Get().(*bytes.Reader)
|
|
rdr.Reset(b)
|
|
defer rdrPool.Put(rdr)
|
|
return NewDecoder(rdr).Decode(v)
|
|
}
|
|
|
|
func Marshal(v interface{}) ([]byte, error) {
|
|
return MsgPack.Marshal(v)
|
|
}
|
|
|
|
func Unmarshal(b []byte, v interface{}) error {
|
|
return MsgPack.Unmarshal(b, v)
|
|
}
|