🐛 image type judge

hello-world-1 · Aug 14, 2018 · 4dff10c · 4dff10c
1 parent a1f85e4
commit 4dff10c
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/src/main/java/com/tale/controller/admin/IndexController.java b/src/main/java/com/tale/controller/admin/IndexController.java
@@ -24,6 +24,7 @@
 import lombok.extern.slf4j.Slf4j;
 
 import javax.swing.*;
+import java.io.File;
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Paths;
@@ -93,14 +94,18 @@ public RestResponse<?> upload(Request request) {
             if ((fileItem.getLength() / 1024) <= TaleConst.MAX_FILE_SIZE) {
                 String fkey = TaleUtils.getFileKey(fname);
 
-                String ftype             = fileItem.getContentType().contains("image") ? Types.IMAGE : Types.FILE;
-                String filePath          = TaleUtils.UP_DIR + fkey;
-                String newFileName       = TaleUtils.getFileName(fkey);
-                String thumbnailFilePath = TaleUtils.UP_DIR + fkey.replace(newFileName, "thumbnail_" + newFileName);
+                String ftype    = fileItem.getContentType().contains("image") ? Types.IMAGE : Types.FILE;
+                String filePath = TaleUtils.UP_DIR + fkey;
+
 
                 try {
                     Files.write(Paths.get(filePath), fileItem.getData());
-                    ImageUtils.cutCenterImage(CLASSPATH + fkey, thumbnailFilePath, 270, 380);
+                    if (TaleUtils.isImage(new File(filePath))) {
+                        String newFileName       = TaleUtils.getFileName(fkey);
+                        String thumbnailFilePath = TaleUtils.UP_DIR + fkey.replace(newFileName, "thumbnail_" + newFileName);
+                        ImageUtils.cutCenterImage(CLASSPATH + fkey, thumbnailFilePath, 270, 380);
+
+                    }
                 } catch (IOException e) {
                     log.error("", e);
                 }