Merge pull request #9873 from stuartwdouglas/blocking-websockets

Allow websocket requests to run on a worker thread

Author: aloubyansky

extensions/undertow-websockets/deployment/src/main/java/io/quarkus/undertow/websockets/deployment/UndertowWebsocketProcessor.java

@@ -134,7 +134,8 @@ public ServletContextAttributeBuildItem deploy(final CombinedIndexBuildItem inde
                 new ReflectiveClassBuildItem(true, true, ClientEndpointConfig.Configurator.class.getName()));
 
         return new ServletContextAttributeBuildItem(WebSocketDeploymentInfo.ATTRIBUTE_NAME,
-                recorder.createDeploymentInfo(annotated, endpoints, config, websocketConfig.maxFrameSize));
+                recorder.createDeploymentInfo(annotated, endpoints, config, websocketConfig.maxFrameSize,
+                        websocketConfig.dispatchToWorker));
     }
 
     private void registerCodersForReflection(BuildProducer<ReflectiveClassBuildItem> reflection,

extensions/undertow-websockets/deployment/src/main/java/io/quarkus/undertow/websockets/deployment/WebsocketConfig.java

@@ -9,9 +9,16 @@ public class WebsocketConfig {
 
     /**
      * The maximum amount of data that can be sent in a single frame.
-     * 
+     *
      * Messages larger than this must be broken up into continuation frames.
      */
     @ConfigItem(defaultValue = "65536")
     public int maxFrameSize;
+
+    /**
+     * If the websocket methods should be run in a worker thread. This allows them to run
+     * blocking tasks, however it will not be as fast as running directly in the IO thread.
+     */
+    @ConfigItem(defaultValue = "false")
+    public boolean dispatchToWorker;
 }

extensions/undertow-websockets/runtime/src/main/java/io/quarkus/undertow/websockets/runtime/UndertowWebsocketRecorder.java

@@ -24,10 +24,10 @@ public void setupWorker(Executor executor) {
 
     @SuppressWarnings("unchecked")
     public WebSocketDeploymentInfo createDeploymentInfo(Set<String> annotatedEndpoints, Set<String> endpoints,
-            Set<String> serverApplicationConfigClasses, int maxFrameSize) {
+            Set<String> serverApplicationConfigClasses, int maxFrameSize, boolean dispatchToWorker) {
         WebSocketDeploymentInfo container = new WebSocketDeploymentInfo();
         container.setMaxFrameSize(maxFrameSize);
-        container.setDispatchToWorkerThread(false);
+        container.setDispatchToWorkerThread(dispatchToWorker);
         container.setExecutor(new ExecutorSupplier());
         Set<Class<? extends Endpoint>> allScannedEndpointImplementations = new HashSet<>();
         for (String i : endpoints) {