From 70b737fb611fbcb4f76c97a8877ca1691da92193 Mon Sep 17 00:00:00 2001
From: sws <15810472099@163.com>
Date: 星期三, 23 十一月 2022 21:11:21 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/show/ExportController.java |   18 ++++++++----------
 1 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/show/ExportController.java b/src/main/java/com/lf/server/controller/show/ExportController.java
index abd0c49..f1c4d80 100644
--- a/src/main/java/com/lf/server/controller/show/ExportController.java
+++ b/src/main/java/com/lf/server/controller/show/ExportController.java
@@ -112,14 +112,16 @@
     @RequestMapping(value = "/downloadFile", method = RequestMethod.GET)
     public void downloadFile(String guid, HttpServletRequest req, HttpServletResponse res) {
         try {
-            UserEntity ue = tokenService.getCurrentUser(req);
-            if (ue == null) {
-                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�"));
+            if (StringHelper.isEmpty(guid)) {
+                WebHelper.writeInfo(HttpStatus.UNAUTHORIZED, "鎵句笉鍒版枃浠禝D", res);
+                return;
             }
 
+            UserEntity ue = tokenService.getCurrentUser(req);
             DownloadEntity de = downloadService.selectByGuid(guid);
-            if (de == null) {
-                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�"));
+            if (null == de) {
+                WebHelper.writeInfo(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�", res);
+                return;
             }
 
             de.setDcount(de.getDcount() + 1);
@@ -129,11 +131,7 @@
             String filePath = exportService.getDownloadFilePath(de);
             WebHelper.download(filePath, de.getName(), res);
         } catch (Exception ex) {
-            try {
-                WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage()));
-            } catch (Exception e) {
-                log.error(e.getMessage(), e);
-            }
+            WebHelper.writeInfo(HttpStatus.ERROR, ex.getMessage(), res);
         }
     }
 }

--
Gitblit v1.9.3