fix(codec): ThriftMarshal

Change-Id: Ia6d24fc15450f04d77765eadb9d37817404ae3d0
mebratuA · Jun 25, 2019 · 5951dbb · 5951dbb
1 parent a6224fa
commit 5951dbb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/codec/thrift_codec.go b/codec/thrift_codec.go
@@ -67,9 +67,11 @@ func ThriftMarshal(v interface{}) ([]byte, error) {
 	p := thrift.NewTBinaryProtocol(trans, false, false)
 	switch s := v.(type) {
 	case thrift.TStruct:
-		return trans.Buffer.Bytes(), s.Write(p)
+		err := s.Write(p)
+		return trans.Buffer.Bytes(), err
 	case nil, *struct{}, struct{}:
-		return trans.Buffer.Bytes(), ThriftEmptyStruct.Write(p)
+		err := ThriftEmptyStruct.Write(p)
+		return trans.Buffer.Bytes(), err
 	}
 	return nil, fmt.Errorf("thrift codec: %T does not implement thrift.TStruct", v)
 }