Administrator
/
FenXiNspJavaMain
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

下载方法改造

This commit is contained in:
zy 2025-11-13 19:01:12 +08:00
parent 537c9d9d1b
commit 79af94e648
2 changed files with 410 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

326
jeecg-boot-base-core/src/main/java/org/jeecg/common/util/ResponseOutputFile.java Normal file
View File

@ -0,0 +1,326 @@
package org.jeecg.common.util;
import cn.hutool.core.util.StrUtil;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.ss.usermodel.Workbook;
import org.jeecg.common.exception.JeecgBootException;
import org.springframework.http.MediaType;
import org.springframework.http.MediaTypeFactory;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
/**
* @Author wgx
*/
@Slf4j
public class ResponseOutputFile {
public static void output(HttpServletRequest request, HttpServletResponse response, Workbook sheets,
String fileName) {
ServletOutputStream outputStream = null;
try {
//设置下载模式
String contentType = getContentType(fileName);
response.setContentType(contentType);
//设置下载模式
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8") + "");
outputStream = response.getOutputStream();
sheets.write(outputStream);
} catch (IOException e) {
e.printStackTrace();
} finally {
if(outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
/**
* 导出文件
* @param request
* @param response
* @param in
* @param fileSize
* @param fileName
*/
public static void output(HttpServletRequest request, HttpServletResponse response, InputStream in,
long fileSize, String fileName) {
ServletOutputStream outputStream = null;
BufferedInputStream bufIn = null;
try {
//设置下载模式
String contentType = getContentType(fileName);
response.setContentLengthLong(fileSize);
response.setContentType(contentType);
// 设置强制下载不打开
// response.setContentType("application/force-download");
// 在线打开方式 文件名应该编码成UTF-8
// response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(filename, "UTF-8"));
//设置下载模式
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8") + "");
// 读到流中
bufIn = new BufferedInputStream(in);
byte[] buffer = new byte[2048];
int num;
outputStream = response.getOutputStream();
while ((num = bufIn.read(buffer)) != -1) {
outputStream.write(buffer, 0, num);
}
response.flushBuffer();
} catch (IOException e) {
e.printStackTrace();
}finally {
if(bufIn != null){
try {
bufIn.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if(in != null){
try {
in.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if(outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
//断点续传
public static void output(HttpServletRequest request, HttpServletResponse response, File file, boolean del){
if(file == null || !file.exists()){
throw new JeecgBootException("文件错误");
}
InputStream inputStream = null;
RandomAccessFile in = null;
ServletOutputStream outputStream = null;
try {
//设置下载模式
String fileName = file.getName();
// String type = fileName.substring(fileName.indexOf(".") + 1);
String contentType = getContentType(fileName);
String range = request.getHeader("Range");
long fileSize = file.length();
long downloadSize = fileSize;
long fromPos = 0, toPos = fileSize - 1;
if(range == null){
response.setContentLengthLong(downloadSize);
}else{// 断点续传
// https://blog.csdn.net/liang19890820/article/details/53215087#%E6%96%AD%E7%82%B9%E7%BB%AD%E4%BC%A0%E7%9A%84%E7%94%A8%E9%80%94
// https://blog.csdn.net/qq_33868430/article/details/83308315
// 若客户端传来Range，说明之前下载了一部分，设置206状态(SC_PARTIAL_CONTENT)
//Range 头部的格式有以下几种情况：
//Range: bytes=0-499 表示第 0-499 字节范围的内容
//Range: bytes=500-999 表示第 500-999 字节范围的内容
//Range: bytes=-500 表示最后 500 字节的内容
//Range: bytes=500- 表示从第 500 字节开始到文件结束部分的内容
//Range: bytes=0-0,-1 表示第一个和最后一个字节
//Range: bytes=500-600,601-999 同时指定几个范围
response.setStatus(HttpServletResponse.SC_PARTIAL_CONTENT);
RangeDTO rangeDTO = executeRangeInfo(range, fileSize);
if(rangeDTO == null){
throw new JeecgBootException("range error");
}
fromPos = rangeDTO.getStartByte();
toPos = rangeDTO.getEndByte();
// String bytes = range.replaceAll("bytes=", "");
// String[] ary = bytes.split("-");
// fromPos = Long.parseLong(ary[0]);
// if (ary.length == 2) {
// toPos = Long.parseLong(ary[1]);
// }
int size;
if (toPos > fromPos) {
size = (int) (toPos - fromPos + 1);
} else {
size = (int) (fileSize - fromPos);
}
response.setContentLengthLong(size);
//Content-Range 说明
//Content-Range: bytes 0-499/22400
//0－499 是指当前发送的数据的范围，而 22400 则是文件的总大小。
response.setHeader("Content-Range", "bytes " + fromPos + "-" + toPos + "/" + fileSize);
downloadSize = size;
}
response.setContentType(contentType);
// 设置强制下载不打开
// response.setContentType("application/force-download");
// 在线打开方式 文件名应该编码成UTF-8
// response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(filename, "UTF-8"));
//设置下载模式
if(!(contentType.contains("image") || contentType.contains("video"))){
//非图片或视频则下载
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(file.getName(), "UTF-8") + "");
}
response.setHeader("Accept-Ranges", "bytes");
// 读到流中
in = new RandomAccessFile(file, "rw");
// 设置下载起始位置
if (fromPos > 0) {
in.seek(fromPos);
}
byte[] buffer = new byte[2048];
int num;
outputStream = response.getOutputStream();
while ((num = in.read(buffer)) != -1) {
outputStream.write(buffer, 0, num);
}
response.flushBuffer();
// inputStream = new BufferedInputStream(new FileInputStream(file));
// outputStream = response.getOutputStream();
// byte[] buf = new byte[1024];
// int len;
// while ((len = inputStream.read(buf)) > 0) {
// outputStream.write(buf, 0, len);
// }
// response.flushBuffer();
} catch (IOException e) {
e.printStackTrace();
}finally {
if(inputStream != null){
try {
inputStream.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if(in != null){
try {
in.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if(outputStream != null){
try {
outputStream.close();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
if(del){
try {
file.delete();
}catch (Exception e){
log.error(e.getMessage());
}
}
}
}
private static RangeDTO executeRangeInfo(String range, Long fileSize) {
if (StrUtil.isEmpty(range) || !range.contains("bytes=") || !range.contains("-")) {
return null;
}
long startByte = 0;
long endByte = fileSize - 1;
range = range.substring(range.lastIndexOf("=") + 1).trim();
String[] ranges = range.split("-");
if (ranges.length <= 0 || ranges.length > 2) {
return null;
}
try {
if (ranges.length == 1) {
if (range.startsWith("-")) {
//1. 如：bytes=-1024 从开始字节到第1024个字节的数据
endByte = Long.parseLong(ranges[0]);
} else if (range.endsWith("-")) {
//2. 如：bytes=1024- 第1024个字节到最后字节的数据
startByte = Long.parseLong(ranges[0]);
}
} else {
//3. 如：bytes=1024-2048 第1024个字节到2048个字节的数据
startByte = Long.parseLong(ranges[0]);
endByte = Long.parseLong(ranges[1]);
}
} catch (NumberFormatException e) {
startByte = 0;
endByte = fileSize - 1;
}
if (startByte >= fileSize) {
log.error("range error, startByte >= fileSize. " +
"startByte: {}, fileSize: {}", startByte, fileSize);
return null;
}
return new RangeDTO(startByte, endByte);
}
public static String getContentType(String fileName){
return MediaTypeFactory.getMediaType(fileName)
.orElse(MediaType.APPLICATION_OCTET_STREAM).toString();
// //String contentType = "application/force-download"
// if(StrUtil.isBlank(fileName)){
// return contentType;
// }
// MediaTypeFactory.getMediaType(fileName);
// // fileName 文件名
//
// String type = fileName.substring(fileName.indexOf(".") + 1);
// if (Objects.equals(type, "jpg") || Objects.equals(type, "JPG")) {
// contentType = "image/jpeg";
// }else if(Objects.equals(type, "png") || Objects.equals(type, "PNG")) {
// contentType = "image/png";
// }else if(Objects.equals(type, "xlsx") || Objects.equals(type, "xls")){
// contentType = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
// }else if(Objects.equals(type, "txt")){
// contentType = "text/plain; charset=utf-8";
// }else if(Objects.equals(type, "pdf")){
// contentType = "application/pdf";
// }else if(Objects.equals(type, "doc")){
// contentType = "application/msword";
// }else if(Objects.equals(type, "docx")){
// contentType = "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
// }else if(Objects.equals(type, "zip")){
// contentType = "application/zip";
// }
// return contentType;
}
@Data
static class RangeDTO {
private long startByte;
private long endByte;
public RangeDTO(long startByte, long endByte){
this.startByte = startByte;
this.endByte = endByte;
}
}
}

103
jeecg-module-system/jeecg-system-biz/src/main/java/org/jeecg/modules/system/controller/CommonController.java
View File

@ -7,11 +7,8 @@ import org.jeecg.common.api.vo.Result;
import org.jeecg.common.constant.CommonConstant;
import org.jeecg.common.constant.SymbolConstant;
import org.jeecg.common.exception.JeecgBootException;
import org.jeecg.common.util.CommonUtils;
import org.jeecg.common.util.RestUtil;
import org.jeecg.common.util.TokenUtils;
import org.jeecg.common.util.*;
import org.jeecg.common.util.filter.FileTypeFilter;
import org.jeecg.common.util.oConvertUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
@ -205,6 +202,69 @@ public class CommonController {
// return result;
// }
// /**
// * 预览图片&下载文件
// * 请求地址：http://localhost:8080/common/static/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
// *
// * @param request
// * @param response
// */
// @GetMapping(value = "/static/**")
// public void view(HttpServletRequest request, HttpServletResponse response) {
// // ISO-8859-1 ==> UTF-8 进行编码转换
// String imgPath = extractPathFromPattern(request);
// if(oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)){
// return;
// }
// // 其余处理略
// InputStream inputStream = null;
// OutputStream outputStream = null;
// try {
// imgPath = imgPath.replace("..", "").replace("../","");
// if (imgPath.endsWith(SymbolConstant.COMMA)) {
// imgPath = imgPath.substring(0, imgPath.length() - 1);
// }
// String filePath = uploadpath + File.separator + imgPath;
// File file = new File(filePath);
// if(!file.exists()){
// response.setStatus(404);
// throw new RuntimeException("文件["+imgPath+"]不存在..");
// }
// // 设置强制下载不打开
// response.setContentType("application/force-download");
// response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
// inputStream = new BufferedInputStream(new FileInputStream(filePath));
// outputStream = response.getOutputStream();
// byte[] buf = new byte[1024];
// int len;
// while ((len = inputStream.read(buf)) > 0) {
// outputStream.write(buf, 0, len);
// }
// response.flushBuffer();
// } catch (IOException e) {
// log.error("预览文件失败" + e.getMessage());
// response.setStatus(404);
// e.printStackTrace();
// } finally {
// if (inputStream != null) {
// try {
// inputStream.close();
// } catch (IOException e) {
// log.error(e.getMessage(), e);
// }
// }
// if (outputStream != null) {
// try {
// outputStream.close();
// } catch (IOException e) {
// log.error(e.getMessage(), e);
// }
// }
// }
//
// }
/**
* 预览图片&下载文件
* 请求地址：http://localhost:8080/common/static/{user/20190119/e1fe9925bc315c60addea1b98eb1cb1349547719_1547866868179.jpg}
@ -216,35 +276,40 @@ public class CommonController {
public void view(HttpServletRequest request, HttpServletResponse response) {
// ISO-8859-1 ==> UTF-8 进行编码转换
String imgPath = extractPathFromPattern(request);
if(oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)){
if (oConvertUtils.isEmpty(imgPath) || CommonConstant.STRING_NULL.equals(imgPath)) {
return;
}
// 其余处理略
InputStream inputStream = null;
OutputStream outputStream = null;
try {
imgPath = imgPath.replace("..", "").replace("../","");
imgPath = imgPath.replace("..", "").replace("../", "");
if (imgPath.endsWith(SymbolConstant.COMMA)) {
imgPath = imgPath.substring(0, imgPath.length() - 1);
}
String filePath = uploadpath + File.separator + imgPath;
File file = new File(filePath);
if(!file.exists()){
if (!file.exists()) {
response.setStatus(404);
throw new RuntimeException("文件["+imgPath+"]不存在..");
log.info("文件[" + imgPath + "]不存在..");
//throw new RuntimeException("文件["+imgPath+"]不存在..");
return;
}
ResponseOutputFile.output(request, response, file, false);
// 设置强制下载不打开
response.setContentType("application/force-download");
response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
inputStream = new BufferedInputStream(new FileInputStream(filePath));
outputStream = response.getOutputStream();
byte[] buf = new byte[1024];
int len;
while ((len = inputStream.read(buf)) > 0) {
outputStream.write(buf, 0, len);
}
response.flushBuffer();
} catch (IOException e) {
// response.setContentType("application/force-download");
// response.setContentType(ResponseOutputFile.getContentType(file.getName()));
// response.setContentLengthLong(file.length());
// response.addHeader("Content-Disposition", "attachment;fileName=" + new String(file.getName().getBytes("UTF-8"),"iso-8859-1"));
// inputStream = new BufferedInputStream(new FileInputStream(filePath));
// outputStream = response.getOutputStream();
// byte[] buf = new byte[1024];
// int len;
// while ((len = inputStream.read(buf)) > 0) {
// outputStream.write(buf, 0, len);
// }
// response.flushBuffer();
} catch (Exception e) {
log.error("预览文件失败" + e.getMessage());
response.setStatus(404);
e.printStackTrace();