sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement Server read and write timeouts

Browse Source
This commit is contained in:
Adnan Maolood
2020-10-31 21:07:02 -04:00
parent 42c95f8c8d
commit 7d470c5fb1
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
client.go
View File

@@ -65,13 +65,6 @@ func (c *Client) Do(req *Request) (*Response, error) {
return c.do(req, nil)
}
func (c *Client) deadline() time.Time {
if c.Timeout > 0 {
return time.Now().Add(c.Timeout)
}
return time.Time{}
}
func (c *Client) do(req *Request, via []*Request) (*Response, error) {
// Connect to the host
config := &tls.Config{
@@ -89,10 +82,8 @@ func (c *Client) do(req *Request, via []*Request) (*Response, error) {
return nil, err
}
// Set connection deadline
if deadline := c.deadline(); !deadline.IsZero() {
if err := conn.SetDeadline(deadline); err != nil {
return nil, err
}
if d := c.Timeout; d != 0 {
conn.SetDeadline(time.Now().Add(d))
}
// Write the request