From 4ae99b7d87644c70e9a778c79b0dc2edaf826116 Mon Sep 17 00:00:00 2001
From: Corey <1339555+CoreyD97@users.noreply.github.com>
Date: Thu, 26 Oct 2023 10:55:32 +0100
Subject: [PATCH] Partial implementation for #109. Thanks to @justinsteven.

---
 .../logview/processor/LogProcessor.java             | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/main/java/com/nccgroup/loggerplusplus/logview/processor/LogProcessor.java b/src/main/java/com/nccgroup/loggerplusplus/logview/processor/LogProcessor.java
index 037110bd..c2c37bf9 100644
--- a/src/main/java/com/nccgroup/loggerplusplus/logview/processor/LogProcessor.java
+++ b/src/main/java/com/nccgroup/loggerplusplus/logview/processor/LogProcessor.java
@@ -234,6 +234,13 @@ private void updateRequestWithResponse(Integer entryIdentifier, Date arrivalTime
         if (entriesPendingProcessing.containsKey(entryIdentifier)) {
             //Not yet started processing the entry, we can add the response so it is processed in the first pass
             final LogEntry logEntry = entriesPendingProcessing.get(entryIdentifier);
+
+            if (response instanceof InterceptedResponse) {
+                // Raises java.lang.UnsupportedOperationException: Not yet implemented
+//                logEntry.setClientIP(String.valueOf(((InterceptedResponse) response).sourceIpAddress()));
+                logEntry.setListenerInterface(((InterceptedResponse) response).listenerInterface());
+            }
+
             //Update the response with the new one, and tell it when it arrived.
             logEntry.addResponse(response, arrivalTime);
 
@@ -330,6 +337,12 @@ private RunnableFuture<LogEntry> createEntryUpdateRunnable(final Future<LogEntry
             }
 
             //Request was processed successfully... now process the response.
+            if (requestResponse instanceof InterceptedResponse) {
+                // Raises java.lang.UnsupportedOperationException: Not yet implemented
+//                logEntry.setClientIP(String.valueOf(((InterceptedResponse) requestResponse).sourceIpAddress()));
+                logEntry.setListenerInterface(((InterceptedResponse) requestResponse).listenerInterface());
+            }
+
             logEntry.addResponse(requestResponse, arrivalTime);
             LogEntry updatedEntry = processEntry(logEntry);