106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
|
package integration
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/integration/try"
|
||
|
|
"github.com/go-check/check"
|
||
|
|
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
|
||
|
|
}
|