package integration import ( "math" "net" "net/http" "net/http/httptest" "os" "time" "github.com/go-check/check" "github.com/traefik/traefik/v2/integration/try" checker "github.com/vdemeester/shakers" ) type KeepAliveSuite struct { BaseSuite } type KeepAliveConfig struct { KeepAliveServer string IdleConnTimeout string } type connStateChangeEvent struct { key string state http.ConnState } func (s *KeepAliveSuite) TestShouldRespectConfiguredBackendHttpKeepAliveTime(c *check.C) { idleTimeout := time.Duration(75) * time.Millisecond connStateChanges := make(chan connStateChangeEvent) noMoreRequests := make(chan bool, 1) completed := make(chan bool, 1) // keep track of HTTP connections and their status changes and measure their idle period go func() { connCount := 0 idlePeriodStartMap := make(map[string]time.Time) idlePeriodLengthMap := make(map[string]time.Duration) maxWaitDuration := 5 * time.Second maxWaitTimeExceeded := time.After(maxWaitDuration) moreRequestsExpected := true // Ensure that all idle HTTP connections are closed before verification phase for moreRequestsExpected || len(idlePeriodLengthMap) < connCount { select { case event := <-connStateChanges: switch event.state { case http.StateNew: connCount++ case http.StateIdle: idlePeriodStartMap[event.key] = time.Now() case http.StateClosed: idlePeriodLengthMap[event.key] = time.Since(idlePeriodStartMap[event.key]) } case <-noMoreRequests: moreRequestsExpected = false case <-maxWaitTimeExceeded: c.Logf("timeout waiting for all connections to close, waited for %v, configured idle timeout was %v", maxWaitDuration, idleTimeout) c.Fail() close(completed) return } } c.Check(connCount, checker.Equals, 1) for _, idlePeriod := range idlePeriodLengthMap { // Our method of measuring the actual idle period is not precise, so allow some sub-ms deviation c.Check(math.Round(idlePeriod.Seconds()), checker.LessOrEqualThan, idleTimeout.Seconds()) } close(completed) }() server := httptest.NewUnstartedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(200) })) server.Config.ConnState = func(conn net.Conn, state http.ConnState) { connStateChanges <- connStateChangeEvent{key: conn.RemoteAddr().String(), state: state} } server.Start() defer server.Close() config := KeepAliveConfig{KeepAliveServer: server.URL, IdleConnTimeout: idleTimeout.String()} file := s.adaptFile(c, "fixtures/timeout/keepalive.toml", config) defer os.Remove(file) cmd, display := s.traefikCmd(withConfigFile(file)) defer display(c) err := cmd.Start() c.Check(err, checker.IsNil) defer cmd.Process.Kill() err = try.GetRequest("http://127.0.0.1:8000/keepalive", time.Duration(1)*time.Second, try.StatusCodeIs(200)) c.Check(err, checker.IsNil) close(noMoreRequests) <-completed }