public class FileDownLoad {
public static void download(byte[] content, String fileName, boolean isOnLine, HttpServletResponse response) throws Exception {
BufferedInputStream br = new BufferedInputStream(getStringInputStream(content));
byte[] buf = new byte[1024];
int len = 0;
response.reset();
// 在线打开方式
if (isOnLine) {
String fileType = fileName.substring(fileName.indexOf(".") + 1);
response.setContentType("image/" + fileType);
//解决跨域
response.addHeader("Access-Control-Allow-Origin", "*");
response.setCharacterEncoding("UTF-8");
// 文件名应该编码成UTF-8
response.setHeader("Content-Disposition", "inline; filename=" + URLEncoder.encode(fileName, "UTF-8"));
} else {
// 纯下载方式
response.setContentType("application/octet-stream");
//解决跨域
response.addHeader("Access-Control-Allow-Origin", "*");
response.setCharacterEncoding("UTF-8");
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(fileName, "UTF-8"));
}
OutputStream out = response.getOutputStream();
while ((len = br.read(buf)) > 0) {
out.write(buf, 0, len);
}
br.close();
out.close();
}
public static InputStream getStringInputStream(byte[] content) {
if (content != null && content.length > 0) {
try {
ByteArrayInputStream stringInputStream = new ByteArrayInputStream(
content);
return stringInputStream;
} catch (Exception e) {
e.printStackTrace();
}
}
return null;
}
}
返回会话:
@Override
public void download(String fileId,boolean isPreview, HttpServletRequest request, HttpServletResponse response) throws IOException {
EdomsDesignContentEntity resultByContentId = null;
if (storageType == Constant.INT_ONE) {
resultByContentId = getEdomsDesignContentEntityByContentId(fileId);
if (Objects.isNull(resultByContentId)) {
log.error(SystemError.SYS_10016.getMessageWithCode());
response.sendError(404, "File not found!");
}
} else {
//从文件服务器获取内容
}
if (Objects.nonNull(resultByContentId)) {
byte[] file = resultByContentId.getContent();
String fileName = resultByContentId.getFileName();
try {
FileDownLoad.download(file, fileName, isPreview, response);
} catch (Exception e) {
log.error("文件");
}
} else {
try {
response.sendError(404, "File not found!");
return;
} catch (Exception e) {
}
}
}