From a3fb0dd528063447b1c96f69cdec5b63ec85e895 Mon Sep 17 00:00:00 2001
From: MAC <1162714483@qq.com>
Date: Sun, 2 Jan 2022 18:25:26 +0800
Subject: [PATCH] =?UTF-8?q?fix(=E6=96=87=E6=A1=A3=E9=A2=84=E8=A7=88):=20?=
 =?UTF-8?q?=E4=BF=AE=E6=94=B9=E9=A2=84=E8=A7=88=E6=8A=A5=E9=94=99=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../file/controller/FiletransferController.java     | 13 ++++++++-----
 .../com/qiwenshare/file/dto/file/PreviewDTO.java    |  1 +
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/qiwenshare/file/controller/FiletransferController.java b/src/main/java/com/qiwenshare/file/controller/FiletransferController.java
index c05cba6..31806c7 100644
--- a/src/main/java/com/qiwenshare/file/controller/FiletransferController.java
+++ b/src/main/java/com/qiwenshare/file/controller/FiletransferController.java
@@ -152,12 +152,15 @@ public class FiletransferController {
             filetransferService.previewPictureFile(httpServletResponse, previewDTO);
             return ;
         }
-
-        Cookie[] cookieArr = httpServletRequest.getCookies();
         String token = "";
-        for (Cookie cookie : cookieArr) {
-            if ("token".equals(cookie.getName())) {
-                token = cookie.getValue();
+        if (StringUtils.isNotEmpty(previewDTO.getToken())) {
+            token = previewDTO.getToken();
+        } else {
+            Cookie[] cookieArr = httpServletRequest.getCookies();
+            for (Cookie cookie : cookieArr) {
+                if ("token".equals(cookie.getName())) {
+                    token = cookie.getValue();
+                }
             }
         }
 
diff --git a/src/main/java/com/qiwenshare/file/dto/file/PreviewDTO.java b/src/main/java/com/qiwenshare/file/dto/file/PreviewDTO.java
index 64c909d..34a50aa 100644
--- a/src/main/java/com/qiwenshare/file/dto/file/PreviewDTO.java
+++ b/src/main/java/com/qiwenshare/file/dto/file/PreviewDTO.java
@@ -14,4 +14,5 @@ public class PreviewDTO {
     private String isMin;
     private Integer platform;
     private String url;
+    private String token;
 }