clean up implementation

src/mod/dynamicproxy/Server.go

@@ -73,11 +73,12 @@ func (h *ProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		}
 
 		// Rate Limit Check
-		// if sep.RequireBasicAuth {
-		if err := handleRateLimit(w, r, sep); err != nil {
-			return
+		if sep.RequireRateLimit {
+			err := h.handleRateLimitRouting(w, r, sep)
+			if err != nil {
+				return
+			}
 		}
-		// }
 
 		//Validate basic auth
 		if sep.RequireBasicAuth {

src/mod/dynamicproxy/ratelimit.go

@@ -50,6 +50,14 @@ func (t *IpTable) Clear() {
 
 var ipTable = IpTable{}
 
+func (h *ProxyHandler) handleRateLimitRouting(w http.ResponseWriter, r *http.Request, pe *ProxyEndpoint) error {
+	err := handleRateLimit(w, r, pe)
+	if err != nil {
+		h.logRequest(r, false, 429, "ratelimit", pe.Domain)
+	}
+	return err
+}
+
 func handleRateLimit(w http.ResponseWriter, r *http.Request, pe *ProxyEndpoint) error {
 	ip, _, err := net.SplitHostPort(r.RemoteAddr)
 	if err != nil {
@@ -60,7 +68,7 @@ func handleRateLimit(w http.ResponseWriter, r *http.Request, pe *ProxyEndpoint)
 
 	ipTable.Increment(ip)
 
-	if ipTable.Exceeded(ip, 10) {
+	if ipTable.Exceeded(ip, int64(pe.RateLimit)) {
 		w.WriteHeader(429)
 		return errors.New("rate limit exceeded")
 	}

src/mod/dynamicproxy/typedef.go

@@ -125,8 +125,8 @@ type ProxyEndpoint struct {
 	BasicAuthExceptionRules []*BasicAuthExceptionRule //Path to exclude in a basic auth enabled proxy target
 
 	// Rate Limiting
-	EnableRateLimiting bool
-	RateLimiting       int // Rate limit in requests per second
+	RequireRateLimit bool
+	RateLimit        int64 // Rate limit in requests per second
 
 	//Access Control
 	AccessFilterUUID string //Access filter ID