Add throttled

Author: Katsute

src/main/java/com/kttdevelopment/simplehttpserver/handler/ConnectionThrottler.java

@@ -0,0 +1,60 @@
+package com.kttdevelopment.simplehttpserver.handler;
+
+import com.kttdevelopment.simplehttpserver.SimpleHttpServer;
+import com.sun.net.httpserver.HttpExchange;
+
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.function.Predicate;
+
+public class ConnectionThrottler {
+
+    private final SimpleHttpServer server;
+
+    private final Predicate<HttpExchange> contributeToLimit;
+    private final AtomicInteger           connections = new AtomicInteger(0);
+
+    private int maxConnections = 0;
+
+    public ConnectionThrottler(final SimpleHttpServer server){
+        this.server = server;
+        contributeToLimit = (exchange) -> true;
+    }
+
+    public ConnectionThrottler(final SimpleHttpServer server, final Predicate<HttpExchange> counts){
+        this.server = server;
+        this.contributeToLimit = counts;
+    }
+
+
+    public synchronized final boolean addConnection(final HttpExchange exchange){
+        if(!contributeToLimit.test(exchange)){
+            return true;
+        }else if(connections.get() + 1 <= maxConnections){
+            connections.incrementAndGet();
+            return true;
+        }
+        return false;
+    }
+
+    public synchronized final void deleteConnection(final HttpExchange exchange){
+        if(contributeToLimit.test(exchange))
+            connections.decrementAndGet();
+    }
+
+    //
+
+
+    @Override
+    public String toString(){
+        final StringBuilder OUT = new StringBuilder();
+
+        OUT.append("ConnectionThrottler")   .append('{');
+        OUT.append("server")                .append('=')    .append(server.toString())  .append(", ");
+        OUT.append("connections")           .append('=')    .append(connections)        .append(", ");
+        OUT.append("maxConnections")        .append('=')    .append(maxConnections);
+        OUT.append('}');
+
+        return OUT.toString();
+    }
+
+}

src/main/java/com/kttdevelopment/simplehttpserver/handler/ThrottledHandler.java

@@ -0,0 +1,28 @@
+package com.kttdevelopment.simplehttpserver.handler;
+
+import com.sun.net.httpserver.HttpExchange;
+import com.sun.net.httpserver.HttpHandler;
+
+import java.io.IOException;
+
+public class ThrottledHandler implements HttpHandler {
+
+    private final HttpHandler handler;
+    private final ConnectionThrottler throttler;
+
+    public ThrottledHandler(final HttpHandler handler, final ConnectionThrottler throttler){
+        this.handler = handler;
+        this.throttler = throttler;
+    }
+
+    @Override
+    public synchronized final void handle(final HttpExchange exchange) throws IOException{
+        if(throttler.addConnection(exchange)){
+            handler.handle(exchange);
+            throttler.deleteConnection(exchange);
+        }else{
+            exchange.close();
+        }
+    }
+
+}