Merge pull request #1753 from akto-api-security/feature/add_api_count…

…er_logging add more logs
akto-api-security · Nov 21, 2024 · 5f63268 · 5f63268
2 parents c323a34 + 54fa1a2
commit 5f63268
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/apps/dashboard/src/main/java/com/akto/filter/LoggingFilter.java b/apps/dashboard/src/main/java/com/akto/filter/LoggingFilter.java
@@ -1,5 +1,6 @@
 package com.akto.filter;
 
+import com.akto.dao.context.Context;
 import com.akto.dto.User;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -9,16 +10,20 @@
 import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpSession;
 import java.io.IOException;
+import java.util.concurrent.atomic.AtomicInteger;
 
 public class LoggingFilter implements Filter {
 
     private static final Logger logger = LoggerFactory.getLogger(LoggingFilter.class);
+    private static AtomicInteger apiCounter = new AtomicInteger();
 
     @Override
     public void init(FilterConfig filterConfig) { }
 
     @Override
     public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
+        int startTs = Context.now();
+        apiCounter.incrementAndGet();
         chain.doFilter(request, response);
 
         try {
@@ -28,7 +33,7 @@ public void doFilter(ServletRequest request, ServletResponse response, FilterCha
             int statusCode = httpServletResponse.getStatus();
             String uri = httpServletRequest.getRequestURI();
             if (uri.contains("fetchActiveLoaders") ||
-                    uri.contains("fetchActiveTestRunsStatus") || uri.contains("metrics")) {
+                    uri.contains("fetchActiveTestRunsStatus") || uri.contains("metrics") || uri.contains("favicon")) {
                 return;
             }
             String method = httpServletRequest.getMethod();
@@ -46,12 +51,14 @@ public void doFilter(ServletRequest request, ServletResponse response, FilterCha
                 ip = httpServletRequest.getRemoteAddr();
             }
 
-            String result = "url="+uri + ";method="+method + ";statusCode="+statusCode + ";username="+username + ";ip="+ ip;
-
+            int endTs = Context.now();
+            String result = "url="+uri + ";method="+method + ";statusCode="+statusCode + ";username="+username + ";ip="+ ip + ";totalTime=" + (endTs - startTs) + ";apiCounterVal=" + apiCounter.get();
             logger.info(result);
 
         } catch (Exception e) {
             logger.error("Error: ", e);
+        } finally {
+            apiCounter.decrementAndGet();
         }
 
     }