From 9b7341aa3d45b985e1e2fef75964b1c1bcde1dab Mon Sep 17 00:00:00 2001
From: huangzj <826100833@qq.com>
Date: Thu, 31 Aug 2023 14:16:08 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=BF=94=E5=9B=9E=E5=89=8D?=
 =?UTF-8?q?=E7=AB=AF=E5=9B=BE=E7=89=87=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/ImageToStreamController.java   | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/pqs-system/system-boot/src/main/java/com/njcn/system/controller/ImageToStreamController.java b/pqs-system/system-boot/src/main/java/com/njcn/system/controller/ImageToStreamController.java
index f58d095f7..5c297eec2 100644
--- a/pqs-system/system-boot/src/main/java/com/njcn/system/controller/ImageToStreamController.java
+++ b/pqs-system/system-boot/src/main/java/com/njcn/system/controller/ImageToStreamController.java
@@ -53,9 +53,10 @@ public class ImageToStreamController extends BaseController {
         try {
             //获取部分参数，用于从minIO上将文件下载到tmp（此步骤不是必须）
             InputStream fileStream = fileStorageUtil.getFileStream(bgImage);
-
+            //获取文件格式
+            String format = bgImage.substring(bgImage.lastIndexOf('.') + 1);
             response.setContentType("application/octet-stream");
-            response.setHeader("content-type", "image/png");
+            response.setHeader("content-type", "image/"+getFileFormat(format));
             response.setHeader("Content-Disposition", "inline;fileName=download" +System.currentTimeMillis() + "");// 设置文件名
             //返回流处理
             bis = new BufferedInputStream(fileStream);
@@ -82,4 +83,21 @@ public class ImageToStreamController extends BaseController {
             }
         }
     }
+
+    private String getFileFormat(String format){
+        switch (format) {
+            case "png":
+                return "png";
+            case "svg":
+                return "svg+xml";
+            case "jpg":
+                return "jpeg";
+            case "gif":
+                return "gif";
+
+            default:
+                break;
+        }
+        return format;
+    }
 }