diff options
| -rw-r--r-- | src/main/java/com/geedgenetworks/utils/IPLocator.java | 9 | ||||
| -rw-r--r-- | src/test/java/com/geedgenetworks/test/IPBenchMark.java | 4 |
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); |