initial commit

msgpack/msgpack.go

@@ -0,0 +1,45 @@
+package msgpack
+
+import (
+	"bytes"
+	"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) },
+}
+
+type encoder struct{}
+
+var MsgPack = encoder{}
+
+func (encoder) Unmarshal(b []byte, v interface{}) error {
+	rdr := rdrPool.Get().(*bytes.Reader)
+	rdr.Reset(b)
+	defer rdrPool.Put(rdr)
+	return codec.NewDecoder(rdr, &codec.MsgpackHandle{}).Decode(v)
+}
+
+func (encoder) Marshal(v interface{}) ([]byte, error) {
+	buf := bufPool.Get().(*bytes.Buffer)
+	buf.Reset()
+	if err := codec.NewEncoder(buf, &codec.MsgpackHandle{}).Encode(v); err != nil {
+		bufPool.Put(buf)
+		return nil, err
+	}
+	return buf.Bytes(), nil
+}
+
+func Unmarshal(b []byte, v interface{}) error {
+	return MsgPack.Unmarshal(b, v)
+}
+
+func Marshal(v interface{}) ([]byte, error) {
+	return MsgPack.Marshal(v)
+}