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)
}