对put,delete的restful请求进行过滤处理的工具类

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);
 }

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值