update

Author: ipipdotnet

src/main/java/net/ipip/ipdb/Risk.java

@@ -0,0 +1,65 @@
+package net.ipip.ipdb;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Map;
+
+public class Risk {
+
+    /**
+     * @var Reader
+     */
+    private Reader reader;
+
+    public Risk(String name) throws IOException,InvalidDatabaseException {
+        this.reader = new Reader(name);
+    }
+
+    public Risk(InputStream in) throws IOException, InvalidDatabaseException {
+        this.reader = new Reader(in);
+    }
+
+    public boolean reload(String name) {
+        try {
+            Reader r = new Reader(name);
+            this.reader = r;
+        } catch (Exception e) {
+            return false;
+        }
+
+        return true;
+    }
+
+    public Map<String, String> findMap(String addr, String language) throws IPFormatException, InvalidDatabaseException {
+        String[] data = this.reader.find(addr, language);
+        if (data == null) {
+            return null;
+        }
+
+        Map<String, String> m = new HashMap<String, String>();
+
+        String[] fields = this.reader.getSupportFields();
+
+        for (int i = 0, l = data.length; i < l; i++) {
+            m.put(fields[i], data[i]);
+        }
+
+        return m;
+    }
+
+    public RiskInfo findInfo(String addr) throws IPFormatException, InvalidDatabaseException {
+
+        Map<String, String> data = this.findMap(addr, "CN");
+        if (data == null) {
+            return null;
+        }
+
+        return new RiskInfo(data);
+    }
+
+    public String fields() {
+        return Arrays.toString(this.reader.getSupportFields());
+    }
+}

src/main/java/net/ipip/ipdb/RiskInfo.java

@@ -0,0 +1,38 @@
+package net.ipip.ipdb;
+
+import java.util.Map;
+
+public class RiskInfo {
+
+    private Map<String,String> data;
+
+    public RiskInfo(Map<String,String> map) {
+        this.data = map;
+    }
+
+    public int getScore() {
+        String s = this.data.get("score");
+        if (s == null) {
+            return 0;
+        }
+        return Integer.parseInt(s);
+    }
+
+    public String getBehavior() {
+        String s = this.data.get("behavior");
+        if (s == null) {
+            return "";
+        } else {
+            return s;
+        }
+    }
+
+    public String getCountryCode() {
+        String s = this.data.get("country_code");
+        if (s == null) {
+            return "";
+        } else {
+            return s;
+        }
+    }
+}

src/test/java/net/ipip/ipdb/RiskTest.java

@@ -0,0 +1,21 @@
+package net.ipip.ipdb;
+
+import org.junit.Test;
+
+import java.io.IOException;
+
+public class RiskTest {
+    @Test
+    public void testRisk() {
+        try {
+            Risk db = new Risk("c:/work/ipdb/v6risk.ipdb");
+
+            System.out.println(db.fields());
+
+            System.out.println(db.findInfo("2001:240:2a3b:7100::"));
+
+        } catch (Exception ioe) {
+            ioe.printStackTrace();
+        }
+    }
+}