开发笔记:Hive的存储和MapReduce处理——数据清洗（Part3）

1 public class IPUtil {
2
3 public static String getCityInfo(String ip){
4
5 //db
6 //String dbPath = "data/ip2region.db";
7 String dbPath = IPUtil.class.getResource("../data/ip2region.db").getPath();
8
9 File file = new File(dbPath);
10 if ( file.exists() == false ) {
11 System.out.println("Error: Invalid ip2region.db file");
12 }
13
14 //查询算法
15 int algorithm = DbSearcher.BTREE_ALGORITHM; //B-tree
16 //DbSearcher.BINARY_ALGORITHM //Binary
17 //DbSearcher.MEMORY_ALGORITYM //Memory
18 try {
19 DbConfig cOnfig= new DbConfig();
20 DbSearcher searcher = new DbSearcher(config, dbPath);
21
22 //define the method
23 Method method = null;
24 switch ( algorithm )
25 {
26 case DbSearcher.BTREE_ALGORITHM:
27 method = searcher.getClass().getMethod("btreeSearch", String.class);
28 break;
29 case DbSearcher.BINARY_ALGORITHM:
30 method = searcher.getClass().getMethod("binarySearch", String.class);
31 break;
32 case DbSearcher.MEMORY_ALGORITYM:
33 method = searcher.getClass().getMethod("memorySearch", String.class);
34 break;
35 }
36
37 DataBlock dataBlock = null;
38 if ( Util.isIpAddress(ip) == false ) {
39 System.out.println("Error: Invalid ip address");
40 }
41
42 dataBlock = (DataBlock) method.invoke(searcher, ip);
43
44 return dataBlock.getRegion();
45
46 } catch (Exception e) {
47 e.printStackTrace();
48 }
49
50 return null;
51 }
52
53 public static void main(String[] args) throws Exception{
54 System.out.println(getCityInfo("106.39.41.166"));
55 }
56 }