From fbd2d577daa7a09bc4bf9489bae420e48ea998e0 Mon Sep 17 00:00:00 2001
From: Slavisa Avramovic <slavisaavramovic@yahoo.com>
Date: Tue, 17 May 2016 07:09:31 +0200
Subject: [PATCH] image-download - Fine grained exception handling

---
 .../java/com/baeldung/spring/controller/ImageController.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/spring-mvc-xml/src/main/java/com/baeldung/spring/controller/ImageController.java b/spring-mvc-xml/src/main/java/com/baeldung/spring/controller/ImageController.java
index c9e846905a..75a1a91f82 100644
--- a/spring-mvc-xml/src/main/java/com/baeldung/spring/controller/ImageController.java
+++ b/spring-mvc-xml/src/main/java/com/baeldung/spring/controller/ImageController.java
@@ -12,6 +12,7 @@ import org.springframework.web.context.support.ServletContextResource;
 
 import javax.servlet.ServletContext;
 import javax.servlet.http.HttpServletResponse;
+import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 
@@ -49,7 +50,9 @@ public class ImageController {
             byte[] media = IOUtils.toByteArray(in);
             headers.setCacheControl(CacheControl.noCache().getHeaderValue());
             responseEntity = new ResponseEntity<>(media, headers, HttpStatus.OK);
-        } catch (IOException ioe) {
+        } catch (FileNotFoundException fnfe) {
+            responseEntity = new ResponseEntity<>(null, headers, HttpStatus.NOT_FOUND);
+        } catch (Exception e) {
             responseEntity = new ResponseEntity<>(null, headers, HttpStatus.INTERNAL_SERVER_ERROR);
         }
         return responseEntity;