RestExcpetion

README.md

@@ -332,7 +332,7 @@ public interface VoteRepository extends CrudRepository<Vote, Long> {
 * At runtime, Spring Data JPA replaces the `?1` placeholder with the passed-in `pollId` parameter value.
 
 
-### Part 3.2.3 - Modify `VoteController`
+### Part 3.2.3 - Modify `VoteCo*_****_*ntroller`
 
 * Create a `getAllVotes` method in the `VoteController`
 

src/main/java/io/zipcoder/tc_spring_poll_application/controller/PollController.java

@@ -43,6 +43,7 @@ public class PollController {
 
     @RequestMapping(value="/polls/{pollId}", method=RequestMethod.GET)
     public ResponseEntity<?> getPoll(@PathVariable Long pollId) {
+        verifyPoll(pollId);
         Poll p = pollRepository.findOne(pollId);
         return new ResponseEntity<> (p, HttpStatus.OK);
     }
@@ -50,17 +51,24 @@ public class PollController {
     @RequestMapping(value="/polls/{pollId}", method=RequestMethod.PUT)
     public ResponseEntity<?> updatePoll(@RequestBody Poll poll, @PathVariable Long pollId) {
         // Save the entity
+        verifyPoll(pollId);
         Poll p = pollRepository.save(poll);
         return new ResponseEntity<>(HttpStatus.OK);
     }
 
     @RequestMapping(value="/polls/{pollId}", method=RequestMethod.DELETE)
     public ResponseEntity<?> deletePoll(@PathVariable Long pollId) {
+        verifyPoll(pollId);
         pollRepository.delete(pollId);
         return new ResponseEntity<>(HttpStatus.OK);
     }
 
-    public void verifyPoll(){
+   // @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.GET)
+    public void verifyPoll(Long pollId) throws ResourceNotFoundException{
+
+        if(pollRepository.findOne(pollId) == null){
+            throw new ResourceNotFoundException();
+        }
     }
 
 }

src/main/java/io/zipcoder/tc_spring_poll_application/dtos/error/ErrorDetail.java

@@ -0,0 +1,60 @@
+package io.zipcoder.tc_spring_poll_application.dtos.error;
+
+public class ErrorDetail {
+    private String title;
+    private int status;
+    private String detail;
+    private long timestamp;
+    private String developerMessage;
+
+    public ErrorDetail(String title, int status, String detail, long timestamp, String developerMessage) {
+        this.title = title;
+        this.status = status;
+        this.detail = detail;
+        this.timestamp = timestamp;
+        this.developerMessage = developerMessage;
+    }
+
+    public ErrorDetail() {
+    }
+
+    public String getTitle() {
+        return title;
+    }
+
+    public void setTitle(String title) {
+        this.title = title;
+    }
+
+    public int getStatus() {
+        return status;
+    }
+
+    public void setStatus(int status) {
+        this.status = status;
+    }
+
+    public String getDetail() {
+        return detail;
+    }
+
+    public void setDetail(String detail) {
+        this.detail = detail;
+    }
+
+    public long getTimestamp() {
+        return timestamp;
+    }
+
+    public void setTimestamp(long timestamp) {
+        this.timestamp = timestamp;
+    }
+
+    public String getDeveloperMessage() {
+        return developerMessage;
+    }
+
+    public void setDeveloperMessage(String developerMessage) {
+        this.developerMessage = developerMessage;
+    }
+}

src/main/java/io/zipcoder/tc_spring_poll_application/dtos/error/RestExceptionHandler.java

@@ -0,0 +1,24 @@
+package io.zipcoder.tc_spring_poll_application.dtos.error;
+import io.zipcoder.tc_spring_poll_application.exception.ResourceNotFoundException;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Date;
+
+@ControllerAdvice
+public class RestExceptionHandler {
+
+    @ExceptionHandler(ResourceNotFoundException.class)
+    public ResponseEntity<?> handleResourceNotFoundException(ResourceNotFoundException rnfe, HttpServletRequest request) {
+        ErrorDetail errorDetail = new ErrorDetail();
+        errorDetail.setTimestamp(new Date().getTime());
+        errorDetail.setDeveloperMessage(rnfe.getMessage());
+        errorDetail.setDetail(rnfe.getStackTrace().toString());
+
+
+        return new ResponseEntity<>(errorDetail, HttpStatus.NOT_FOUND);
+    }
+}