From 6d834d8754731d60e6964c26eea0ad3f45b6b86f Mon Sep 17 00:00:00 2001
From: Stuart Douglas <stuart.w.douglas@gmail.com>
Date: Thu, 2 Jul 2015 18:18:25 +0200
Subject: [PATCH] WFLY-4861 add http2 update listener to support upgrade
 requests as well as prior knowledge

---
 .../wildfly/extension/undertow/HttpListenerService.java  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/undertow/src/main/java/org/wildfly/extension/undertow/HttpListenerService.java b/undertow/src/main/java/org/wildfly/extension/undertow/HttpListenerService.java
index 2a977622f66c..33315d436aa0 100644
--- 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