兼容本地绝对路径文件读取

jodconverter-web/src/main/java/cn/keking/utils/DownloadUtils.java

@@ -33,6 +33,7 @@ public class DownloadUtils {
     private static final String URL_PARAM_FTP_USERNAME = "ftp.username";
     private static final String URL_PARAM_FTP_PASSWORD = "ftp.password";
     private static final String URL_PARAM_FTP_CONTROL_ENCODING = "ftp.control.encoding";
+    private static final String[] URL_PARAM_LOCALURL_START={"c:","d:","e:","f:","g:","/"};
 
     /**
      * @param fileAttribute fileAttribute
@@ -55,6 +56,11 @@ public class DownloadUtils {
             dirFile.mkdirs();
         }
         try {
+            if(isLocalUrl(urlStr)){
+                byte[] bytes = getBytesFromLocalUrl(urlStr);
+                OutputStream os = new FileOutputStream(new File(realPath));
+                saveBytesToOutStream(bytes, os);
+            }else{
                 URL url = new URL(urlStr);
                 if (url.getProtocol() != null && url.getProtocol().toLowerCase().startsWith("http")) {
                     byte[] bytes = getBytesFromUrl(urlStr);
@@ -70,6 +76,7 @@ public class DownloadUtils {
                     response.setContent(null);
                     response.setMsg("url不能识别url" + urlStr);
                 }
+            }
             response.setContent(realPath);
             response.setMsg(fileName);
             if(FileType.simText.equals(fileAttribute.getType())){
@@ -104,6 +111,21 @@ public class DownloadUtils {
         }
     }
 
+    public byte[] getBytesFromLocalUrl(String urlStr) throws IOException {
+        File file = new File(urlStr);
+        FileInputStream fis = new FileInputStream(file);
+        ByteArrayOutputStream bos = new ByteArrayOutputStream(1024);
+        byte[] b = new byte[1024];
+        int n;
+        while ((n = fis.read(b)) != -1) {
+            bos.write(b, 0, n);
+        }
+        fis.close();
+        byte[] data = bos.toByteArray();
+        bos.close();
+        return data;
+    }
+
     public void saveBytesToOutStream(byte[] b, OutputStream os) throws IOException {
         os.write(b);
         os.close();
@@ -173,4 +195,16 @@ public class DownloadUtils {
       }
     }
   }
+
+  /**
+  * @param url url
+  * @return 是否是本地文件的绝对路径
+  */
+  private boolean isLocalUrl(String url){
+      for(String str:URL_PARAM_LOCALURL_START){
+          if(url.toLowerCase().startsWith(str))
+              return true;
+      }
+      return false;
+  }
 }