github/ttrpc
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add client side context.Done support

Browse Source 
Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
This commit is contained in:
Michael Crosby 2018-08-28 10:54:57 -04:00
parent 94dde38880
commit d77f111e2e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
client.go
View File

@ -110,12 +110,16 @@ func (c *Client) dispatch(ctx context.Context, req *Request, resp *Response) err
} }
select { select {
case <-ctx.Done():
return ctx.Err()
case c.calls <- call: case c.calls <- call:
case <-c.done: case <-c.done:
return c.err return c.err
} }
select { select {
case <-ctx.Done():
return ctx.Err()
case err := <-errs: case err := <-errs:
return filterCloseErr(err) return filterCloseErr(err)
case <-c.done: case <-c.done: