X Tutup
import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.nio.ByteBuffer; import java.nio.ByteOrder; import java.nio.charset.Charset; import java.util.Arrays; import java.util.concurrent.locks.ReentrantLock; public class IPCounty { private static int offset; private static int[] index = new int[256]; private static ByteBuffer dataBuffer; private static ByteBuffer indexBuffer; private static File ipFile ; private static ReentrantLock lock; static { lock = new ReentrantLock(); } public static void main(String[] args) throws Exception { load("C:\\lovebizhi\\tiantexin\\framework\\library\\ip\\quxian.dat"); System.out.println(Arrays.toString(find("222.180.195.194"))); Long st = System.nanoTime(); for (int i = 0; i < 10000; i++) { try { find("222.180.195.194"); // } catch (Exception e) { e.printStackTrace(); } } Long et = System.nanoTime(); System.out.println((et - st) / 1000 / 1000); } public static void load(String name) { ipFile = new File(name); load(); } private static void load() { FileInputStream fin = null; lock.lock(); try { dataBuffer = ByteBuffer.allocate(Long.valueOf(ipFile.length()).intValue()); fin = new FileInputStream(ipFile); int readBytesLength; byte[] chunk = new byte[4096]; while (fin.available() > 0) { readBytesLength = fin.read(chunk); dataBuffer.put(chunk, 0, readBytesLength); } dataBuffer.position(0); int indexLength = dataBuffer.getInt(); byte[] indexBytes = new byte[indexLength]; dataBuffer.get(indexBytes, 0, indexLength - 4); indexBuffer = ByteBuffer.wrap(indexBytes); indexBuffer.order(ByteOrder.LITTLE_ENDIAN); offset = indexLength; int loop = 0; while (loop++ < 256) { index[loop - 1] = indexBuffer.getInt(); } indexBuffer.order(ByteOrder.BIG_ENDIAN); } catch (IOException ioe) { ioe.printStackTrace(); } finally { try { if (fin != null) { fin.close(); } } catch (IOException e){ e.printStackTrace(); } lock.unlock(); } } public static String[] find(String ip) throws Exception { int ip_prefix_value = new Integer(ip.substring(0, ip.indexOf("."))); long ip2long_value = ip2long(ip); int start = index[ip_prefix_value]; int max_comp_len = offset - 1028; long index_offset = -1; int index_length = -1; byte b = 0; for (start = start * 12 + 1024; start < max_comp_len; start += 12) { if (int2long(indexBuffer.getInt(start)) <= ip2long_value) { if (int2long(indexBuffer.getInt(start + 4)) >= ip2long_value) { index_offset = bytesToLong(b, indexBuffer.get(start + 10), indexBuffer.get(start + 9), indexBuffer.get(start + 8)); index_length = 0xFF & indexBuffer.get(start + 11); break; } } else { break; } } if (index_offset == -1 && index_length == -1) { throw new Exception("IP Data Not Found."); } byte[] areaBytes; lock.lock(); try { dataBuffer.position(offset + (int) index_offset - 1024); areaBytes = new byte[index_length]; dataBuffer.get(areaBytes, 0, index_length); } finally { lock.unlock(); } return new String(areaBytes, Charset.forName("UTF-8")).split("\t", -1); } private static long bytesToLong(byte a, byte b, byte c, byte d) { return int2long((((a & 0xff) << 24) | ((b & 0xff) << 16) | ((c & 0xff) << 8) | (d & 0xff))); } private static int str2Ip(String ip) { String[] ss = ip.split("\\."); int a, b, c, d; a = Integer.parseInt(ss[0]); b = Integer.parseInt(ss[1]); c = Integer.parseInt(ss[2]); d = Integer.parseInt(ss[3]); return (a << 24) | (b << 16) | (c << 8) | d; } private static long ip2long(String ip) { return int2long(str2Ip(ip)); } private static long int2long(int i) { long l = i & 0x7fffffffL; if (i < 0) { l |= 0x080000000L; } return l; } }
X Tutup