mirror of
https://github.com/go-micro/go-micro.git
synced 2024-12-18 08:26:38 +02:00
d1806e2883
* Update marshaler.go Byes codec always return error "invalid message" now. * Update go.mod Update package name * Update go.mod
41 lines
634 B
Go
41 lines
634 B
Go
package bytes
|
|
|
|
import (
|
|
"go-micro.dev/v4/codec"
|
|
)
|
|
|
|
type Marshaler struct{}
|
|
|
|
type Message struct {
|
|
Header map[string]string
|
|
Body []byte
|
|
}
|
|
|
|
func (n Marshaler) Marshal(v interface{}) ([]byte, error) {
|
|
switch ve := v.(type) {
|
|
case *[]byte:
|
|
return *ve, nil
|
|
case []byte:
|
|
return ve, nil
|
|
case *Message:
|
|
return ve.Body, nil
|
|
}
|
|
return nil, codec.ErrInvalidMessage
|
|
}
|
|
|
|
func (n Marshaler) Unmarshal(d []byte, v interface{}) error {
|
|
switch ve := v.(type) {
|
|
case *[]byte:
|
|
*ve = d
|
|
return nil
|
|
case *Message:
|
|
ve.Body = d
|
|
return nil
|
|
}
|
|
return codec.ErrInvalidMessage
|
|
}
|
|
|
|
func (n Marshaler) String() string {
|
|
return "bytes"
|
|
}
|