unistack-org/micro-codec-proto
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 15 Wiki Activity

add nil check

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2020-11-25 12:24:14 +03:00
parent 979b824b2f
commit f627fea89e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
proto.go
View File

@ -13,6 +13,9 @@ import (
type protoCodec struct{} type protoCodec struct{}
func (c *protoCodec) Marshal(v interface{}) ([]byte, error) { func (c *protoCodec) Marshal(v interface{}) ([]byte, error) {
if v == nil {
return nil, nil
}
switch m := v.(type) { switch m := v.(type) {
case *codec.Frame: case *codec.Frame:
return m.Data, nil return m.Data, nil
@ -25,7 +28,7 @@ func (c *protoCodec) Marshal(v interface{}) ([]byte, error) {
} }
func (c *protoCodec) Unmarshal(d []byte, v interface{}) error { func (c *protoCodec) Unmarshal(d []byte, v interface{}) error {
if d == nil { if d == nil || v == nil {
return nil return nil
} }
switch m := v.(type) { switch m := v.(type) {