add stats handler

lyft · Jul 3, 2024 · da0eb30 · da0eb30
1 parent 2cb6d66
commit da0eb30
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/proxyserver/src/main/java/com/lyft/data/proxyserver/ProxyServer.java b/proxyserver/src/main/java/com/lyft/data/proxyserver/ProxyServer.java
@@ -20,6 +20,7 @@
 import org.eclipse.jetty.server.Server;
 import org.eclipse.jetty.server.ServerConnector;
 import org.eclipse.jetty.server.SslConnectionFactory;
+import org.eclipse.jetty.server.handler.StatisticsHandler;
 import org.eclipse.jetty.servlet.ServletContextHandler;
 import org.eclipse.jetty.servlet.ServletHolder;
 import org.eclipse.jetty.util.ssl.SslContextFactory;
@@ -97,7 +98,10 @@ private void setupContext(ProxyServerConfiguration config) {
 
     // Setup proxy handler to handle CONNECT methods
     ConnectHandler proxyConnectHandler = new ConnectHandler();
-    this.server.setHandler(proxyConnectHandler);
+    StatisticsHandler proxyStatsHandler = new StatisticsHandler();
+
+    proxyStatsHandler.setHandler(proxyConnectHandler);
+    this.server.setHandler(proxyStatsHandler);
 
     if (proxyHandler != null) {
       proxy.setProxyHandler(proxyHandler);