From 7265ef3c86d3e817ff30edc86b4a19ac4e495308 Mon Sep 17 00:00:00 2001
From: Andy Kwok <andy.kwok@improving.com>
Date: Mon, 25 Nov 2024 10:10:35 -0800
Subject: [PATCH] Style fix

Signed-off-by: Andy Kwok <andy.kwok@improving.com>
---
 .../opensearch/geospatial/action/IpEnrichmentResponse.java  | 6 +++---
 .../ip2geo/action/IpEnrichmentTransportAction.java          | 3 +--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/client/src/main/java/org/opensearch/geospatial/action/IpEnrichmentResponse.java b/client/src/main/java/org/opensearch/geospatial/action/IpEnrichmentResponse.java
index 0a84ff8f..826e8249 100644
--- a/client/src/main/java/org/opensearch/geospatial/action/IpEnrichmentResponse.java
+++ b/client/src/main/java/org/opensearch/geospatial/action/IpEnrichmentResponse.java
@@ -36,24 +36,24 @@ public class IpEnrichmentResponse extends ActionResponse {
     private Map<String, Object> geoLocationData;
 
     /**
-     * Private method to be used by fromActionResponse call to populate this Response class.
+     * Public method to be called by fromActionResponse( ) to populate this Response class.
      * @param streamInput Stream object which contain the geoLocationData.
      * @throws IOException Exception being thrown when given stremInput doesn't contain what IpEnrichmentResponse is expecting.
      */
     public IpEnrichmentResponse(StreamInput streamInput) throws IOException {
         super(streamInput);
         geoLocationData = streamInput.readMap();
+        log.trace("Constructing IP Enrichment response with values: [{}]", geoLocationData);
     }
 
     /**
-     * Overridden method used by OpenSearch runtime to write result into listener.
+     * Overridden method used by OpenSearch runtime to serialise this class content into stream.
      * @param streamOutput the streamOutput used to construct this response object.
      * @throws IOException the IOException.
      */
     @Override
     public void writeTo(StreamOutput streamOutput) throws IOException {
         streamOutput.writeMap(geoLocationData);
-        log.trace("Constructing IP Enrichment response with values: [{}]", geoLocationData);
     }
 
     /**
diff --git a/src/main/java/org/opensearch/geospatial/ip2geo/action/IpEnrichmentTransportAction.java b/src/main/java/org/opensearch/geospatial/ip2geo/action/IpEnrichmentTransportAction.java
index ff80c51d..aed46866 100644
--- a/src/main/java/org/opensearch/geospatial/ip2geo/action/IpEnrichmentTransportAction.java
+++ b/src/main/java/org/opensearch/geospatial/ip2geo/action/IpEnrichmentTransportAction.java
@@ -34,8 +34,6 @@ public class IpEnrichmentTransportAction extends HandledTransportAction<ActionRe
 
     private final Ip2GeoCachedDao ip2GeoCachedDao;
 
-    // private final String defaultDataSourceName;
-
     private final DatasourceDao datasourceDao;
 
     /**
@@ -43,6 +41,7 @@ public class IpEnrichmentTransportAction extends HandledTransportAction<ActionRe
      * @param transportService the transport service
      * @param actionFilters the action filters
      * @param cachedDao the cached datasource facade
+     * @param datasourceDao the datasourceDao to read the default datasource from
      */
     @Inject
     public IpEnrichmentTransportAction(