From daa697719eb0ddfd170f1ab94c5422a4f5b93951 Mon Sep 17 00:00:00 2001
From: zhaoxiaohao <279049017@qq.com>
Date: Fri, 18 Sep 2020 14:36:54 +0800
Subject: [PATCH] Merge branch 'dev'

---
 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