[Fix][IPLookupV2] additional commitsV1.2.3

2 files changed, 9 insertions, 4 deletions

diff --git a/src/main/java/com/geedgenetworks/utils/IPLocator.java b/src/main/java/com/geedgenetworks/utils/IPLocator.java
index 445e91b..2d0c196 100644
--- a/src/main/java/com/geedgenetworks/utils/IPLocator.java
+++ b/src/main/java/com/geedgenetworks/utils/IPLocator.java
@@ -1,5 +1,8 @@
 package com.geedgenetworks.utils;
 
+import com.alibaba.fastjson2.JSONObject;
+import com.geedgenetworks.model.LocationResponse;
+
 public interface IPLocator {
 
     /**
@@ -125,14 +128,14 @@ public interface IPLocator {
      * @param ip
      * @return
      */
-    Object infoLookupToJSON(String ip);
+    JSONObject infoLookupToJSON(String ip);
 
     /**
-     * This method returns all fields for the MMDB Response Object..
+     * This method returns all fields for the MMDB Response Object.
      * @param ip
      * @return
      */
-    Object infoLookup(String ip);
+    LocationResponse infoLookup(String ip);
 
     /**
      *
diff --git a/src/test/java/com/geedgenetworks/test/IPBenchMark.java b/src/test/java/com/geedgenetworks/test/IPBenchMark.java
index 8a34e89..e0ce4b7 100644
--- a/src/test/java/com/geedgenetworks/test/IPBenchMark.java
+++ b/src/test/java/com/geedgenetworks/test/IPBenchMark.java
@@ -1,5 +1,7 @@
 package com.geedgenetworks.test;
 
+import com.alibaba.fastjson2.JSONObject;
+import com.geedgenetworks.model.LocationResponse;
 import com.geedgenetworks.utils.IpLookupV2;
 import org.apache.log4j.Logger;
 
@@ -73,7 +75,7 @@ public class IPBenchMark {
         for (int i = 0; i < count; i++) {
             random.nextBytes(address);
             InetAddress ip = InetAddress.getByAddress(address);
-            Object ipInfo = ipLookup.infoLookup(ip.getHostAddress());
+            LocationResponse ipInfo = ipLookup.infoLookup(ip.getHostAddress());
             if (TRACE) {
                 if (i % 50000 == 0) {
                     logger.info(i + " " + ip);