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