From 8e76d451dd39b5a194c7affeafa3be9c27559d71 Mon Sep 17 00:00:00 2001
From: iamqizhao <toqizhao@gmail.com>
Date: Thu, 14 Jan 2016 17:01:22 -0800
Subject: [PATCH] Add missing trace finish on an uncommon code path

---
 stream.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/stream.go b/stream.go
index 3ac5f6a6..d8bdc16b 100644
--- 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