Resource文件读取

diagbotman-service/src/main/java/com/diagbot/util/IPUtil.java

@@ -5,8 +5,10 @@ import org.lionsoul.ip2region.DataBlock;
 import org.lionsoul.ip2region.DbConfig;
 import org.lionsoul.ip2region.DbSearcher;
 import org.lionsoul.ip2region.Util;
+import org.springframework.util.ResourceUtils;
 
 import java.io.File;
+import java.io.FileNotFoundException;
 import java.lang.reflect.Method;
 
 /**
@@ -20,11 +22,15 @@ public class IPUtil {
     public static String getCityInfo(String ip) {
 
         //db
-        String dbPath = IPUtil.class.getResource("/ip2region.db").getPath();
-
-        File file = new File(dbPath);
-        if (file.exists() == false) {
-            System.out.println("Error: Invalid ip2region.db file");
+        String dbPath = "";
+        try {
+            File file = ResourceUtils.getFile("classpath:ip2region.db");
+            if (file.exists() == false) {
+                System.out.println("Error: Invalid ip2region.db file");
+            }
+            dbPath = file.getPath();
+        } catch (FileNotFoundException e) {
+            System.out.println("文件读取错误");
         }
 
         //查询算法
@@ -60,6 +66,7 @@ public class IPUtil {
 
         } catch (Exception e) {
             e.printStackTrace();
+            System.out.println("Ip解析错误");
         }
 
         return null;