update throttler

Author: Katsute

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

@@ -2,6 +2,9 @@
 
 import com.sun.net.httpserver.HttpExchange;
 
+/**
+ *
+ */
 abstract class ConnectionThrottler {
 
     abstract boolean addConnection(final HttpExchange exchange);

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

@@ -1,7 +1,6 @@
 package com.kttdevelopment.simplehttpserver.handler;
 
-import com.kttdevelopment.simplehttpserver.HttpSession;
-import com.kttdevelopment.simplehttpserver.SimpleHttpServer;
+import com.kttdevelopment.simplehttpserver.*;
 import com.sun.net.httpserver.HttpExchange;
 
 import java.util.*;
@@ -10,38 +9,38 @@
 
 public class SessionThrottler extends ConnectionThrottler {
 
-    private final SimpleHttpServer server;
+    private final HttpSessionHandler sessionHandler;
 
     private final Predicate<HttpSession> usesLimit;
     private final Map<HttpSession, AtomicInteger> sessions = Collections.synchronizedMap(new HashMap<>());
 
     private int maxConnections = 0;
 
-    public SessionThrottler(final SimpleHttpServer server){
-        this.server = server;
+    public SessionThrottler(final HttpSessionHandler sessionHandler){
+        this.sessionHandler = sessionHandler;
         usesLimit = (exchange) -> true;
     }
 
-    public SessionThrottler(final SimpleHttpServer server, final int maxConnections){
-        this.server = server;
+    public SessionThrottler(final HttpSessionHandler sessionHandler, final int maxConnections){
+        this.sessionHandler = sessionHandler;
         usesLimit = (exchange) -> true;
         this.maxConnections = maxConnections;
     }
 
-    public SessionThrottler(final SimpleHttpServer server, final Predicate<HttpSession> usesLimit){
-        this.server = server;
+    public SessionThrottler(final HttpSessionHandler sessionHandler, final Predicate<HttpSession> usesLimit){
+        this.sessionHandler = sessionHandler;
         this.usesLimit = usesLimit;
     }
 
-    public SessionThrottler(final SimpleHttpServer server, final Predicate<HttpSession> usesLimit, final int maxConnections){
-        this.server = server;
+    public SessionThrottler(final HttpSessionHandler sessionHandler, final Predicate<HttpSession> usesLimit, final int maxConnections){
+        this.sessionHandler = sessionHandler;
         this.usesLimit = usesLimit;
         this.maxConnections = maxConnections;
     }
 
     @Override
     final boolean addConnection(final HttpExchange exchange){
-        final HttpSession session = server.getHttpSession(exchange);
+        final HttpSession session = sessionHandler.getSession(exchange);
         if(!usesLimit.test(session)){
             return true;
         }else{
@@ -58,7 +57,7 @@ final boolean addConnection(final HttpExchange exchange){
 
     @Override
     final void deleteConnection(final HttpExchange exchange){
-        final HttpSession session = server.getHttpSession(exchange);
+        final HttpSession session = sessionHandler.getSession(exchange);
         if(usesLimit.test(session))
             synchronized(this){
                 sessions.get(session).decrementAndGet();