All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.giants.web.springmvc.advice.JsonResultResponseAdvice Maven / Gradle / Ivy

Go to download

The newest version!
/**
 * 
 */
package com.giants.web.springmvc.advice;

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

import com.giants.common.collections.CollectionUtils;
import org.apache.commons.lang.ArrayUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.support.ResourceBundleMessageSource;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.http.server.ServletServerHttpRequest;
import org.springframework.http.server.ServletServerHttpResponse;
import org.springframework.web.context.support.WebApplicationContextUtils;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

import com.giants.common.GiantsConstants;
import com.giants.common.regex.Pattern;
import com.giants.web.springmvc.json.JsonResult;
import com.giants.web.springmvc.json.JsonpResult;

import java.util.List;

/**
 * @author vencent.lu
 *
 */
public class JsonResultResponseAdvice implements ResponseBodyAdvice {
	
    private static final Pattern CALLBACK_PARAM_PATTERN = Pattern.compile("[0-9A-Za-z_\\.]*");
    protected final Logger logger = LoggerFactory.getLogger(getClass());    
    private String[] jsonpQueryParamNames;
    
	private ResourceBundleMessageSource resourceBundleMessageSource;

	private List uriExcludeList;
		
	@Override
	public boolean supports(MethodParameter returnType,
			Class> converterType) {
		return true;
	}

	@Override
	public Object beforeBodyWrite(Object body, MethodParameter returnType,
			MediaType selectedContentType,
			Class> selectedConverterType,
			ServerHttpRequest request, ServerHttpResponse response) {
		HttpServletRequest servletRequest = ((ServletServerHttpRequest) request).getServletRequest();
		HttpServletResponse servletResponse = ((ServletServerHttpResponse)response).getServletResponse();
		if (this.resourceBundleMessageSource == null) {
			this.resourceBundleMessageSource = WebApplicationContextUtils
					.getWebApplicationContext(servletRequest.getServletContext())
					.getBean(ResourceBundleMessageSource.class);
		}
		Object result = null;
		if (CollectionUtils.isEmpty(this.uriExcludeList) || !this.uriExcludeList.contains(servletRequest.getRequestURI())) {
			JsonResult jsonResult = new JsonResult();
			jsonResult.setData(body);
			if (servletResponse.getStatus() == 200) {
				jsonResult.setCode(GiantsConstants.ERROR_CODE_SUCCESS);
				jsonResult.setMessage(this.resourceBundleMessageSource.getMessage("operation.success", null, servletRequest.getLocale()));
			} else {
				jsonResult.setCode(GiantsConstants.ERROR_CODE_SYSTEM_ERROR);
				jsonResult.setMessage(this.resourceBundleMessageSource.getMessage("operation.systemerror", null, servletRequest.getLocale()));
			}
			result = jsonResult;
		} else {
			result = body;
		}
		if (ArrayUtils.isNotEmpty(this.jsonpQueryParamNames)) {
		    for (String name : this.jsonpQueryParamNames) {
	            String value = servletRequest.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");
	                response.getHeaders().setContentType(contentTypeToUse);
	                JsonpResult jsonpResult = new JsonpResult();
	                jsonpResult.setJsonpFunction(value);
	                jsonpResult.setValue(result);
	                return jsonpResult;
	            }
	        }
		}
		return result;
	}
	
	public void setJsonpQueryParamName(String jsonpQueryParamName) {
        this.jsonpQueryParamNames = new String[]{jsonpQueryParamName};
    }

    public void setJsonpQueryParamNames(String... jsonpQueryParamNames) {
        this.jsonpQueryParamNames = jsonpQueryParamNames;
    }

	public void setUriExcludeList(List uriExcludeList) {
		this.uriExcludeList = uriExcludeList;
	}
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy