when the stream errors cleanup the connection (#1199)

This commit is contained in:
Asim Aslam 2020-02-15 11:35:08 +00:00 committed by Vasiliy Tolstov
parent 5657fd07d2
commit 6460839705

View File

@ -218,6 +218,12 @@ func (g *grpcClient) stream(ctx context.Context, node *registry.Node, req client
st, err := cc.NewStream(newCtx, desc, methodToGRPC(req.Service(), req.Endpoint()), grpcCallOptions...)
if err != nil {
// we need to cleanup as we dialled and created a context
// cancel the context
cancel()
// close the connection
cc.Close()
// now return the error
return nil, errors.InternalServerError("go.micro.client", fmt.Sprintf("Error creating stream: %v", err))
}