package com.lovo.util;
import java.io.BufferedReader; import java.io.IOException; import java.io.InputStreamReader; import java.net.URLDecoder; import java.util.ArrayList; import java.util.Collections; import java.util.Enumeration; import java.util.HashMap; import java.util.Iterator; import java.util.List; import java.util.Map;
import javax.servlet.Filter; import javax.servlet.FilterChain; import javax.servlet.FilterConfig; import javax.servlet.ServletException; import javax.servlet.ServletRequest; import javax.servlet.ServletResponse; import javax.servlet.annotation.WebFilter; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletRequestWrapper;
/** * 对put,delete的restful请求进行过滤处理,主要是解析请求参数 * * @author ljh Servlet Filter implementation class PutMethodFilter */ @WebFilter("/*") public class PutMethodFilter implements Filter {
/** * Default constructor. */ public PutMethodFilter() {
}
/** * @see Filter#destroy() */ public void destroy() {
}
/** * @see Filter#doFilter(ServletRequest, ServletResponse, FilterChain) */ public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { request.setCharacterEncoding("utf-8"); response.setContentType("text/html;charset=utf-8");
HttpServletRequest req = (HttpServletRequest) request;
// 得到请求提交方式 String method = req.getMethod().toUpperCase();
// PUT请求不能通过 request.getparameter()得到参数 if ("GET".equals(method) || "PUT".equals(method) || "DELETE".equals(method)) {
StringBuilder content = new StringBuilder();
if ("GET".equals(method)) { if (req.getQueryString() != null) { content.append(req.getQueryString()); } } else { BufferedReader in = new BufferedReader(new InputStreamReader( request.getInputStream())); String org_req_content;
while ((org_req_content = in.readLine()) != null) { content.append(org_req_content); } } Map<String, String[]> map = parseRequestParameter(URLDecoder .decode(content.toString(), "UTF-8")); request.setAttribute(PutRequestWrapper.CUSTOMER_ATTR_KEY, map);
// 包装成自定义的请�? request = new PutRequestWrapper(req); }
chain.doFilter(request, response); }
/** * 自定义参数解�? String queryStr = "c=8&d=100&f=&a=1&a=2&b=1&b=&c=&d=1&q=999&c="; * * @param content * @return */ public Map<String, String[]> parseRequestParameter(String content) { Map<String, String[]> params = new HashMap<String, String[]>(); if (content == null || "".equals(content.trim())) { return params; }
String[] tmp = content.split("&");
// 处理多�?情况 Map<String, List<String>> multiValues = null;
for (int i = 0; i < tmp.length; i++) { String[] keyValue = tmp[i].split("="); if (params.containsKey(keyValue[0])) { // 处理checkbox之类的多值情�?,比如:ch1=111&ch1=222 List valuesList; if (multiValues == null) { multiValues = new HashMap<String, List<String>>(); } if (multiValues.containsKey(keyValue[0])) { valuesList = multiValues.get(keyValue[0]); if (keyValue.length >= 2) { valuesList.add(keyValue[1]); } else { // 处理提交的参数中没有值的情况,比如:?aa=&bb=111,其中aa就没有取�? valuesList.add(""); } } else { valuesList = new ArrayList(); valuesList.add(params.get(keyValue[0])[0]); if (keyValue.length >= 2) { valuesList.add(keyValue[1]); } else { // 处理提交的参数中没有值的情况,比如:?aa=&bb=111,其中aa就没有取�? valuesList.add(""); } multiValues.put(keyValue[0], valuesList); }
} else { // 普�?情况,没有多�? if (keyValue.length >= 2) { params.put(keyValue[0], new String[] { keyValue[1] }); } else { // 处理提交的参数中没有�?��情况,比如:?aa=&bb=111,其中aa就没有取�? params.put(keyValue[0], new String[] { "" }); } } }// end for
if (multiValues != null) { // 有多个�?的情�? for (Iterator<String> its = multiValues.keySet().iterator(); its .hasNext();) { String key = its.next(); List<String> values = multiValues.get(key); String[] strArray = new String[values.size()]; for (int i = 0; i < values.size(); i++) { strArray[i] = values.get(i); } params.put(key, strArray); } }
return params; }
/** * @see Filter#init(FilterConfig) */ public void init(FilterConfig fConfig) throws ServletException { }
}
/** * put请求包装�? * * @author ljh * */ class PutRequestWrapper extends HttpServletRequestWrapper {
// 自定义解析完成后的参数,放入request中的key public final static String CUSTOMER_ATTR_KEY = "customer.attribute.parameter";
private Map<String, String[]> reqParams;
public PutRequestWrapper(HttpServletRequest request) { super(request); reqParams = (Map<String, String[]>) request .getAttribute(CUSTOMER_ATTR_KEY); }
@Override public Map getParameterMap() { return reqParams; }
@Override public Enumeration getParameterNames() { return Collections.enumeration(reqParams.keySet()); }
@Override public String[] getParameterValues(String name) { return reqParams.get(name); }
}