72 lines
2.4 KiB
Java
72 lines
2.4 KiB
Java
package cn.celess.blog.util;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.apache.commons.io.FileUtils;
|
|
import org.lionsoul.ip2region.DataBlock;
|
|
import org.lionsoul.ip2region.DbConfig;
|
|
import org.lionsoul.ip2region.DbSearcher;
|
|
import org.lionsoul.ip2region.Util;
|
|
|
|
import java.io.File;
|
|
import java.lang.reflect.Method;
|
|
import java.util.Objects;
|
|
|
|
/**
|
|
* @author : xiaohai
|
|
* @date : 2020/09/04 9:36
|
|
*/
|
|
|
|
@Slf4j
|
|
public class AddressUtil {
|
|
|
|
public static String getCityInfo(String ip) {
|
|
File file;
|
|
try {
|
|
//db
|
|
String dbPath = AddressUtil.class.getResource("/ip2region/ip2region.db").getPath();
|
|
file = new File(dbPath);
|
|
if (!file.exists()) {
|
|
String tmpDir = System.getProperties().getProperty("java.io.tmpdir");
|
|
dbPath = tmpDir + "ip.db";
|
|
file = new File(dbPath);
|
|
FileUtils.copyInputStreamToFile(Objects.requireNonNull(AddressUtil.class.getClassLoader().getResourceAsStream("classpath:ip2region/ip2region.db")), file);
|
|
}
|
|
//查询算法
|
|
//B-tree
|
|
int algorithm = DbSearcher.BTREE_ALGORITHM;
|
|
try {
|
|
DbConfig config = new DbConfig();
|
|
DbSearcher searcher = new DbSearcher(config, dbPath);
|
|
Method method = null;
|
|
switch (algorithm) {
|
|
case DbSearcher.BTREE_ALGORITHM:
|
|
method = searcher.getClass().getMethod("btreeSearch", String.class);
|
|
break;
|
|
case DbSearcher.BINARY_ALGORITHM:
|
|
method = searcher.getClass().getMethod("binarySearch", String.class);
|
|
break;
|
|
case DbSearcher.MEMORY_ALGORITYM:
|
|
method = searcher.getClass().getMethod("memorySearch", String.class);
|
|
break;
|
|
}
|
|
|
|
DataBlock dataBlock;
|
|
if (!Util.isIpAddress(ip)) {
|
|
System.out.println("Error: Invalid ip address");
|
|
}
|
|
|
|
dataBlock = (DataBlock) method.invoke(searcher, ip);
|
|
|
|
return dataBlock.getRegion();
|
|
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
}
|