mirror of https://gitee.com/xiaonuobase/snowy
109 lines
3.8 KiB
Java
109 lines
3.8 KiB
Java
/*
|
||
* Copyright [2022] [https://www.xiaonuo.vip]
|
||
*
|
||
* Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
|
||
*
|
||
* 1.请不要删除和修改根目录下的LICENSE文件。
|
||
* 2.请不要删除和修改Snowy源码头部的版权声明。
|
||
* 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
|
||
* 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
|
||
* 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
|
||
* 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
|
||
*/
|
||
package vip.xiaonuo.common.util;
|
||
|
||
import cn.hutool.core.io.FileUtil;
|
||
import cn.hutool.core.net.Ipv4Util;
|
||
import cn.hutool.core.util.ObjectUtil;
|
||
import cn.hutool.extra.servlet.ServletUtil;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.lionsoul.ip2region.xdb.Searcher;
|
||
import vip.xiaonuo.common.exception.CommonException;
|
||
|
||
import javax.servlet.http.HttpServletRequest;
|
||
import java.io.File;
|
||
import java.io.InputStream;
|
||
|
||
/**
|
||
* 根据ip地址定位工具类，离线方式
|
||
* 参考地址：https://gitee.com/lionsoul/ip2region/tree/master/binding/java
|
||
*
|
||
* @author xuyuxiang
|
||
* @date 2020/3/16 11:25
|
||
*/
|
||
@Slf4j
|
||
public class CommonIpAddressUtil {
|
||
|
||
private static final String LOCAL_REMOTE_HOST = "0:0:0:0:0:0:0:1";
|
||
|
||
private static final Searcher searcher;
|
||
|
||
static {
|
||
String fileName = "/ip2region.xdb";
|
||
File existFile = FileUtil.file(FileUtil.getTmpDir() + FileUtil.FILE_SEPARATOR + fileName);
|
||
if(!FileUtil.exist(existFile)) {
|
||
InputStream resourceAsStream = CommonIpAddressUtil.class.getResourceAsStream(fileName);
|
||
if(ObjectUtil.isEmpty(resourceAsStream)) {
|
||
throw new CommonException("CommonIpAddressUtil初始化失败，原因：IP地址库数据不存在");
|
||
}
|
||
FileUtil.writeFromStream(resourceAsStream, existFile);
|
||
}
|
||
|
||
String dbPath = existFile.getPath();
|
||
|
||
// 1、从 dbPath 加载整个 xdb 到内存。
|
||
byte[] cBuff;
|
||
try {
|
||
cBuff = Searcher.loadContentFromFile(dbPath);
|
||
} catch (Exception e) {
|
||
log.error(">>> CommonIpAddressUtil初始化异常：", e);
|
||
throw new CommonException("CommonIpAddressUtil初始化异常");
|
||
}
|
||
|
||
// 2、使用上述的 cBuff 创建一个完全基于内存的查询对象。
|
||
try {
|
||
searcher = Searcher.newWithBuffer(cBuff);
|
||
} catch (Exception e) {
|
||
log.error(">>> CommonIpAddressUtil初始化异常：", e);
|
||
throw new CommonException("CommonIpAddressUtil初始化异常");
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取客户端ip
|
||
*
|
||
* @author xuyuxiang
|
||
* @date 2020/3/19 9:32
|
||
*/
|
||
public static String getIp(HttpServletRequest request) {
|
||
if (ObjectUtil.isEmpty(request)) {
|
||
return Ipv4Util.LOCAL_IP;
|
||
} else {
|
||
try {
|
||
String remoteHost = ServletUtil.getClientIP(request);
|
||
return LOCAL_REMOTE_HOST.equals(remoteHost) ? Ipv4Util.LOCAL_IP : remoteHost;
|
||
} catch (Exception e) {
|
||
log.error(">>> 获取客户端ip异常：", e);
|
||
return Ipv4Util.LOCAL_IP;
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 根据IP地址离线获取城市
|
||
*
|
||
* @author xuyuxiang
|
||
* @date 2022/4/27 23:14
|
||
*/
|
||
public static String getCityInfo(String ip) {
|
||
try {
|
||
ip = ip.trim();
|
||
// 3、执行查询
|
||
String region = searcher.searchByStr(ip);
|
||
return region.replace("0|", "").replace("|0", "");
|
||
} catch (Exception e) {
|
||
return "未知";
|
||
}
|
||
}
|
||
}
|