grpc
/
grpc-go
mirror of https://github.com/grpc/grpc-go.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Specialize connection error handling to avoid goroutine leaking in some cases

This commit is contained in:
iamqizhao 2016-01-14 12:24:00 -08:00
parent 7ed4629849
commit 47fc4a2936
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
stream.go
View File

@ -133,8 +133,12 @@ func NewClientStream(ctx context.Context, desc *StreamDesc, cc *ClientConn, meth
// Listen on ctx.Done() to detect cancellation when there is no pending
// I/O operations on this stream.
go func() {
<-s.Context().Done()
cs.closeTransportStream(transport.ContextErr(s.Context().Err()))
select {
case <-t.Error():
// Incur transport error, simply exit.
case <-s.Context().Done():
cs.closeTransportStream(transport.ContextErr(s.Context().Err()))
}
}()
return cs, nil
}