Merge pull request #7698 from stuartwdouglas/WFLY-4861

WFLY-4861 add http2 update listener to support upgrade requests as we…
wildfly · Jul 6, 2015 · 50fe22f · 50fe22f
2 parents 444a8d3 + 6d834d8
commit 50fe22f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/undertow/src/main/java/org/wildfly/extension/undertow/HttpListenerService.java b/undertow/src/main/java/org/wildfly/extension/undertow/HttpListenerService.java
@@ -34,6 +34,7 @@
 import io.undertow.server.handlers.ProxyPeerAddressHandler;
 import io.undertow.server.handlers.SSLHeaderHandler;
 import io.undertow.server.protocol.http.HttpOpenListener;
+import io.undertow.server.protocol.http2.Http2UpgradeHandler;
 import org.jboss.msc.service.ServiceName;
 import org.jboss.msc.service.StartContext;
 import org.jboss.msc.service.ValueService;
@@ -71,6 +72,14 @@ public HttpHandler wrap(final HttpHandler handler) {
                 return httpUpgradeHandler;
             }
         });
+        if(listenerOptions.get(UndertowOptions.ENABLE_HTTP2, false)) {
+            addWrapperHandler(new HandlerWrapper() {
+                @Override
+                public HttpHandler wrap(HttpHandler handler) {
+                    return new Http2UpgradeHandler(handler);
+                }
+            });
+        }
         if (certificateForwarding) {
             addWrapperHandler(new HandlerWrapper() {
                 @Override