Merge pull request #492 from iamqizhao/master

Add missing trace finish on an uncommon code path
2016-01-14 17:41:25 -08:00
parent b24c357536 8e76d451dd
commit 5da22b92e9
1 changed files with 3 additions and 1 deletions
--- a/stream.go
+++ b/stream.go
@ -138,7 +138,9 @@ func NewClientStream(ctx context.Context, desc *StreamDesc, cc *ClientConn, meth
 		case <-t.Error():
 			// Incur transport error, simply exit.
 		case <-s.Context().Done():
-			cs.closeTransportStream(transport.ContextErr(s.Context().Err()))
+			err := s.Context().Err()
+			cs.finish(err)
+			cs.closeTransportStream(transport.ContextErr(err))
 		}
 	}()
 	return cs, nil