|
|
package codec
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
|
|
|
codecpkg "github.com/noahlann/nnet/pkg/codec"
|
|
|
)
|
|
|
|
|
|
// MessagePackCodec MessagePack编解码器
|
|
|
type MessagePackCodec struct {
|
|
|
encoder MessagePackEncoder
|
|
|
decoder MessagePackDecoder
|
|
|
}
|
|
|
|
|
|
// NewMessagePackCodec 创建MessagePack编解码器
|
|
|
func NewMessagePackCodec() codecpkg.Codec {
|
|
|
return &MessagePackCodec{
|
|
|
encoder: &defaultMessagePackEncoder{},
|
|
|
decoder: &defaultMessagePackDecoder{},
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Encode 编码
|
|
|
func (c *MessagePackCodec) Encode(v interface{}) ([]byte, error) {
|
|
|
return c.encoder.Encode(v)
|
|
|
}
|
|
|
|
|
|
// Decode 解码
|
|
|
func (c *MessagePackCodec) Decode(data []byte, v interface{}) error {
|
|
|
return c.decoder.Decode(data, v)
|
|
|
}
|
|
|
|
|
|
// Name 获取编解码器名称
|
|
|
func (c *MessagePackCodec) Name() string {
|
|
|
return "msgpack"
|
|
|
}
|
|
|
|
|
|
// MessagePackEncoder MessagePack编码器接口
|
|
|
type MessagePackEncoder interface {
|
|
|
Encode(v interface{}) ([]byte, error)
|
|
|
}
|
|
|
|
|
|
// MessagePackDecoder MessagePack解码器接口
|
|
|
type MessagePackDecoder interface {
|
|
|
Decode(data []byte, v interface{}) error
|
|
|
}
|
|
|
|
|
|
// defaultMessagePackEncoder 默认MessagePack编码器（使用JSON作为fallback）
|
|
|
type defaultMessagePackEncoder struct{}
|
|
|
|
|
|
func (e *defaultMessagePackEncoder) Encode(v interface{}) ([]byte, error) {
|
|
|
// 如果没有安装msgpack库，使用JSON作为fallback
|
|
|
// 在实际使用中，用户应该安装github.com/vmihailenco/msgpack/v5
|
|
|
// 这里提供一个简单的实现，使用JSON格式但标记为msgpack
|
|
|
return json.Marshal(v)
|
|
|
}
|
|
|
|
|
|
// defaultMessagePackDecoder 默认MessagePack解码器（使用JSON作为fallback）
|
|
|
type defaultMessagePackDecoder struct{}
|
|
|
|
|
|
func (d *defaultMessagePackDecoder) Decode(data []byte, v interface{}) error {
|
|
|
// 如果没有安装msgpack库，使用JSON作为fallback
|
|
|
return json.Unmarshal(data, v)
|
|
|
}
|
|
|
|