GitHub-issue#253 : Implemented GeoIP processor functionality. Address…

…ed Code review comments Signed-off-by: venkataraopasyavula <[email protected]>
opensearch-project · Jul 10, 2023 · 2599461 · 2599461
1 parent 456560b
commit 2599461
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.../main/java/org/opensearch/dataprepper/plugins/processor/databaseenrich/GetGeoIP2Data.java b/.../main/java/org/opensearch/dataprepper/plugins/processor/databaseenrich/GetGeoIP2Data.java
@@ -121,7 +121,7 @@ public Map<String, Object> getGeoData(InetAddress inetAddress, List<String> attr
                 for (String attribute : attributes) {
                     switch (attribute) {
                         case IP:
-                            enrichData(geoData, IP, inetAddress.toString());
+                            enrichData(geoData, IP, inetAddress.getHostAddress());
                             break;
                         case COUNTRY_ISO_CODE:
                             enrichData(geoData, COUNTRY_ISO_CODE, country.getIsoCode());
@@ -156,7 +156,7 @@ public Map<String, Object> getGeoData(InetAddress inetAddress, List<String> attr
                 }
             } else {
 
-                enrichData(geoData, IP, inetAddress.toString());
+                enrichData(geoData, IP, inetAddress.getHostAddress());
                 enrichData(geoData, COUNTRY_ISO_CODE, country.getIsoCode());
                 enrichData(geoData, COUNTRY_NAME, country.getName());
                 enrichData(geoData, CONTINENT_NAME, continent.getName());

diff --git a/...ain/java/org/opensearch/dataprepper/plugins/processor/databaseenrich/GetGeoLite2Data.java b/...ain/java/org/opensearch/dataprepper/plugins/processor/databaseenrich/GetGeoLite2Data.java
@@ -145,7 +145,7 @@ public Map<String, Object> getGeoData(InetAddress inetAddress, List<String> attr
                 for (String attribute : attributes) {
                     switch (attribute) {
                         case IP:
-                            enrichData(geoData, IP, inetAddress.toString());
+                            enrichData(geoData, IP, inetAddress.getHostAddress());
                             break;
                         case COUNTRY_ISO_CODE:
                             enrichData(geoData, COUNTRY_ISO_CODE, country.getIsoCode());
@@ -188,7 +188,7 @@ public Map<String, Object> getGeoData(InetAddress inetAddress, List<String> attr
                 }
             } else {
 
-                enrichData(geoData, IP, inetAddress.toString());
+                enrichData(geoData, IP, inetAddress.getHostAddress());
                 enrichData(geoData, COUNTRY_ISO_CODE, country.getIsoCode());
                 enrichData(geoData, COUNTRY_NAME, country.getName());
                 enrichData(geoData, CONTINENT_NAME, continent.getName());