micro/codec/bytes/marshaler.go

39 lines
625 B
Go
Raw Normal View History

2019-01-10 12:42:02 +03:00
package bytes
import (
"github.com/unistack-org/micro/v3/codec"
2019-01-10 12:42:02 +03:00
)
type Marshaler struct{}
type Message struct {
Header map[string]string
Body []byte
}
func (n Marshaler) Marshal(v interface{}) ([]byte, error) {
2019-12-03 10:25:58 +03:00
switch ve := v.(type) {
2019-01-15 00:30:43 +03:00
case *[]byte:
return *ve, nil
2019-01-10 12:42:02 +03:00
case []byte:
2019-12-03 10:25:58 +03:00
return ve, nil
2019-01-10 12:42:02 +03:00
case *Message:
2019-12-03 10:25:58 +03:00
return ve.Body, nil
2019-01-10 12:42:02 +03:00
}
return nil, codec.ErrInvalidMessage
2019-01-10 12:42:02 +03:00
}
func (n Marshaler) Unmarshal(d []byte, v interface{}) error {
2019-12-03 10:25:58 +03:00
switch ve := v.(type) {
2019-01-10 12:42:02 +03:00
case *[]byte:
*ve = d
case *Message:
ve.Body = d
}
return codec.ErrInvalidMessage
2019-01-10 12:42:02 +03:00
}
func (n Marshaler) String() string {
return "bytes"
}