Add missing trace finish on an uncommon code path

2016-01-14 17:01:22 -08:00
parent 7172d5c7a6
commit 8e76d451dd
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