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