package gob

import (
	"bytes"
	"encoding/gob"
)

//Register 注册Gob类型
func Register(o ...interface{}) {
	for _, v := range o {
		gob.Register(v)
	}
	mapGob := make(map[string]interface{})
	gob.Register(mapGob)
}

//Marshal 编码
func Marshal(v interface{}) ([]byte, error) {
	b := new(bytes.Buffer)
	err := gob.NewEncoder(b).Encode(v)
	if err != nil {
		return nil, err
	}
	return b.Bytes(), nil
}

//Unmarshal 解码
func Unmarshal(data []byte, v interface{}) error {
	b := bytes.NewBuffer(data)
	return gob.NewDecoder(b).Decode(v)
}