$.RestExceptionHandler.java Source code

Introduction

Here is the source code for $.RestExceptionHandler.java

Source

    #set($symbol_pound='#')#set($symbol_dollar='$')#set($symbol_escape='\')
/*******************************************************************************
 * Copyright (c) 2005, 2014 springside.github.io
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *******************************************************************************/
package ${package}.rest;

    import java.util.Map;

    import javax.validation.ConstraintViolationException;

    import org.springframework.http.HttpHeaders;
    import org.springframework.http.HttpStatus;
    import org.springframework.http.MediaType;
    import org.springframework.http.ResponseEntity;
    import org.springframework.web.bind.annotation.ControllerAdvice;
    import org.springframework.web.bind.annotation.ExceptionHandler;
    import org.springframework.web.context.request.WebRequest;
    import org.springframework.web.servlet.mvc.method.annotation.ResponseEntityExceptionHandler;
    import org.springside.modules.beanvalidator.BeanValidators;
    import org.springside.modules.mapper.JsonMapper;
    import org.springside.modules.web.MediaTypes;

    /**
     * ExceptionHandler?Restful.
     * 
     * @author calvin
     */
    // Spring-MVC????Controllerannotation
    @ControllerAdvice
    public class RestExceptionHandler extends ResponseEntityExceptionHandler {
        private JsonMapper jsonMapper = new JsonMapper();

        /**
         * ?RestException.
         */
        @ExceptionHandler(value = { RestException.class })
        public final ResponseEntity<?> handleException(RestException ex, WebRequest request) {
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.parseMediaType(MediaTypes.TEXT_PLAIN_UTF_8));
            return handleExceptionInternal(ex, ex.getMessage(), headers, ex.status, request);
        }

        /**
         * ?JSR311 Validation.
         */
        @ExceptionHandler(value = { ConstraintViolationException.class })
        public final ResponseEntity<?> handleException(ConstraintViolationException ex, WebRequest request) {
            Map<String, String> errors = BeanValidators.extractPropertyAndMessage(ex.getConstraintViolations());
            String body = jsonMapper.toJson(errors);
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.parseMediaType(MediaTypes.TEXT_PLAIN_UTF_8));
            return handleExceptionInternal(ex, body, headers, HttpStatus.BAD_REQUEST, request);
        }
    }