agent: disable HTTP keepalives for checks

This commit is contained in:
Ryan Uber 2015-03-15 13:30:50 -07:00
parent b5253c940c
commit 589a4cc4c7
2 changed files with 29 additions and 5 deletions

View File

@ -286,15 +286,24 @@ func (c *CheckHTTP) Start() {
defer c.stopLock.Unlock() defer c.stopLock.Unlock()
if c.httpClient == nil { if c.httpClient == nil {
// Create the transport. We disable HTTP Keep-Alive's to prevent
// failing checks due to the keepalive interval.
trans := *http.DefaultTransport.(*http.Transport)
trans.DisableKeepAlives = true
// Create the HTTP client.
c.httpClient = &http.Client{
Timeout: 10 * time.Second,
Transport: &trans,
}
// For long (>10s) interval checks the http timeout is 10s, otherwise the // For long (>10s) interval checks the http timeout is 10s, otherwise the
// timeout is the interval. This means that a check *should* return // timeout is the interval. This means that a check *should* return
// before the next check begins. // before the next check begins.
if c.Timeout > 0 && c.Timeout < c.Interval { if c.Timeout > 0 && c.Timeout < c.Interval {
c.httpClient = &http.Client{Timeout: c.Timeout} c.httpClient.Timeout = c.Timeout
} else if c.Interval < 10*time.Second { } else if c.Interval < 10*time.Second {
c.httpClient = &http.Client{Timeout: c.Interval} c.httpClient.Timeout = c.Interval
} else {
c.httpClient = &http.Client{Timeout: 10 * time.Second}
} }
} }
@ -325,7 +334,6 @@ func (c *CheckHTTP) run() {
c.check() c.check()
next = time.After(c.Interval) next = time.After(c.Interval)
case <-c.stopCh: case <-c.stopCh:
http.DefaultTransport.(*http.Transport).CloseIdleConnections()
return return
} }
} }

View File

@ -305,3 +305,19 @@ func TestCheckHTTPTimeout(t *testing.T) {
t.Fatalf("should be critical %v", mock.state) t.Fatalf("should be critical %v", mock.state)
} }
} }
func TestCheckHTTP_disablesKeepAlives(t *testing.T) {
check := &CheckHTTP{
CheckID: "foo",
HTTP: "http://foo.bar/baz",
Interval: 10 * time.Second,
Logger: log.New(os.Stderr, "", log.LstdFlags),
}
check.Start()
defer check.Stop()
if !check.httpClient.Transport.(*http.Transport).DisableKeepAlives {
t.Fatalf("should have disabled keepalives")
}
}