From f67a8bfa2bd4895806bb1686e25df3a2eab8e771 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期五, 11 八月 2023 17:26:47 +0800
Subject: [PATCH] 优化代码

---
 service-api/src/main/java/com/java110/api/controller/app/file/UserDownloadFileController.java |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/service-api/src/main/java/com/java110/api/controller/app/file/UserDownloadFileController.java b/service-api/src/main/java/com/java110/api/controller/app/file/UserDownloadFileController.java
index 5fca68a..8286715 100644
--- a/service-api/src/main/java/com/java110/api/controller/app/file/UserDownloadFileController.java
+++ b/service-api/src/main/java/com/java110/api/controller/app/file/UserDownloadFileController.java
@@ -1,9 +1,8 @@
 package com.java110.api.controller.app.file;
 
-import com.java110.api.controller.app.payment.NotifyPaymentController;
 import com.java110.core.client.FileUploadTemplate;
 import com.java110.core.log.LoggerFactory;
-import com.java110.dto.userDownloadFile.UserDownloadFileDto;
+import com.java110.dto.user.UserDownloadFileDto;
 import com.java110.intf.job.IUserDownloadFileV1InnerServiceSMO;
 import com.java110.utils.util.Assert;
 import org.slf4j.Logger;
@@ -47,8 +46,11 @@
         userDownloadFileDto.setDownloadUserId(userId);
         List<UserDownloadFileDto> userDownloadFileDtos = userDownloadFileV1InnerServiceSMOImpl.queryUserDownloadFiles(userDownloadFileDto);
         Assert.listOnlyOne(userDownloadFileDtos, "鏂囦欢涓嶅瓨鍦�");
+        String tempUrl = userDownloadFileDtos.get(0).getTempUrl();
+        String fileName = userDownloadFileDtos.get(0).getFileTypeName()+tempUrl.substring(tempUrl.lastIndexOf("/"));
 
         response.setHeader("content-type", "application/octet-stream");
+        response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
         response.setContentType("application/octet-stream");
 
         InputStream is = null;

--
Gitblit v1.8.0