fix: improve EOF detection

powerman · Apr 23, 2020 · 6f3efa7 · 6f3efa7
1 parent c2902e8
commit 6f3efa7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jsonrpc2/client.go b/jsonrpc2/client.go
@@ -172,7 +172,7 @@ func (c *clientCodec) ReadResponseHeader(r *rpc.Response) error {
 	// - client will be shutdown
 	// So, return io.EOF as is, return *Error for all other errors.
 	if err := c.dec.Decode(&c.resp); err != nil {
-		if err == io.EOF {
+		if err == io.EOF || err == io.ErrUnexpectedEOF {
 			return err
 		}
 		return NewError(errInternal.Code, err.Error())