add exception handler for the download error

jesuspg · May 5, 2014 · 69b52b8 · 69b52b8
1 parent b24d316
commit 69b52b8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...c/main/java/com/telefonica/euro_iaas/sdc/puppetwrapper/controllers/GenericController.java b/...c/main/java/com/telefonica/euro_iaas/sdc/puppetwrapper/controllers/GenericController.java
@@ -24,15 +24,14 @@
 
 package com.telefonica.euro_iaas.sdc.puppetwrapper.controllers;
 
-import java.io.IOException;
 import java.util.NoSuchElementException;
 
 import javax.servlet.http.HttpServletResponse;
 
 import org.springframework.web.bind.annotation.ExceptionHandler;
 import org.springframework.web.servlet.ModelAndView;
 
-import com.telefonica.euro_iaas.sdc.puppetwrapper.common.Error;
+import com.telefonica.euro_iaas.sdc.puppetwrapper.data.ModuleDownloaderException;
 
 public class GenericController {
 
@@ -41,6 +40,12 @@ public ModelAndView handleNoSuchElementException(NoSuchElementException ex, Http
         response.setStatus(HttpServletResponse.SC_NOT_FOUND);
         return handleModelAndView(ex.getMessage());
     }
+
+    @ExceptionHandler(ModuleDownloaderException.class)
+    public ModelAndView handleModuleDownloaderException(ModuleDownloaderException ex, HttpServletResponse response) {
+        response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
+        return handleModelAndView(ex.getMessage());
+    }
 
     public ModelAndView handleModelAndView(String error) {
         ModelAndView model = new ModelAndView();