兼容本地绝对路径文件读取,采用file标准协议读取文件

pull/19/head
13540823418 4 years ago
parent 4c708f3cbd
commit fb09a8c00f

@ -33,7 +33,6 @@ public class DownloadUtils {
private static final String URL_PARAM_FTP_USERNAME = "ftp.username"; 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_PASSWORD = "ftp.password";
private static final String URL_PARAM_FTP_CONTROL_ENCODING = "ftp.control.encoding"; 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 * @param fileAttribute fileAttribute
@ -56,13 +55,8 @@ public class DownloadUtils {
dirFile.mkdirs(); dirFile.mkdirs();
} }
try { try {
if(isLocalUrl(urlStr)){
byte[] bytes = getBytesFromLocalUrl(urlStr);
OutputStream os = new FileOutputStream(new File(realPath));
saveBytesToOutStream(bytes, os);
}else{
URL url = new URL(urlStr); URL url = new URL(urlStr);
if (url.getProtocol() != null && url.getProtocol().toLowerCase().startsWith("http")) { if (url.getProtocol() != null && (url.getProtocol().toLowerCase().startsWith("file")||url.getProtocol().toLowerCase().startsWith("http"))) {
byte[] bytes = getBytesFromUrl(urlStr); byte[] bytes = getBytesFromUrl(urlStr);
OutputStream os = new FileOutputStream(new File(realPath)); OutputStream os = new FileOutputStream(new File(realPath));
saveBytesToOutStream(bytes, os); saveBytesToOutStream(bytes, os);
@ -76,7 +70,6 @@ public class DownloadUtils {
response.setContent(null); response.setContent(null);
response.setMsg("url不能识别url" + urlStr); response.setMsg("url不能识别url" + urlStr);
} }
}
response.setContent(realPath); response.setContent(realPath);
response.setMsg(fileName); response.setMsg(fileName);
if(FileType.simText.equals(fileAttribute.getType())){ if(FileType.simText.equals(fileAttribute.getType())){
@ -111,21 +104,6 @@ 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 { public void saveBytesToOutStream(byte[] b, OutputStream os) throws IOException {
os.write(b); os.write(b);
os.close(); os.close();
@ -195,16 +173,4 @@ 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;
}
} }

Loading…
Cancel
Save