Extend from AbstractHandlerContainer instead of AbstractHandler (#2460)

dropwizard · Aug 23, 2018 · 5a78323 · 5a78323
1 parent 1ee6401
commit 5a78323
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/dropwizard-jetty/src/main/java/io/dropwizard/jetty/ContextRoutingHandler.java b/dropwizard-jetty/src/main/java/io/dropwizard/jetty/ContextRoutingHandler.java
@@ -2,9 +2,8 @@
 
 import org.eclipse.jetty.server.Handler;
 import org.eclipse.jetty.server.Request;
-import org.eclipse.jetty.server.handler.AbstractHandler;
+import org.eclipse.jetty.server.handler.AbstractHandlerContainer;
 import org.eclipse.jetty.util.ArrayTernaryTrie;
-import org.eclipse.jetty.util.Trie;
 
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;
@@ -15,8 +14,8 @@
 /**
  * A Jetty router which routes requests based on context path.
  */
-public class ContextRoutingHandler extends AbstractHandler {
+public class ContextRoutingHandler extends AbstractHandlerContainer {
-    private final Trie<Handler> handlers;
+    private final ArrayTernaryTrie<Handler> handlers;
 
     public ContextRoutingHandler(Map<String, ? extends Handler> handlers) {
         this.handlers = new ArrayTernaryTrie<>(false);
@@ -54,4 +53,9 @@ protected void doStop() throws Exception {
             handlers.get(key).stop();
         }
     }
+
+    @Override
+    public Handler[] getHandlers() {
+        return handlers.entrySet().stream().map(e -> e.getValue()).toArray(Handler[]::new);
+    }
 }