From 1c25d758140a75e1b93a3a5264b3e269ca7aae1d Mon Sep 17 00:00:00 2001
From: Ritesh Varyani <riteshvaryani@lyft.com>
Date: Wed, 3 Jul 2024 15:07:12 -0700
Subject: [PATCH] try request logging

---
 .../com/lyft/data/proxyserver/ProxyServer.java  | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 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
index b3ac5a42..1fe5d647 100644
--- a/proxyserver/src/main/java/com/lyft/data/proxyserver/ProxyServer.java
+++ b/proxyserver/src/main/java/com/lyft/data/proxyserver/ProxyServer.java
@@ -10,16 +10,22 @@
 
 import lombok.extern.slf4j.Slf4j;
 
+import org.apache.http.client.ResponseHandler;
 import org.apache.http.util.TextUtils;
+import org.eclipse.jetty.http.HttpParser;
 import org.eclipse.jetty.http.HttpScheme;
 import org.eclipse.jetty.http.HttpVersion;
 import org.eclipse.jetty.proxy.ConnectHandler;
+import org.eclipse.jetty.server.CustomRequestLog;
+import org.eclipse.jetty.server.Handler;
 import org.eclipse.jetty.server.HttpConfiguration;
 import org.eclipse.jetty.server.HttpConnectionFactory;
 import org.eclipse.jetty.server.SecureRequestCustomizer;
 import org.eclipse.jetty.server.Server;
 import org.eclipse.jetty.server.ServerConnector;
 import org.eclipse.jetty.server.SslConnectionFactory;
+import org.eclipse.jetty.server.handler.HandlerCollection;
+import org.eclipse.jetty.server.handler.RequestLogHandler;
 import org.eclipse.jetty.servlet.ServletContextHandler;
 import org.eclipse.jetty.servlet.ServletHolder;
 import org.eclipse.jetty.util.ssl.SslContextFactory;
@@ -97,7 +103,16 @@ private void setupContext(ProxyServerConfiguration config) {
 
     // Setup proxy handler to handle CONNECT methods
     ConnectHandler proxyConnectHandler = new ConnectHandler();
-    this.server.setHandler(proxyConnectHandler);
+
+    HandlerCollection handlers = new HandlerCollection();
+
+    RequestLogHandler requestLogHandler = new RequestLogHandler();
+    //possible not needed
+    //requestLogHandler.setRequestLog(customRequestLog);
+    handlers.setHandlers(new Handler[] { requestLogHandler, proxyConnectHandler });
+
+
+    this.server.setHandler(handlers);
 
     if (proxyHandler != null) {
       proxy.setProxyHandler(proxyHandler);