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; + } }