FileUtils.downFile优化,支持手机播放视频。感谢Nice味道

This commit is contained in:
thinkgem
2019-06-12 09:37:18 +08:00
parent 6bfe50906e
commit 605c38c74d

View File

@@ -3,7 +3,6 @@
*/ */
package com.jeesite.common.io; package com.jeesite.common.io;
import java.io.BufferedOutputStream;
import java.io.File; import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.io.FileNotFoundException; import java.io.FileNotFoundException;
@@ -16,6 +15,7 @@ import java.util.Enumeration;
import java.util.List; import java.util.List;
import javax.activation.MimetypesFileTypeMap; import javax.activation.MimetypesFileTypeMap;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpServletResponse;
@@ -697,136 +697,79 @@ public class FileUtils extends org.apache.commons.io.FileUtils {
* @return 返回错误信息无错误信息返回null * @return 返回错误信息无错误信息返回null
*/ */
public static String downFile(File file, HttpServletRequest request, HttpServletResponse response, String fileName){ public static String downFile(File file, HttpServletRequest request, HttpServletResponse response, String fileName){
long fileLength = file.length(); // 记录文件大小 if (file == null || !file.exists() || file.length() <= 0){
long pastLength = 0; // 记录已下载文件大小 return "文件为空或不存在!";
int rangeSwitch = 0; // 0从头开始的全文下载1从某字节开始的下载bytes=27000-2从某字节开始到某字节结束的下载bytes=27000-39000
long toLength = 0; // 记录客户端需要下载的字节段的最后一个字节偏移量比如bytes=27000-39000则这个值是为39000
long contentLength = 0; // 客户端请求的字节总量
String rangeBytes = ""; // 记录客户端传来的形如“bytes=27000-”或者“bytes=27000-39000”的内容
RandomAccessFile raf = null; // 负责读取数据
OutputStream os = null; // 写出数据
OutputStream out = null; // 缓冲
byte b[] = new byte[1024]; // 暂存容器
if (request.getHeader("Range") != null) { // 客户端请求的下载的文件块的开始字节
response.setStatus(javax.servlet.http.HttpServletResponse.SC_PARTIAL_CONTENT);
logger.debug("request.getHeader(\"Range\") = " + request.getHeader("Range"));
rangeBytes = request.getHeader("Range").replaceAll("bytes=", "");
if (rangeBytes.indexOf('-') == rangeBytes.length() - 1) {// bytes=969998336-
rangeSwitch = 1;
rangeBytes = rangeBytes.substring(0, rangeBytes.indexOf('-'));
pastLength = Long.parseLong(rangeBytes.trim());
contentLength = fileLength - pastLength; // 客户端请求的是 969998336 之后的字节
} else { // bytes=1275856879-1275877358
rangeSwitch = 2;
String temp0 = rangeBytes.substring(0, rangeBytes.indexOf('-'));
String temp2 = rangeBytes.substring(rangeBytes.indexOf('-') + 1, rangeBytes.length());
pastLength = Long.parseLong(temp0.trim()); // bytes=1275856879-1275877358从第 1275856879 个字节开始下载
toLength = Long.parseLong(temp2); // bytes=1275856879-1275877358到第 1275877358 个字节结束
contentLength = toLength - pastLength; // 客户端请求的是 1275856879-1275877358 之间的字节
}
} else { // 从开始进行下载
contentLength = fileLength; // 客户端要求全文下载
} }
try(RandomAccessFile randomFile = new RandomAccessFile(file, "r");
// 如果设设置了Content-Length则客户端会自动进行多线程下载。如果不希望支持多线程则不要设置这个参数。 响应的格式是: ServletOutputStream out = response.getOutputStream()) {
// Content-Length: [文件的总大小] - [客户端请求的下载的文件块的开始字节] long contentLength = randomFile.length();
// ServletActionContext.getResponse().setHeader("Content- Length", new Long(file.length() - p).toString()); String range = request.getHeader("Range");
response.reset(); // 告诉客户端允许断点续传多线程连接下载,响应的格式是:Accept-Ranges: bytes long start = 0, end = 0;
if (pastLength != 0) { if (range != null && range.startsWith("bytes=")) {
response.setHeader("Accept-Ranges", "bytes");// 如果是第一次下,还没有断点续传,状态是默认的 200,无需显式设置;响应的格式是:HTTP/1.1 200 OK String[] values = range.split("=")[1].split("-");
// 不是从最开始下载, 响应的格式是: Content-Range: bytes [文件块的开始字节]-[文件的总大小 - 1]/[文件的总大小] start = Long.parseLong(values[0]);
logger.debug("服务器即将开始断点续传..."); if (values.length > 1) {
switch (rangeSwitch) { end = Long.parseLong(values[1]);
case 1: { // 针对 bytes=27000- 的请求
String contentRange = new StringBuffer("bytes ").append(new Long(pastLength).toString()).append("-")
.append(new Long(fileLength - 1).toString()).append("/").append(new Long(fileLength).toString()).toString();
response.setHeader("Content-Range", contentRange);
break;
}
case 2: { // 针对 bytes=27000-39000 的请求
String contentRange = rangeBytes + "/" + new Long(fileLength).toString();
response.setHeader("Content-Range", contentRange);
break;
}
default: {
break;
} }
} }
} int requestSize = 0;
if (end != 0 && end > start) {
try { requestSize = Long.valueOf(end - start + 1).intValue();
} else {
requestSize = Integer.MAX_VALUE;
}
response.setContentType(FileUtils.getContentType(file.getName()));
boolean isPreview = "preview".equalsIgnoreCase(request.getParameter("source")); boolean isPreview = "preview".equalsIgnoreCase(request.getParameter("source"));
response.addHeader("Content-Disposition", (!isPreview ? "attachment; " : "") + "filename=\"" + String fileNameEncode = EncodeUtils.encodeUrl(StringUtils.isBlank(fileName) ? file.getName() : fileName);
EncodeUtils.encodeUrl(StringUtils.isBlank(fileName) ? file.getName() : fileName) + "\""); response.addHeader("Content-Disposition", (!isPreview ? "attachment; " : "") + "filename=\"" + fileNameEncode + "\"");
response.setContentType(FileUtils.getContentType(file.getName())); // set the MIME type. response.setHeader("Accept-Ranges", "bytes");
response.addHeader("Content-Length", String.valueOf(contentLength)); // 第一次请求只返回 content length 来让客户端请求多次实际数据
os = response.getOutputStream(); if (range == null) {
out = new BufferedOutputStream(os); response.setHeader("Content-Length", String.valueOf(contentLength));
raf = new RandomAccessFile(file, "r"); } else {
try { // 以后的多次以断点续传的方式来返回视频数据
switch (rangeSwitch) { response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT); // 206
case 0: { // 普通下载,或者从头开始的下载 同1 long requestStart = 0, requestEnd = 0;
String[] ranges = range.split("=");
if (ranges.length > 1) {
String[] rangeDatas = ranges[1].split("-");
requestStart = Long.parseLong(rangeDatas[0]);
if (rangeDatas.length > 1) {
requestEnd = Long.parseLong(rangeDatas[1]);
} }
case 1: { // 针对 bytes=27000- 的请求 }
raf.seek(pastLength); // 形如 bytes=969998336- 的客户端请求,跳过 969998336 个字节 long length = 0;
int n = 0; if (requestEnd > 0) {
while ((n = raf.read(b, 0, 1024)) != -1) { length = requestEnd - requestStart + 1;
out.write(b, 0, n); response.setHeader("Content-Length", String.valueOf(length));
} response.setHeader("Content-Range", "bytes " + requestStart + "-" + requestEnd + "/" + contentLength);
break; } else {
} length = contentLength - requestStart;
case 2: { // 针对 bytes=27000-39000 的请求 response.setHeader("Content-Length", String.valueOf(length));
raf.seek(pastLength); // 形如 bytes=1275856879-1275877358 的客户端请求,找到第 1275856879 个字节 response.setHeader("Content-Range", "bytes " + requestStart + "-" + (contentLength - 1) + "/" + contentLength);
int n = 0; }
long readLength = 0; // 记录已读字节数 }
while (readLength <= contentLength - 1024) {// 大部分字节在这里读取 randomFile.seek(start);
n = raf.read(b, 0, 1024); int needSize = requestSize;
readLength += 1024; while (needSize > 0) {
out.write(b, 0, n); byte[] buffer = new byte[1024];
} int len = randomFile.read(buffer);
if (readLength <= contentLength) { // 余下的不足 1024 个字节在这里读取 if (needSize < buffer.length) {
n = raf.read(b, 0, (int) (contentLength - readLength)); out.write(buffer, 0, needSize);
out.write(b, 0, n); } else {
} out.write(buffer, 0, len);
break; if (len < buffer.length) {
}
default: {
break; break;
} }
} }
out.flush(); needSize -= buffer.length;
logger.debug("下载完成!" + file.getAbsolutePath());
} catch (IOException ie) {
/**
* 在写数据的时候, 对于 ClientAbortException 之类的异常,
* 是因为客户端取消了下载,而服务器端继续向浏览器写入数据时, 抛出这个异常,这个是正常的。
* 尤其是对于迅雷这种吸血的客户端软件, 明明已经有一个线程在读取 bytes=1275856879-1275877358
* 如果短时间内没有读取完毕,迅雷会再启第二个、第三个。。。线程来读取相同的字节段, 直到有一个线程读取完毕,迅雷会 KILL
* 掉其他正在下载同一字节段的线程, 强行中止字节读出,造成服务器抛 ClientAbortException。
* 所以,我们忽略这种异常
*/
logger.debug("提醒向客户端传输时出现IO异常但此异常是允许的有可能客户端取消了下载导致此异常不用关心");
} }
} catch (Exception e) { out.flush();
return null;
} catch (IOException e) {
logger.debug(e.getMessage(), e); logger.debug(e.getMessage(), e);
} finally { return e.getMessage();
if (out != null) {
try {
out.close();
} catch (IOException e) {
// logger.error(e.getMessage(), e);
}
}
if (raf != null) {
try {
raf.close();
} catch (IOException e) {
// logger.error(e.getMessage(), e);
}
}
} }
return null;
} }
/** /**