From 001ab4186873c76ac1e6f97f8e02b6980ac8eb2a Mon Sep 17 00:00:00 2001
From: bingbing <zhaobingliang@aliyun.com>
Date: Wed, 28 Oct 2020 19:00:22 +0800
Subject: [PATCH] 回退

---
 kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java |   16 +++++++++++++++-
 1 files changed, 15 insertions(+), 1 deletions(-)

diff --git a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java
index 0e9d05e..3cac468 100644
--- a/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java
+++ b/kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java
@@ -32,7 +32,21 @@
         fileInfo.setId(IdUtil.fastSimpleUUID());
         fileInfo.setName(file.getOriginalFilename());
         fileInfo.setContentType(file.getContentType());
-        fileInfo.setIsImg(fileInfo.getContentType().startsWith("image/"));
+        if (fileInfo.getContentType().startsWith("image/")) {
+            fileInfo.setIsImg(true);
+        }
+        else if (fileInfo.getContentType().startsWith("multipart")){
+            String extendName = file.getOriginalFilename().substring(file.getOriginalFilename().indexOf(".")+1,file.getOriginalFilename().length()).toLowerCase();
+            if (extendName.equals("png") || extendName.equals("bmp") || extendName.equals("jpg") || extendName.equals("jpeg") || extendName.equals("gif")) {
+                fileInfo.setIsImg(true);
+            } else  {
+                fileInfo.setIsImg(false);
+            }
+
+        } else  {
+            fileInfo.setIsImg(false);
+        }
+
         fileInfo.setSize(file.getSize());
         fileInfo.setCreateTime(new Date());
         return fileInfo;

--
Gitblit v1.8.0