opensource/grpc-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use user context instead of creating new context for client side stream

Browse Source
This commit is contained in:
Menghan Li
2016-09-12 13:23:05 -07:00
parent 28707e14b1
commit 5f47f9f5cc
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
transport/http2_client.go
View File

@ -252,8 +252,7 @@ func (t *http2Client) newStream(ctx context.Context, callHdr *CallHdr) *Stream {
s.windowHandler = func(n int) { s.windowHandler = func(n int) {
t.updateWindow(s, uint32(n)) t.updateWindow(s, uint32(n))
} }
// Make a stream be able to cancel the pending operations by itself. s.ctx = ctx
s.ctx, s.cancel = context.WithCancel(ctx)
s.dec = &recvBufferReader{ s.dec = &recvBufferReader{
ctx: s.ctx, ctx: s.ctx,
goAway: s.goAway, goAway: s.goAway,

1
transport/transport.go
View File

@ -170,6 +170,7 @@ type Stream struct {
st ServerTransport st ServerTransport
// ctx is the associated context of the stream. // ctx is the associated context of the stream.
ctx context.Context ctx context.Context
// cancel is always nil for client side Stream.
cancel context.CancelFunc cancel context.CancelFunc
// done is closed when the final status arrives. // done is closed when the final status arrives.
done chan struct{} done chan struct{}