com.giants.web.springmvc.resolver.JsonResultExceptionResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of giants-web-springmvc Show documentation
The newest version!
/**
 * 
 */
package com.giants.web.springmvc.resolver;

import java.io.IOException;
import java.util.Collections;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.ArrayUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.support.ResourceBundleMessageSource;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.http.server.ServletServerHttpResponse;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.support.WebApplicationContextUtils;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerExceptionResolver;
import org.springframework.web.servlet.ModelAndView;

import com.giants.common.regex.Pattern;
import com.giants.web.springmvc.exception.BuildExceptionJsonResult;
import com.giants.web.springmvc.json.JsonpResult;

/**
 * 将异常解析为JsonResult {@link com.giants.web.springmvc.json.JsonResult}
 * 最终序列化成 json 输出
 * @author vencent.lu
 * @since 1.0.0
 */

public class JsonResultExceptionResolver implements HandlerExceptionResolver {

    private static final Pattern CALLBACK_PARAM_PATTERN = Pattern.compile("[0-9A-Za-z_\\.]*");
    protected final Logger logger = LoggerFactory.getLogger(getClass());    
    private String[] jsonpQueryParamNames;
	
	private List> messageConverters;
	
	private boolean includeModelAndView = false;
	
	private ResourceBundleMessageSource resourceBundleMessageSource;

	/* (non-Javadoc)
	 * @see org.springframework.web.servlet.HandlerExceptionResolver#resolveException(javax.servlet.http.HttpServletRequest, javax.servlet.http.HttpServletResponse, java.lang.Object, java.lang.Exception)
	 */
	@Override
	public ModelAndView resolveException(HttpServletRequest request,
			HttpServletResponse response, Object handler, Exception ex) {
		boolean isReturnJson = false;
		HandlerMethod handlerMethod = (HandlerMethod) handler;
		if (this.includeModelAndView) {
			isReturnJson = true;
		} else if (handler != null
				&& (AnnotationUtils.findAnnotation(handlerMethod.getBeanType(), RestController.class) != null
				|| AnnotationUtils.findAnnotation(handlerMethod.getMethod(), ResponseBody.class) != null)) {
			isReturnJson = true;
		}
		
		if (isReturnJson) {
	        if (this.messageConverters != null) {
	        	if (this.resourceBundleMessageSource == null) {
					this.resourceBundleMessageSource = WebApplicationContextUtils
							.getWebApplicationContext(request.getServletContext())
							.getBean(ResourceBundleMessageSource.class);
				}
	        	Object result = BuildExceptionJsonResult.build(ex,this.resourceBundleMessageSource,request.getLocale());
	        	HttpOutputMessage outputMessage = new ServletServerHttpResponse(response);
	        	if (ArrayUtils.isNotEmpty(this.jsonpQueryParamNames)) {
	                for (String name : this.jsonpQueryParamNames) {
	                    String value = request.getParameter(name);
	                    if (value != null) {
	                        if (!CALLBACK_PARAM_PATTERN.matches(value)) {
	                            if (logger.isDebugEnabled()) {
	                                logger.debug("Ignoring invalid jsonp parameter value: " + value);
	                            }
	                            continue;
	                        }
	                        MediaType contentTypeToUse = new MediaType("application", "javascript");
	                        outputMessage.getHeaders().setContentType(contentTypeToUse);
	                        JsonpResult jsonpResult = new JsonpResult();
	                        jsonpResult.setJsonpFunction(value);
	                        jsonpResult.setValue(result);
	                        result = jsonpResult;
	                    }
	                }
	            }	        	
				HttpInputMessage inputMessage = new ServletServerHttpRequest(request);  
		        List acceptedMediaTypes = inputMessage.getHeaders().getAccept();  
		        if (acceptedMediaTypes.isEmpty()) {  
		            acceptedMediaTypes = Collections.singletonList(MediaType.ALL);
		        }  
		        MediaType.sortByQualityValue(acceptedMediaTypes);		        
		        Class returnValueType = result.getClass();
	            for (MediaType acceptedMediaType : acceptedMediaTypes) {
	                for (HttpMessageConverter