MyFavMirrors/minio
1
0
Fork 0
mirror of https://github.com/minio/minio.git synced 2025-11-26 04:26:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

A small logical change in messaging and logging

Browse Source
This commit is contained in:
Harshavardhana
2015-04-29 17:15:54 -07:00
parent 8be3b92ab7
commit 25d4c0f6fa
2 changed files with 20 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/api/quota/conn_limit.go
View File

@@ -20,6 +20,8 @@ import (
"net"
"net/http"
"sync"
"github.com/minio-io/minio/pkg/utils/log"
)
// requestLimitHandler
@@ -30,6 +32,10 @@ type connLimit struct {
limit int
}
func (c *connLimit) GetUsed(ip uint32) int {
return c.connections[ip]
}
func (c *connLimit) TestAndAdd(ip uint32) bool {
c.Lock()
defer c.Unlock()
@@ -58,12 +64,13 @@ func (c *connLimit) Remove(ip uint32) {
func (c *connLimit) ServeHTTP(w http.ResponseWriter, req *http.Request) {
host, _, _ := net.SplitHostPort(req.RemoteAddr)
longIP := longIP{net.ParseIP(host)}.IptoUint32()
if c.TestAndAdd(longIP) {
defer c.Remove(longIP)
c.handler.ServeHTTP(w, req)
} else {
writeErrorResponse(w, req, ConnectionLimitExceeded, req.RequestURI)
if !c.TestAndAdd(longIP) {
hosts, _ := net.LookupAddr(uint32ToIP(longIP).String())
log.Debug.Printf("Offending Host: %s, ConnectionsUSED: %d\n", hosts, c.GetUsed(longIP))
writeErrorResponse(w, req, ConnectionLimitExceeded, req.URL.Path)
}
defer c.Remove(longIP)
c.handler.ServeHTTP(w, req)
}
// ConnectionLimit limits the number of concurrent connections