diff options
Diffstat (limited to 'src/main/java')
| -rw-r--r-- | src/main/java/cn/ac/iie/bolt/segmentation/InternalBolt.java | 1 | ||||
| -rw-r--r-- | src/main/java/cn/ac/iie/utils/TopUtils.java | 15 |
2 files changed, 10 insertions, 6 deletions
diff --git a/src/main/java/cn/ac/iie/bolt/segmentation/InternalBolt.java b/src/main/java/cn/ac/iie/bolt/segmentation/InternalBolt.java index 09f5f34..c765135 100644 --- a/src/main/java/cn/ac/iie/bolt/segmentation/InternalBolt.java +++ b/src/main/java/cn/ac/iie/bolt/segmentation/InternalBolt.java @@ -46,5 +46,4 @@ public class InternalBolt extends BaseBasicBolt { } - } diff --git a/src/main/java/cn/ac/iie/utils/TopUtils.java b/src/main/java/cn/ac/iie/utils/TopUtils.java index c145fd1..5ada340 100644 --- a/src/main/java/cn/ac/iie/utils/TopUtils.java +++ b/src/main/java/cn/ac/iie/utils/TopUtils.java @@ -8,15 +8,20 @@ import com.zdjizhi.utils.IpLookup; */ public class TopUtils { private static IpLookup ipLookup = new IpLookup.Builder(false) - .loadDataFileV4(TopNCountConfig.IP_LIBRARY + "Kazakhstan.mmdb") + .loadDataFileV4("Kazakhstan.mmdb") .loadDataFileV6(TopNCountConfig.IP_LIBRARY + "Kazakhstan.mmdb") .build(); public static boolean isKazakhstan(String ip) { - String[] ipAddress = ipLookup.cityLookupDetail(ip).split("\t"); - String country = ipAddress[2]; - return "Kazakhstan".equals(country) || "N/A".equals(country); -// return "China".equals(country) || "N/A".equals(country); + String address = ipLookup.cityLookupDetail(ip); + String pri = "Private IP"; + if (pri.equals(address)) { + return true; + } else { + String[] ipAddress = address.split("\t"); + String country = ipAddress[2]; + return "Kazakhstan".equals(country); + } } } |