1
0
mirror of https://github.com/go-micro/go-micro.git synced 2025-06-12 22:07:47 +02:00

fix broken pipe error

This commit is contained in:
Asim Aslam
2019-06-03 15:55:47 +01:00
parent 850f8bafdf
commit f80f0eb38e
2 changed files with 16 additions and 16 deletions

View File

@ -38,7 +38,11 @@ func (r *rpcStream) Send(msg interface{}) error {
Type: codec.Response,
}
return r.codec.Write(&resp, msg)
if err := r.codec.Write(&resp, msg); err != nil {
r.err = err
}
return nil
}
func (r *rpcStream) Recv(msg interface{}) error {
@ -51,12 +55,18 @@ func (r *rpcStream) Recv(msg interface{}) error {
if err := r.codec.ReadHeader(req, req.Type); err != nil {
// discard body
r.codec.ReadBody(nil)
r.err = err
return err
}
// we need to stay up to date with sequence numbers
r.id = req.Id
return r.codec.ReadBody(msg)
if err := r.codec.ReadBody(msg); err != nil {
r.err = err
return err
}
return nil
}
func (r *rpcStream) Error() error {