Add missing trace finish on an uncommon code path

This commit is contained in:
iamqizhao
2016-01-14 17:01:22 -08:00
parent 7172d5c7a6
commit 8e76d451dd

View File

@ -138,7 +138,9 @@ func NewClientStream(ctx context.Context, desc *StreamDesc, cc *ClientConn, meth
case <-t.Error(): case <-t.Error():
// Incur transport error, simply exit. // Incur transport error, simply exit.
case <-s.Context().Done(): 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 return cs, nil