Merge pull request #32 from psaab/develop

Limit number of concurrent connections
This commit is contained in:
Justin Li 2014-09-25 14:31:57 -04:00
commit 1ae60da852
2 changed files with 5 additions and 0 deletions

View file

@ -82,6 +82,7 @@ type Config struct {
RequestTimeout Duration `json:"request_timeout"`
HttpReadTimeout Duration `json:"http_read_timeout"`
HttpWriteTimeout Duration `json:"http_write_timeout"`
HttpListenLimit int `json:"http_listen_limit"`
NumWantFallback int `json:"default_num_want"`
ClientWhitelistEnabled bool `json:"client_whitelist_enabled"`

View file

@ -117,10 +117,14 @@ func Serve(cfg *config.Config, tkr *tracker.Tracker) {
}
glog.V(0).Info("Starting on ", cfg.Addr)
if cfg.HttpListenLimit != 0 {
glog.V(0).Info("Limiting connections to ", cfg.HttpListenLimit)
}
grace := &graceful.Server{
Timeout: cfg.RequestTimeout.Duration,
ConnState: srv.connState,
ListenLimit: cfg.HttpListenLimit,
Server: &http.Server{
Addr: cfg.Addr,
Handler: newRouter(srv),