根据IP地址查出对应城市

package com.jd.train.service.util;

import com.jd.common.util.StringUtils;
import com.jd.train.manager.util.impl.HttpClientUtilManagerImpl;
import org.apache.log4j.Logger;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;

import javax.servlet.http.HttpServletRequest;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.RandomAccessFile;

public class IPParser {
private final Logger log = Logger.getLogger(IPParser.class);
private String url; //ip查询请求地址
private String dbPath; //本地IP数据库地址
private String country, localStr;
private long IPN;
private int recordCount, countryFlag;
private long rangE, rangB, offSet, startIP, endIP, firstStartIP,lastStartIP, endIPOff;
private RandomAccessFile fis;
private byte[] buff;

private long B2L(byte[] b) {
long ret = 0;
for (int i = 0; i < b.length; i++) {
long t = 1L;
for (int j = 0; j < i; j++)
t = t * 256L;
ret += ((b[i] < 0) ? 256 + b[i] : b[i]) * t;
}
return ret;
}

private long ipToInt(String ip) {
String[] arr = ip.split("\\.");
long ret = 0;
for (int i = 0; i < arr.length; i++) {
long l = 1;
for (int j = 0; j < i; j++)
l *= 256;
try {
ret += Long.parseLong(arr[arr.length - i - 1]) * l;
} catch (Exception e) {
ret += 0;
}
}
return ret;
}

private void seek(String ip) throws Exception {
IPN = ipToInt(ip);
fis = new RandomAccessFile(dbPath, "r");
buff = new byte[4];
fis.seek(0);
fis.read(buff);
firstStartIP = B2L(buff);
fis.read(buff);
lastStartIP = B2L(buff);
recordCount = (int) ((lastStartIP - firstStartIP) / 7);
if (recordCount <= 1) {
localStr = country = "未知";
throw new Exception();
}

rangB = 0;
rangE = recordCount;
long RecNo;

do {
RecNo = (rangB + rangE) / 2;
getStartIP(RecNo);
if (IPN == startIP) {
rangB = RecNo;
break;
}
if (IPN > startIP)
rangB = RecNo;
else
rangE = RecNo;
} while (rangB < rangE - 1);

getStartIP(rangB);
getEndIP();
getCountry(IPN);

fis.close();
}

private String getFlagStr(long OffSet) throws IOException {
int flag = 0;
do {
fis.seek(OffSet);
buff = new byte[1];
fis.read(buff);
flag = (buff[0] < 0) ? 256 + buff[0] : buff[0];
if (flag == 1 || flag == 2) {
buff = new byte[3];
fis.read(buff);
if (flag == 2) {
countryFlag = 2;
endIPOff = OffSet - 4;
}
OffSet = B2L(buff);
} else
break;
} while (true);

if (OffSet < 12) {
return "";
} else {
fis.seek(OffSet);
return getStr();
}
}

private String getStr() throws IOException {
long l = fis.length();
ByteArrayOutputStream byteout = new ByteArrayOutputStream();
byte c = fis.readByte();
do {
byteout.write(c);
c = fis.readByte();
} while (c != 0 && fis.getFilePointer() < l);
return byteout.toString();
}

private void getCountry(long ip) throws IOException {
if (countryFlag == 1 || countryFlag == 2) {
country = getFlagStr(endIPOff + 4);
if (countryFlag == 1) {
localStr = getFlagStr(fis.getFilePointer());
if (IPN >= ipToInt("255.255.255.0")
&& IPN <= ipToInt("255.255.255.255")) {
localStr = getFlagStr(endIPOff + 21);
country = getFlagStr(endIPOff + 12);
}
} else {
localStr = getFlagStr(endIPOff + 8);
}
} else {
country = getFlagStr(endIPOff + 4);
localStr = getFlagStr(fis.getFilePointer());
}
}

private long getEndIP() throws IOException {
fis.seek(endIPOff);
buff = new byte[4];
fis.read(buff);
endIP = B2L(buff);
buff = new byte[1];
fis.read(buff);
countryFlag = (buff[0] < 0) ? 256 + buff[0] : buff[0];
return endIP;
}

private long getStartIP(long RecNo) throws IOException {
offSet = firstStartIP + RecNo * 7;
fis.seek(offSet);
buff = new byte[4];
fis.read(buff);
startIP = B2L(buff);
buff = new byte[3];
fis.read(buff);
endIPOff = B2L(buff);
return startIP;
}

private String getUserIp(HttpServletRequest request) {
String ip = CodesUtil.getIpAddr(request);
if (ip != null && ip.indexOf(",") > 0) {
log.info("取到客户多个ip1====================" + ip);
String[] arr = ip.split(",");
ip = arr[arr.length - 1].trim();//有多个ip时取最后一个ip
log.info("取到客户多个ip2====================" + ip);
}
return ip;
}

private String findCity(HttpServletRequest request , String ipStr) {
if(null == request && StringUtils.isEmpty(ipStr)){
log.info("IPParser,findIpAddr(), request and ip is empty !");
return null;
}

String ip = ipStr;
if(request != null){
ip = getUserIp(request);
}
if(StringUtils.isEmpty(ip)){
log.info("IPParser,findIpAddr(), ip is empty !");
return null;
}

String result;
String location;
try {
seek(ip.trim());
} catch (Exception e) {
log.error("IPParser,findIpAddr() error ! " + e);
}
location = country;
//查不到1
if (StringUtils.isEmpty(location)) {
log.info("IPParser,findIpAddr(), location is empty ! Trying remote interface ...");
location = findIpAddrRemote(ip);
if (StringUtils.isEmpty(location)) {
log.info("IPParser,findIpAddr(), location is empty !");
return null;
}
}
//查不到2
location = location.trim();
if (location.length() == 0) {
log.info("IPParser,findIpAddr(), location length is 0 ! Trying remote interface ...");
location = findIpAddrRemote(ip);
if (StringUtils.isEmpty(location)) {
log.info("IPParser,findIpAddr(), location is empty !");
return null;
}
if (location.length() == 0) {
log.info("IPParser,findIpAddr(), location length is 0 !");
return null;
}
}
//查不到3
if (location.contains("未知")) {
log.info("IPParser,findIpAddr(), location is not found ! Trying remote interface ...");
location = findIpAddrRemote(ip);
if (StringUtils.isEmpty(location)) {
log.info("IPParser,findIpAddr(), location is empty !");
return null;
}
if (location.contains("未知")) {
log.info("IPParser,findIpAddr(), location is not found !");
return null;
}
}

int cityIndex = location.indexOf("市");
int cityIndex2 = location.indexOf("市", cityIndex + 1);
int provinceIndex = location.indexOf("省");
if (cityIndex > -1 && cityIndex2 == -1) {
if (provinceIndex > -1) {
location = location.substring(provinceIndex + 1, cityIndex);
} else {
location = location.substring(0, cityIndex);
}
}
//处理有2个市的地址
if (cityIndex > -1 && cityIndex2 > -1) {
location = location.substring(cityIndex + 1, cityIndex2);
}

//自治区处理
if (location.contains("宁夏")) {
location = location.replaceAll("宁夏", "");
}
if (location.indexOf("内蒙古") > -1) {
location = location.replaceAll("内蒙古", "");
int districtIndex = location.indexOf("盟");
if (districtIndex > -1) {
location = location.substring(districtIndex + 1, location.length());
}
}
if (location.indexOf("广西") > -1) {
location = location.replaceAll("广西", "");
}
if (location.indexOf("西藏") > -1) {
location = location.replaceAll("西藏", "");
}
if (location.indexOf("新疆") > -1) {
location = location.replaceAll("新疆", "");
int districtIndex = location.indexOf("州");
if (districtIndex > -1) {
location = location.substring(districtIndex + 1, location.length());
}
}

//港澳台处理
if (location.contains("省")) {
location = location.replaceAll("省", "");
}

//其他
if (location.startsWith("清华大学")) {
location = "北京";
}

//去掉CZ88.NET字样，去掉空格
location = location.replaceAll("CZ88.NET", "");
result = location.replaceAll(" ", "");

log.info(ip + " --> " + location);
return result;
}

private String findIpAddrRemote(String ip) {
String result;
HttpClientUtilManagerImpl httpClientUtilManager = new HttpClientUtilManagerImpl();
httpClientUtilManager.setTimeout(1000); //超时时间 1s
String ipXML = httpClientUtilManager.executeHttpRequestString(url + ip, null);
if (ipXML == null) {
log.info("IPParser,findIpAddrRemote(),ipXML is null !");
return null;
}
Document doc;
try {
doc = DocumentHelper.parseText(ipXML.trim());
} catch (DocumentException e) {
log.error("IPParser,findIpAddrRemote() error !" + e);
return null;
}
if (doc == null) {
log.info("IPParser,findIpAddrRemote(),doc is null !");
return null;
}
Element rootElement = doc.getRootElement();
Element productElement = rootElement.element("product");
if (productElement == null) {
log.info("IPParser,findIpAddrRemote(),productElement is null !");
return null;
}
Element element = productElement.element("location");
if (element == null) {
log.info("IPParser,findIpAddrRemote(),element is null !");
return null;
}
result = element.getTextTrim();
return result;
}

/**
* 根据ip获得城市名称
* @param ip
* @return
*/
public String ipToCity(String ip){
return findCity(null , ip);
}

/**
* 根据请求获得城市名称
* @param request
* @return
*/
public String ipToCity(HttpServletRequest request){
return findCity(request, null);
}

public String getUrl() {
return url;
}

public void setUrl(String url) {
this.url = url;
}

public String getDbPath() {
return dbPath;
}

public void setDbPath(String dbPath) {
this.dbPath = dbPath;
}

public String getCountry() {
return country;
}

public void setCountry(String country) {
this.country = country;
}

public String getLocalStr() {
return localStr;
}

public void setLocalStr(String localStr) {
this.localStr = localStr;
}

public long getIPN() {
return IPN;
}

public void setIPN(long IPN) {
this.IPN = IPN;
}

public int getRecordCount() {
return recordCount;
}

public void setRecordCount(int recordCount) {
this.recordCount = recordCount;
}

public int getCountryFlag() {
return countryFlag;
}

public void setCountryFlag(int countryFlag) {
this.countryFlag = countryFlag;
}

public long getRangE() {
return rangE;
}

public void setRangE(long rangE) {
this.rangE = rangE;
}

public long getRangB() {
return rangB;
}

public void setRangB(long rangB) {
this.rangB = rangB;
}

public long getOffSet() {
return offSet;
}

public void setOffSet(long offSet) {
this.offSet = offSet;
}

public long getStartIP() {
return startIP;
}

public void setStartIP(long startIP) {
this.startIP = startIP;
}

public long getFirstStartIP() {
return firstStartIP;
}

public void setFirstStartIP(long firstStartIP) {
this.firstStartIP = firstStartIP;
}

public long getLastStartIP() {
return lastStartIP;
}

public void setLastStartIP(long lastStartIP) {
this.lastStartIP = lastStartIP;
}

public long getEndIPOff() {
return endIPOff;
}

public void setEndIPOff(long endIPOff) {
this.endIPOff = endIPOff;
}

public RandomAccessFile getFis() {
return fis;
}

public void setFis(RandomAccessFile fis) {
this.fis = fis;
}

public byte[] getBuff() {
return buff;
}

public void setBuff(byte[] buff) {
this.buff = buff;
}

public static void main(String[] args) throws Exception {
long initUsedMemory = (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory());
long start = System.currentTimeMillis();

IPParser w = new IPParser();
// w.setPath(new File("QQWry2.Dat").getAbsolutePath());
//		w.seek("125.10.100.17");
//        String ip = "211.161.190.171";
//        String ip = "14.192.60.1";
//        w.findRequestPhysAddr(ip);

long end = System.currentTimeMillis();
long endUsedMemory = (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory());
System.out.println("time spent:" + (end - start) + " ms");
System.out.println("memory consumes:" + (endUsedMemory - initUsedMemory));

}

}