broken images refuse to upload

ktali · Nov 18, 2018 · 3a10497 · 3a10497
1 parent 76ccfd8
commit 3a10497
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/com/kmk/imageboard/service/ImageService.java b/src/main/java/com/kmk/imageboard/service/ImageService.java
@@ -11,6 +11,7 @@
 
 import org.springframework.web.multipart.MultipartFile;
 
+import javax.imageio.IIOException;
 import javax.imageio.ImageIO;
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
@@ -53,7 +54,12 @@ public ResponseEntity<String> uploadImage(Principal principal, MultipartFile fil
                 Scalr.Mode.AUTOMATIC,
                 128,
                 128);
-        ImageIO.write(thumbnailBufferedImage, "jpg", new File("imagerepository" + File.separator + "thumbnails" + File.separator + newEntity.getId()));
+        try {
+            ImageIO.write(thumbnailBufferedImage, "jpg", new File("imagerepository" + File.separator + "thumbnails" + File.separator + newEntity.getId()));
+        } catch (IIOException e) {
+            imageRepository.delete(newEntity);
+            throw e;
+        }
         bufferedImage.flush();
         thumbnailBufferedImage.flush();