Add context to requests

This commit is contained in:
Adnan Maolood
2020-11-26 00:42:25 -05:00
parent 3b9cc7f168
commit 82688746dd
2 changed files with 10 additions and 3 deletions

View File

@@ -96,10 +96,11 @@ func (c *Client) do(req *Request, via []*Request) (*Response, error) {
return c.verifyConnection(req, cs)
},
}
conn, err := tls.Dial("tcp", req.Host, config)
netConn, err := (&net.Dialer{}).DialContext(req.Context, "tcp", req.Host)
if err != nil {
return nil, err
}
conn := tls.Client(netConn, config)
// Set connection deadline
if d := c.Timeout; d != 0 {
conn.SetDeadline(time.Now().Add(d))