From d77f111e2e2792ffa7cb82585768aa467db07f41 Mon Sep 17 00:00:00 2001
From: Michael Crosby <crosbymichael@gmail.com>
Date: Tue, 28 Aug 2018 10:54:57 -0400
Subject: [PATCH] Add client side context.Done support

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
---
 client.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/client.go b/client.go
index ede2e6b..e40592d 100644
--- a/client.go
+++ b/client.go
@@ -110,12 +110,16 @@ func (c *Client) dispatch(ctx context.Context, req *Request, resp *Response) err
 	}
 
 	select {
+	case <-ctx.Done():
+		return ctx.Err()
 	case c.calls <- call:
 	case <-c.done:
 		return c.err
 	}
 
 	select {
+	case <-ctx.Done():
+		return ctx.Err()
 	case err := <-errs:
 		return filterCloseErr(err)
 	case <-c.done: