Added prehandle

Author: Katsute

src/main/java/com/kttdevelopment/simplehttpserver/SimpleHttpServerImpl.java

@@ -37,7 +37,9 @@ static SimpleHttpServer createSimpleHttpServer(final Integer port, final Integer
 
             private final HttpServer server = HttpServer.create();
 
-            private final HashMap<HttpContext,HttpHandler> contexts = new HashMap<>();
+            private HttpSessionHandler sessionHandler;
+
+            private final Map<HttpContext,HttpHandler> contexts = new HashMap<>();
 
             private boolean running = false;
 
@@ -46,6 +48,11 @@ static SimpleHttpServer createSimpleHttpServer(final Integer port, final Integer
                     server.bind(new InetSocketAddress(port),backlog != null ? backlog : 0);
             }
 
+            private void handle(final HttpExchange exchange){
+                if(sessionHandler != null)
+                    sessionHandler.assignSession(exchange);
+            }
+
         //
 
             @Override
@@ -98,30 +105,54 @@ public final Executor getExecutor(){
                 return server.getExecutor();
             }
 
+            @Override
+            public synchronized final void setHttpSessionHandler(final HttpSessionHandler sessionHandler){
+                this.sessionHandler = sessionHandler;
+            }
+
+            @Override
+            public final HttpSessionHandler getHttpSessionHandler(){
+                return sessionHandler;
+            }
+
+            //
+
         //
 
             @Override
             public synchronized final HttpContext createContext(final String path){
-                final HttpContext context = server.createContext(getContext(path));
-                contexts.put(context,context.getHandler());
-                return context;
+                return createContext(path,(HttpExchange exchange) -> {});
             }
 
             @Override
             public synchronized final HttpContext createContext(final String path, final HttpHandler handler){
                 if(!getContext(path).equals("/") && handler instanceof RootHandler)
                     throw new IllegalArgumentException("RootHandler can only be used at the root '/' context");
-                final HttpContext context = server.createContext(getContext(path),handler);
+
+                final HttpHandler wrapper = exchange -> {
+                    handle(exchange);
+                    handler.handle(exchange);
+                };
+                final HttpContext context = server.createContext(getContext(path),wrapper);
+
                 contexts.put(context,handler);
+
                 return context;
             }
 
             @Override
             public synchronized final HttpContext createContext(final String path, final SimpleHttpHandler handler){
                 if(!getContext(path).equals("/") && handler instanceof RootHandler)
                     throw new IllegalArgumentException("RootHandler can only be used at the root '/' context");
-                final HttpContext context = server.createContext(getContext(path),(exchange) -> handler.handle(SimpleHttpExchange.create(exchange)));
+
+                final HttpHandler wrapper = exchange -> {
+                    handle(exchange);
+                    handler.handle(SimpleHttpExchange.create(exchange));
+                };
+                final HttpContext context = server.createContext(getContext(path),wrapper);
+
                 contexts.put(context,context.getHandler());
+
                 return context;
             }
 
@@ -249,11 +280,9 @@ public synchronized final void removeContext(final HttpContext context){
 
             @Override
             public final HttpHandler getContextHandler(final String path){
-                for(final HttpContext context : contexts.keySet()){
-                    if(context.getPath().equals(getContext(path))){
+                for(final HttpContext context : contexts.keySet())
+                    if(context.getPath().equals(getContext(path)))
                         return context.getHandler();
-                    }
-                }
                 return null;
             }