Added an option to set the maximum worker threads

Author: JanHolger

src/main/java/org/javawebstack/httpserver/HTTPServer.java

@@ -54,6 +54,11 @@ public HTTPServer(IHTTPSocketServer server) {
         routeAutoInjectors.add(DefaultRouteAutoInjector.INSTANCE);
     }
 
+    public HTTPServer maxThreads(int maxThreads) {
+        this.server.setMaxThreads(maxThreads);
+        return this;
+    }
+
     public HTTPServer logger(Logger logger) {
         this.logger = logger;
         return this;

src/main/java/org/javawebstack/httpserver/adapter/IHTTPSocketServer.java

@@ -6,6 +6,7 @@ public interface IHTTPSocketServer {
 
     void setPort(int port);
     int getPort();
+    void setMaxThreads(int maxThreads);
     void start() throws IOException;
     void stop();
     void join();

src/main/java/org/javawebstack/httpserver/adapter/simple/SimpleHTTPSocketServer.java

@@ -7,15 +7,15 @@
 import java.io.IOException;
 import java.net.ServerSocket;
 import java.net.Socket;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
+import java.util.concurrent.*;
 
 public class SimpleHTTPSocketServer implements IHTTPSocketServer {
 
     private final Thread schedulerThread;
     private ExecutorService executorService;
     private ServerSocket serverSocket;
     private int port = 80;
+    private int maxThreads = 64;
     private IHTTPSocketHandler handler;
 
     public SimpleHTTPSocketServer() {
@@ -52,9 +52,13 @@ public void setHandler(IHTTPSocketHandler handler) {
         this.handler = handler;
     }
 
+    public void setMaxThreads(int maxThreads) {
+        this.maxThreads = maxThreads;
+    }
+
     public void start() throws IOException {
         this.serverSocket = new ServerSocket(port);
-        this.executorService = Executors.newCachedThreadPool();
+        this.executorService = new ThreadPoolExecutor(1, maxThreads, 60L, TimeUnit.SECONDS, new SynchronousQueue());
         this.schedulerThread.start();
     }
 

src/main/java/org/javawebstack/httpserver/adapter/undertow/UndertowHTTPSocketServer.java

@@ -11,12 +11,12 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
-import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
+import java.util.concurrent.*;
 
 public class UndertowHTTPSocketServer implements IHTTPSocketServer {
 
     private int port = 80;
+    private int maxThreads = 64;
     private Undertow server;
     private IHTTPSocketHandler handler;
     private ExecutorService executorService;
@@ -30,11 +30,12 @@ public int getPort() {
     }
 
     public void start() throws IOException {
-        executorService = Executors.newCachedThreadPool();
+        executorService = new ThreadPoolExecutor(1, maxThreads, 60L, TimeUnit.SECONDS, new SynchronousQueue());
         server = Undertow.builder()
                 .addHttpListener(port, "0.0.0.0")
                 .setServerOption(Options.KEEP_ALIVE, true)
                 .setHandler(new BlockingHandler(httpServerExchange -> {
+                    httpServerExchange.setDispatchExecutor(executorService);
                     if(httpServerExchange.getRequestHeaders().contains("sec-websocket-key")) {
                         httpServerExchange.upgradeChannel((streamConnection, httpServerExchange1) -> {
                             InputStream inputStream = new StreamSourceInputStream(streamConnection.getSourceChannel());
@@ -68,6 +69,10 @@ public void setHandler(IHTTPSocketHandler handler) {
         this.handler = handler;
     }
 
+    public void setMaxThreads(int maxThreads) {
+        this.maxThreads = maxThreads;
+    }
+
     public boolean isWebSocketSupported() {
         return true;
     }