renun/zfile
1
0
Fork 0
mirror of https://github.com/zfile-dev/zfile.git synced 2025-04-19 05:34:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ 增加本地文件下载/预览行为控制

Browse Source
This commit is contained in:
赵俊
2021-07-13 21:43:25 +08:00
parent 73b42cf654
commit 463f311dd3
3 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/im/zhaojun/zfile/controller/home/LocalController.java
View File

@@ -32,17 +32,21 @@ public class LocalController {
*
* @param driveId
* 驱动器 ID
* @param type
* 附件预览类型:
* download:下载
* default: 浏览器默认行为
*/
@GetMapping("/file/{driveId}/**")
@ResponseBody
public void downAttachment(@PathVariable("driveId") Integer driveId, final HttpServletRequest request, final HttpServletResponse response) {
public void downAttachment(@PathVariable("driveId") Integer driveId, String type, final HttpServletRequest request, final HttpServletResponse response) {
String path = (String) request.getAttribute(HandlerMapping.PATH_WITHIN_HANDLER_MAPPING_ATTRIBUTE);
String bestMatchPattern = (String) request.getAttribute(HandlerMapping.BEST_MATCHING_PATTERN_ATTRIBUTE);
AntPathMatcher apm = new AntPathMatcher();
String filePath = apm.extractPathWithinPattern(bestMatchPattern, path);
LocalServiceImpl localService = (LocalServiceImpl) driveContext.get(driveId);
File file = new File(StringUtils.removeDuplicateSeparator(localService.getFilePath() + ZFileConstant.PATH_SEPARATOR + filePath));
FileUtil.export(request, response, file);
FileUtil.export(request, response, file, type);
}
}

1
src/main/java/im/zhaojun/zfile/model/constant/LocalFileResponseTypeConstant.java Normal file
View File

@@ -0,0 +1 @@
package im.zhaojun.zfile.model.constant;

17
src/main/java/im/zhaojun/zfile/util/FileUtil.java
View File

@@ -1,6 +1,8 @@
package im.zhaojun.zfile.util;
import cn.hutool.core.util.StrUtil;
import cn.hutool.core.util.URLUtil;
import im.zhaojun.zfile.model.constant.LocalFileResponseTypeConstant;
import lombok.extern.slf4j.Slf4j;
import org.apache.catalina.connector.ClientAbortException;
import org.springframework.core.io.FileSystemResource;
@@ -16,6 +18,7 @@ import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.util.Date;
import java.util.Objects;
/**
* @author zhaojun
@@ -63,18 +66,18 @@ public class FileUtil {
* @param response 响应对象
* @param file 下载的文件
*/
public static void export(HttpServletRequest request, HttpServletResponse response, File file) {
export(request, response, file, file.getName());
public static void export(HttpServletRequest request, HttpServletResponse response, File file, String type) {
export(request, response, file, file.getName(), type);
}
/**
* 返回文件给 response，支持断点续传和多线程下载
* 返回文件给 response，支持断点续传和多线程下载 (动态变化的文件不支持)
* @param request 请求对象
* @param response 响应对象
* @param file 下载的文件
* @param fileName 下载的文件名，为空则默认读取文件名称
*/
public static void export(HttpServletRequest request, HttpServletResponse response, File file, String fileName) {
public static void export(HttpServletRequest request, HttpServletResponse response, File file, String fileName, String type) {
if (!file.exists()) {
try {
response.getWriter().write("404 FILE NOT FOUND");
@@ -135,11 +138,15 @@ public class FileUtil {
//文件类型
String contentType = request.getServletContext().getMimeType(fileName);
if (Objects.equals(type, LocalFileResponseTypeConstant.DOWNLOAD) || StrUtil.isEmpty(contentType)) {
contentType = "attachment";
}
response.setHeader(HttpHeaders.ACCEPT_RANGES, "bytes");
response.setHeader(HttpHeaders.CONTENT_TYPE, contentType);
// 这里文件名换你想要的，inline 表示浏览器可以直接使用
// 参考资料：https://developer.mozilla.org/zh-CN/docs/Web/HTTP/Headers/Content-Disposition
response.setHeader(HttpHeaders.CONTENT_DISPOSITION, "inline;filename=" + URLUtil.encode(fileName));
response.setHeader(HttpHeaders.CONTENT_DISPOSITION, contentType + ";filename=" + URLUtil.encode(fileName));
response.setHeader(HttpHeaders.CONTENT_LENGTH, String.valueOf(contentLength));
// [要下载的开始位置]-[结束位置]/[文件总大小]
response.setHeader(HttpHeaders.CONTENT_RANGE, "bytes " + startByte + rangeSeparator + endByte + "/" + file.length());