下载和预览前判断文件是否能到达

dd2c5cdc · chentianzhong · 3fa5bf89 · dd2c5cdc · dd2c5cdc
Commit dd2c5cdc authored Jul 09, 2023 by chentianzhong
Hide whitespace changes
Inline Side-by-side

Showing with 38 additions and 7 deletions

src/main/java/com/zq/email/controller/EmailAttachController.java
+28 -7

src/main/java/com/zq/email/service/impl/EmailAttachServiceImpl.java
+10 -0

No files found.
--- a/src/main/java/com/zq/email/controller/EmailAttachController.java
+++ b/src/main/java/com/zq/email/controller/EmailAttachController.java
@@ -28,6 +28,7 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.core.io.Resource;
 import org.springframework.http.ResponseEntity;
+import org.springframework.util.Assert;
 import org.springframework.web.bind.annotation.*;
 import javax.servlet.http.HttpServletRequest;
@@ -69,6 +70,26 @@ public class EmailAttachController {
        return ResultVo.success(emailAttachService.findByContentId(vo.getContentId()));
    }
+    @ApiOperation("前端调用下载前，判断文件是否是否可以下载")
+    @AnonymousGetMapping("/preDownload/{id}")
+    public ResultVo preDownload(HttpServletRequest request, HttpServletResponse response, @PathVariable("id") Long id) {
+        EmailAttach emailAttach = emailAttachService.getById(id);
+        AssertUtils.notNull(emailAttach, "获取附件信息失败");
+        String download = emailAttachService.getDownloadPath(emailAttach);
+        AssertUtils.hasText(download, "下载地址为空");
+        log.info("下载文件地址:" +download);
+        try {
+            byte[] bytes = HttpUtil.downloadBytes(download);
+            AssertUtils.isTrue(ArrayUtil.isNotEmpty(bytes), "下载文件失败，文件为空");
+            return ResultVo.success();
+        } catch (Exception e) {
+            log.error("下载文件失败:" + e.getLocalizedMessage(), e);
+            return ResultVo.fail("下载文件失败,文件可能不存在");
+        }
+    }
    @ApiOperation("pc统一下载附件和移动端旧附件下载")
    @AnonymousGetMapping("/download/{id}")
    public void download(HttpServletRequest request, HttpServletResponse response, @PathVariable("id") Long id) {
@@ -78,16 +99,18 @@ public class EmailAttachController {
        String download = emailAttachService.getDownloadPath(emailAttach);
        AssertUtils.hasText(download, "下载地址为空");
        log.info("下载文件地址:" +download);
-        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
-            HttpUtil.download(download, outputStream, true);
+            byte[] bytes = HttpUtil.downloadBytes(download);
-            byte[] bytes = outputStream.toByteArray();
            response.setHeader("Content-Disposition", "attachment;filename*=UTF-8''" + URLEncoder.encode(emailAttach.getAttachName(),"UTF-8"));
            response.setCharacterEncoding("UTF-8");
            ServletUtil.write(response, IoUtil.toStream(bytes), "application/octet-stream");
        } catch (Exception e) {
            log.error("下载文件失败:" + e.getLocalizedMessage(), e);
-            AssertUtils.isTrue(false, "下载文件失败,文件可能不存在");
+            if (profile.contains("wan")){
+                AssertUtils.isTrue(false, "文件超过安全时间，禁止从外网访问下载");
+            }else{
+                AssertUtils.isTrue(false, "下载文件失败，文件可能不存在");
+            }
        }
    }
@@ -101,10 +124,8 @@ public class EmailAttachController {
        String download = emailAttachService.getDownloadPath(emailAttach);
        AssertUtils.hasText(download, "下载地址为空");
        log.info("下载文件地址:" +download);
-        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        try {
-            HttpUtil.download(download, outputStream, true);
+            byte[] bytes = HttpUtil.downloadBytes(download);
-            byte[] bytes = outputStream.toByteArray();
            ServletUtil.write(response, IoUtil.toStream(bytes), "application/octet-stream");
        } catch (Exception e) {
            log.error("下载文件失败:" + e.getLocalizedMessage(), e);

--- a/src/main/java/com/zq/email/service/impl/EmailAttachServiceImpl.java
+++ b/src/main/java/com/zq/email/service/impl/EmailAttachServiceImpl.java
@@ -2,8 +2,10 @@ package com.zq.email.service.impl;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.ArrayUtil;
 import cn.hutool.core.util.IdUtil;
 import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpUtil;
 import cn.hutool.json.JSONUtil;
 import com.zq.common.utils.AssertUtils;
 import com.zq.common.utils.TokenUtils;
@@ -167,6 +169,14 @@ public class EmailAttachServiceImpl extends ServiceImpl<EmailAttachMapper, Email
        String downloadPath = this.getDownloadPath(emailAttach);
        String suffix = emailAttach.getAttachName().substring(emailAttach.getAttachName().lastIndexOf(".") + 1);
        AssertUtils.hasText(suffix, "错误的文件名称("+emailAttach.getAttachName()+"),导致获取文件类型失败");
+        log.info("wps预览文件地址:  " + downloadPath);
+        try {
+            byte[] bytes = HttpUtil.downloadBytes(downloadPath);
+            AssertUtils.isTrue(ArrayUtil.isNotEmpty(bytes), "下载文件失败，文件为空");
+        }catch (Exception e){
+            AssertUtils.isTrue(false, "下载文件失败，文件可能不存在");
+        }
        FileViewVo viewVo = new FileViewVo();
        viewVo.setFileId(IdUtil.fastSimpleUUID());