diff --git a/hyp-admin/src/main/java/com/hyp/web/controller/reward/FileRelationController.java b/hyp-admin/src/main/java/com/hyp/web/controller/reward/FileRelationController.java index fe68dca..393e4a9 100644 --- a/hyp-admin/src/main/java/com/hyp/web/controller/reward/FileRelationController.java +++ b/hyp-admin/src/main/java/com/hyp/web/controller/reward/FileRelationController.java @@ -2,10 +2,12 @@ package com.hyp.web.controller.reward; import com.hyp.common.annotation.Log; import com.hyp.common.config.HypConfig; +import com.hyp.common.constant.Constants; import com.hyp.common.core.controller.BaseController; import com.hyp.common.core.domain.AjaxResult; import com.hyp.common.enums.BusinessType; import com.hyp.common.exception.ServiceException; +import com.hyp.common.utils.HypFileUtil; import com.hyp.common.utils.file.FileUploadUtils; import com.hyp.common.utils.file.FileUtils; import com.hyp.framework.config.ServerConfig; @@ -179,19 +181,26 @@ public class FileRelationController extends BaseController { if(Objects.isNull(rewFileRelation)){ throw new ServiceException("下载文件不存在"); } - String fileName = rewFileRelation.getFileName(); + String fileName = rewFileRelation.getFilePath(); if (!FileUtils.checkAllowDownload(fileName)) { - throw new Exception(com.hyp.common.utils.StringUtils.format("文件名称({})非法，不允许下载。 ", fileName)); + throw new Exception(com.hyp.common.utils.StringUtils.format("资源文件({})非法，不允许下载。 ", fileName)); } - String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1); - String filePath = HypConfig.getDownloadPath() + fileName; + // 本地资源路径 + String localPath = HypConfig.getProfile(); + // 数据库资源地址 + String downloadPath = localPath + com.hyp.common.utils.StringUtils.substringAfter(fileName, Constants.RESOURCE_PREFIX); + // 下载名称 + String downloadName = com.hyp.common.utils.StringUtils.substringAfterLast(downloadPath, "/"); response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE); - FileUtils.setAttachmentResponseHeader(response, realFileName); - FileUtils.writeBytes(filePath, response.getOutputStream()); + FileUtils.setAttachmentResponseHeader(response, downloadName); + FileUtils.writeBytes(downloadPath, response.getOutputStream()); } catch (Exception e) { log.error("下载文件失败", e); } } + + + } diff --git a/hyp-common/src/main/java/com/hyp/common/utils/HypFileUtil.java b/hyp-common/src/main/java/com/hyp/common/utils/HypFileUtil.java new file mode 100644 index 0000000..169fd36 --- /dev/null +++ b/hyp-common/src/main/java/com/hyp/common/utils/HypFileUtil.java @@ -0,0 +1,131 @@ +package com.hyp.common.utils; + +import cn.hutool.core.date.DateUtil; +import cn.hutool.core.io.FileUtil; +import cn.hutool.core.util.IdUtil; +import cn.hutool.core.util.RandomUtil; +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.multipart.MultipartFile; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.*; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; + +@Slf4j +public class HypFileUtil extends FileUtil { + + public static final String IE_FLAG_MISE = "msie"; + public static final String IE_FLAG_TRIDENT = "trident"; + public static final String IE_FLAG_EDGE = "edge"; + + /** + * MultipartFile转File + */ + public static File toFile(MultipartFile multipartFile) { + // 获取文件名 + String fileName = multipartFile.getOriginalFilename(); + // 获取文件后缀 + String prefix = "." + getExtensionName(fileName); + File file = null; + try { + // 用uuid作为文件名，防止生成的临时文件重复 + file = File.createTempFile(IdUtil.simpleUUID(), prefix); + // MultipartFile to File + multipartFile.transferTo(file); + } catch (IOException e) { + e.printStackTrace(); + } + return file; + } + + /** + * 获取文件扩展名，不带. + */ + public static String getExtensionName(String filename) { + if ((filename != null) && (filename.length() > 0)) { + int dot = filename.lastIndexOf('.'); + if ((dot > -1) && (dot < (filename.length() - 1))) { + return filename.substring(dot + 1); + } + } + return filename; + } + + + + /** + * 获取不带扩展名的文件名 + */ + public static String getFileNameNoEx(String filename) { + if ((filename != null) && (filename.length() > 0)) { + int dot = filename.lastIndexOf('.'); + if ((dot > -1) && (dot < (filename.length()))) { + return filename.substring(0, dot); + } + } + return filename; + } + + + public static File upload(MultipartFile file, String uploadDir, String applyId){ + String suffix = getExtensionName(file.getOriginalFilename()); + String fileId = RandomUtil.randomNumbers(32); + String dateDir = DateUtil.format(DateUtil.date(),"yyyyMMdd"); + //上传文件以后的存储路径 + String separator = File.separator; + String filePath = uploadDir + separator + dateDir + separator + applyId + separator + fileId + "." + suffix; + try { + File upFile = new File(filePath).getCanonicalFile(); + // 检测是否存在目录 + if (!upFile.getParentFile().exists()) { + upFile.getParentFile().mkdirs(); + } + log.info("上传文件路径：{}", filePath); + // 文件写入 + file.transferTo(upFile); + return upFile; + } catch (IOException e) { + throw new RuntimeException("上传失败"); + } + } + + + public static void downLoadFile(String filePath, String fileName, HttpServletRequest request, HttpServletResponse response) { + + File f = null; + try { + f = FileUtil.newFile(filePath); + if (!f.exists()) { + response.sendError(404, "File not found!"); + return; + } + } catch (IOException e) { + log.error(e.getMessage(), e); + } + try (BufferedInputStream br = new BufferedInputStream(new FileInputStream(f)); + OutputStream out = response.getOutputStream();) { + byte[] buf = new byte[1024]; + int len = 0; + // 在线打开方式 + String userAgent = request.getHeader("User-Agent").toLowerCase(); + fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name()); + response.reset(); + // 解决申报系统异步下载跨域问题 + response.setHeader("Access-Control-Allow-Origin", "*"); + response.setHeader("Access-Control-Allow-Methods", "POST, GET"); + response.setHeader("Access-Control-Expose-Headers", "Content-Disposition"); + // 纯下载方式 + response.setHeader("Content-disposition", String.format("attachment; filename=\"%s\"", fileName)); + response.setContentType("application/x-msdownload;charset=UTF-8"); + response.setCharacterEncoding(StandardCharsets.UTF_8.name()); + while ((len = br.read(buf)) > 0) { + out.write(buf, 0, len); + } + } catch (IOException e) { + log.error(e.getMessage(), e); + } + } + +}