为了多次读取ServletInputStream引发的一系列问题

{

  head: null

body: null

token:xxxxxxxxxxxxxxxxxxxxx

}

public class MyHttpRequest extends HttpServletRequestWrapper {

  private static Logger log=Logger.getLogger(MyHttpRequest. class );

     private   byte [] bytes;

     /**

      * @param request {@link javax.servlet.http.HttpServletRequest} object.

      * @throws IOException

      */

     public MyHttpRequest(HttpServletRequest request) throws IOException {

         super (request);

         bytes= IOUtils.toByteArray( request.getInputStream());     

     }

     @Override

     public ServletInputStream getInputStream() throws IOException {

         final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);

         return new DelegatingServletInputStream(byteArrayInputStream);

     }

     @Override

     public BufferedReader getReader() throws IOException {

      return new BufferedReader( new InputStreamReader(getInputStream()));

     }

public class MyFilter implements Filter{

  @Override

  public void destroy() {   

  }

  @Override

  public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain)

    throws IOException, ServletException {

   HttpServletRequest request=(HttpServletRequest) req;

   MyHttpRequest myrequest= new MyHttpRequest(request);

   chain.doFilter(myrequest, res); 

  }

  @Override

  public void init(FilterConfig arg0) throws ServletException { 

  }

}

package com.hrhs.jyj.filter;

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.UnsupportedEncodingException;

import java.util.ArrayList;

import java.util.Enumeration;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.ServletInputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletRequestWrapper;

import org.apache测试数据mons.io.IOUtils;

import org.apache.log4j.Logger;

import org.springframework.mock.web.DelegatingServletInputStream;

public class MyHttpRequest extends HttpServletRequestWrapper {

  private static Logger log = Logger.getLogger(MyHttpRequest. class );

  private byte [] bytes;

  private String body;

  private Map<String, List<String>> map;

  private int readMap= 0 ;

  private String queryString;

  /**

   * @param request

   *            {@link javax.servlet.http.HttpServletRequest} object.

   * @throws IOException

   */

  public MyHttpRequest(HttpServletRequest request) throws IOException {

   super (request);

   bytes = IOUtils.toByteArray(request.getInputStream());

   queryString = request.getQueryString();

  }

  @Override

  public ServletInputStream getInputStream() throws IOException {

   final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);

   return new DelegatingServletInputStream(byteArrayInputStream);

  }

  @Override

  public BufferedReader getReader() throws IOException {

   return new BufferedReader( new InputStreamReader(getInputStream()));

  }

  @Override

  public String getParameter(String name) {

   log.info( "走了getParameter" );

   return super .getParameter(name);

  }

  @Override

  public Map<String, String[]> getParameterMap() {

   log.info( "走了getParameterMap" );

   return super .getParameterMap();

  }

  @Override

  public Enumeration<String> getParameterNames() {

   log.info( "走了getParameterNames" );

   return super .getParameterNames();

  }

  //暂时先改写这一个,其他的也可以修改

  @Override

  public String[] getParameterValues(String name) {

   log.info( "走了getParameterValues" );

   try {

    Map<String, List<String>> nameVals = doParameter();

    List<String> list = nameVals.get(name);

    if (list!= null &&list.size()> 0 ){

     return list.toArray( new String[]{});

    }

   } catch (UnsupportedEncodingException e) {

    e.printStackTrace();

   }

   return new String[]{};

  }

  //这里获取所有参数值的map

  public Map<String, List<String>> doParameter() throws UnsupportedEncodingException {

   if (readMap== 0 ){

    //这里把post里的参数和地址栏参数结合到一起,然后解析

    body = new String(bytes, getCharacterEncoding())+ "&" +queryString;

    String[] nameVals = body.split( "&" );

    map = new HashMap<String, List<String>>();

    for (String nameVal : nameVals) {

     String name = nameVal.split( "=" )[ 0 ];

     String val = nameVal.split( "=" )[ 1 ];

     if (map.containsKey(name)) {

      List<String> vals = map.get(name);

      vals.add(val);

      map.put(name, vals);

     } else {

      List<String> vals = new ArrayList<String>();

      vals.add(val);

      map.put(name, vals);

     }

    }

    readMap= 1 ;

   }

   return map; 

  }

}

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import javax.servlet.ServletInputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletRequestWrapper;

public class BufferHttpServletRequestWrapper extends HttpServletRequestWrapper {

  private final byte [] body;

  public BufferHttpServletRequestWrapper(HttpServletRequest request) throws IOException {

   super (request);

   InputStream is = request.getInputStream();

         ByteArrayOutputStream baos = new ByteArrayOutputStream();

         byte buff[] = new byte [ 1024 ];

         int read;

         while ( ( read = is.read( buff ) ) > 0 ) {

             baos.write( buff, 0 , read );

         }

         body = baos.toByteArray();

  }

  @Override

  public BufferedReader getReader() throws IOException {

   return new BufferedReader( new InputStreamReader(getInputStream()));

  }

  @Override

  public ServletInputStream getInputStream() throws IOException {

   final ByteArrayInputStream bais = new ByteArrayInputStream(body);

   return new ServletInputStream() {

    @Override

    public int read() throws IOException {

     return bais.read();

    }

   };

  }

}

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

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.http.HttpServletRequest;

/**

  * Servlet Filter implementation class PostDataDumperFilter

  */

public class PostDataDumperFilter implements Filter {

  private FilterConfig filterConfig = null ;

     /**

      * Default constructor.

      */

     public PostDataDumperFilter() {

         // TODO Auto-generated constructor stub

     }

  /**

   * @see Filter#destroy()

   */

  public void destroy() {

   this .filterConfig = null ;

  }

  /**

   * @see Filter#doFilter(ServletRequest, ServletResponse, FilterChain)

   */

  public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {

   if (filterConfig == null )

    return ;

   //备份HttpServletRequest

   ServletRequest requestWrapper = null ; 

         if (request instanceof HttpServletRequest) { 

             requestWrapper = new BufferHttpServletRequestWrapper((HttpServletRequest) request); 

         }

   //使用流

   InputStream reader = requestWrapper.getInputStream();

   ByteArrayOutputStream byteOutput = new ByteArrayOutputStream( 100 );

   int i = 0 ;

   byte [] b = new byte [ 100 ];

   while ((i = reader.read(b))!= - 1 ){

    byteOutput.write(b, 0 , i);

   }

   request.setAttribute( "post" , new String(byteOutput.toByteArray()));

   // pass the request along the filter chain

   if ( null == requestWrapper){

    chain.doFilter(request, response);

   } else { 

             chain.doFilter(requestWrapper, response); 

         }

  }

  /**

   * @see Filter#init(FilterConfig)

   */

  public void init(FilterConfig fConfig) throws ServletException {

   this .filterConfig = fConfig;

  }

}