ServletAnnotationMappingUtils.java

package com.hundsun.jresplus.web.adapter;

import java.util.Iterator;
import java.util.List;

import javax.servlet.http.HttpServletRequest;

import org.springframework.http.MediaType;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.util.WebUtils;

/**
 * @author sagahl copy by fish
 * 
 */
public class ServletAnnotationMappingUtils {

	/**
	 * Check whether the given request matches the specified request methods.
	 * 
	 * @param methods
	 *            the HTTP request methods to check against
	 * @param request
	 *            the current HTTP request to check
	 */
	public static boolean checkRequestMethod(RequestMethod[] methods,
			HttpServletRequest request) {
		if (ObjectUtils.isEmpty(methods)) {
			return true;
		}
		for (RequestMethod method : methods) {
			if (method.name().equals(request.getMethod())) {
				return true;
			}
		}
		return false;
	}

	/**
	 * Check whether the given request matches the specified parameter
	 * conditions.
	 * 
	 * @param params
	 *            the parameter conditions, following
	 *            {@link RequestMapping#params()}
	 * @param request
	 *            the current HTTP request to check
	 */
	public static boolean checkParameters(String[] params,
			HttpServletRequest request) {
		if (!ObjectUtils.isEmpty(params)) {
			for (String param : params) {
				int separator = param.indexOf('=');
				if (separator == -1) {
					if (param.startsWith("!")) {
						if (WebUtils.hasSubmitParameter(request,
								param.substring(1))) {
							return false;
						}
					} else if (!WebUtils.hasSubmitParameter(request, param)) {
						return false;
					}
				} else {
					String key = param.substring(0, separator);
					String value = param.substring(separator + 1);
					if (!value.equals(request.getParameter(key))) {
						return false;
					}
				}
			}
		}
		return true;
	}

	/**
	 * Check whether the given request matches the specified header conditions.
	 * 
	 * @param headers
	 *            the header conditions, following
	 *            {@link RequestMapping#headers()}
	 * @param request
	 *            the current HTTP request to check
	 */
	public static boolean checkHeaders(String[] headers,
			HttpServletRequest request) {
		if (!ObjectUtils.isEmpty(headers)) {
			for (String header : headers) {
				int separator = header.indexOf('=');
				if (separator == -1) {
					if (header.startsWith("!")) {
						if (request.getHeader(header.substring(1)) != null) {
							return false;
						}
					} else if (request.getHeader(header) == null) {
						return false;
					}
				} else {
					String key = header.substring(0, separator);
					String value = header.substring(separator + 1);
					if (isMediaTypeHeader(key)) {
						List<MediaType> requestMediaTypes = MediaType
								.parseMediaTypes(request.getHeader(key));
						List<MediaType> valueMediaTypes = MediaType
								.parseMediaTypes(value);
						boolean found = false;
						for (Iterator<MediaType> valIter = valueMediaTypes
								.iterator(); valIter.hasNext() && !found;) {
							MediaType valueMediaType = valIter.next();
							for (Iterator<MediaType> reqIter = requestMediaTypes
									.iterator(); reqIter.hasNext() && !found;) {
								MediaType requestMediaType = reqIter.next();
								if (valueMediaType.includes(requestMediaType)) {
									found = true;
								}
							}

						}
						if (!found) {
							return false;
						}
					} else if (!value.equals(request.getHeader(key))) {
						return false;
					}
				}
			}
		}
		return true;
	}

	private static boolean isMediaTypeHeader(String headerName) {
		return "Accept".equalsIgnoreCase(headerName)
				|| "Content-Type".equalsIgnoreCase(headerName);
	}

}