多次讀取HttpServletRequest的inputstream方法 問題解決

原因：我要收集所有來自前台請求的參數信息，無論在任何地方的。當前請求參數都是json格式，都寫在httpservlet的body中。這個只能通過流進行獲取。然后問題來了，HttpServletRequest的inputstream只能讀取一次。。。。

解決：重寫request的inputstream方法。。然后在需要部署應用中加上過濾器，在過濾器中加上這個重寫的request的方法，問題解決。

1、讀取流信息的工具類：HttpUtil.java

/**獲取請求Body
     * 
     * @param request
     * @return
     * @author : chewneixian 陳惟鮮
     * @create_date 2016年12月6日 下午12:54:07
     */
    public static String getBodyString(ServletRequest request) {
        StringBuilder sb = new StringBuilder();
        InputStream inputStream = null;
        BufferedReader reader = null;
        try {
            inputStream = request.getInputStream();
            reader = new BufferedReader(new InputStreamReader(inputStream, Charset.forName("UTF-8")));
            String line = "";
            while ((line = reader.readLine()) != null) {
                sb.append(line);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (reader != null) {
                try {
                    reader.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return sb.toString();
    }

 

2、重寫httpservlet的inputstream類：BodyReaderHttpServletRequestWrapper

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

import com.iafclub.baseTools.util.HttpUtil;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.Charset;

/**重寫HttpServletRequestWrapper
 * 
 * @author 陳惟鮮
 * @date 2016年12月7日 下午3:06:52
 *
 */
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {

    private final byte[] body;

    public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        body = HttpUtil.getBodyString(request).getBytes(Charset.forName("UTF-8"));
    }

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

3、過濾器中加上重寫方法：AccessFilter

import java.io.IOException;
import java.util.regex.Pattern;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletContext;
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.HttpServletResponse;

import org.apache.log4j.Logger;

import com.iafclub.baseTools.util.HttpUtil;
import com.iafclub.demo.aop.BodyReaderHttpServletRequestWrapper;
/**訪問過濾器
 * 
 * @author chenweixian
 *
 */
@WebFilter(filterName="accessFilter", urlPatterns={
        "*.do",
//        "*.jsp",
//        "/*",
//        "/layout/*",
//        "/apply/*",
//        "/audit/*",
//        "/authority/*",
//        "/commonWare/*",
//        "/contract/*",
//        "/marketing/*",
//        "/product/*",
//        "/project/*",
//        "/system/*",
//        "/user/*"
        })
public class AccessFilter implements Filter {
    //     日志對象
    private static Logger logger = Logger.getLogger(AccessFilter.class);

    public void init(FilterConfig filterConfig) throws ServletException {
//        ServletContext context = filterConfig.getServletContext();  
    }
    
    public void destroy() {
    }

    public void doFilter(ServletRequest req, ServletResponse res,FilterChain chain) throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) req;
        System.out.println("=====+" + req.hashCode());
        HttpServletResponse response = (HttpServletResponse) res;        
        if ("POST".equalsIgnoreCase(request.getMethod())) {
            
            ServletRequest requestWrapper = new BodyReaderHttpServletRequestWrapper(request);
            System.out.println("===filter==+" + requestWrapper.hashCode());

            String body = HttpUtil.getBodyString(requestWrapper);
            System.out.println("AccessFilter="+body);
            chain.doFilter(requestWrapper, response);
            return ;
        }
        
        chain.doFilter(req, res);
    }
}

4、攔截器：ControllerAopInterceptor（其實我們這里根本用到這個東西，加載這里的目的是想說，過濾器訪問在aop切面之前，過濾器修改request，其他訪問的request都被修改）

import javax.servlet.http.HttpServletRequest;

import org.apache.log4j.Logger;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.AfterThrowing;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Service;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import com.iafclub.baseTools.util.HttpUtil;
import com.iafclub.baseTools.util.MyCollectionUtil;

/**切面  
 * 
 * @author chenweixian
 *
 */
@Aspect
@Service
public class ControllerAopInterceptor {
    
    private Logger logger = Logger.getLogger(ControllerAopInterceptor.class);
    
    @Pointcut("execution(* com.iafclub.demo.web.controller..*(..))")  
    private void anyMethod(){}//定義一個切入點  
      
    @Before("anyMethod() && args(request)") 
    public void doAccessCheck(Object request){
        logger.info(request);  
        logger.info("前置通知");  
    }  
      
    @AfterReturning("anyMethod() && args(request)")  
    public void doAfter(Object request){  
        logger.info(request);  
        logger.info("后置通知");  
    }  
    
    @Around("anyMethod()")  
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        RequestAttributes ra = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes sra = (ServletRequestAttributes) ra;
        HttpServletRequest request = sra.getRequest();
        
        String url = request.getRequestURL().toString();
        String method = request.getMethod();
        String uri = request.getRequestURI();
        String queryString = request.getQueryString();
        Object[] requestObject = pjp.getArgs();
        
        String body = HttpUtil.getBodyString(request);
        System.out.println("====aop=+" + request.hashCode());

        System.out.println("ControllerAopInterceptor="+body);
        logger.info("請求開始, 各個參數, url: {"+url+"}, method: {"+method+"}, uri: {"+uri+"}, params: {"+queryString+"},requestObject{"+requestObject+"}");
        // result的值就是被攔截方法的返回值
        Object result = pjp.proceed();
        logger.info("請求結束，controller的返回值是 " + MyCollectionUtil.objectToMapAll(result));
        return result;
    }
      
    @After("anyMethod()")  
    public void after(){  
        logger.info("最終通知");  
    }  
      
    @AfterThrowing("anyMethod()")  
    public void doAfterThrow(){  
        logger.info("例外通知");  
    }  
    
}

5、真正處理的類也不會受到影響：

/**測試首頁
 * 
 * @author chenweixian
 *
 */
@Controller
public class IndexController {

    @Autowired
    private StringRedisTemplate stringRedisTemplate;
    @Autowired
    private DictionaryService dictionaryService;
    /**
     * 首頁
     * @throws IOException 
     * */
    @RequestMapping("index.do")
    public String index(Model model, HttpServletRequest request) throws IOException{

        String body = HttpUtil.getBodyString(request);
        System.out.println("==index===+" + request.hashCode());

        System.out.println("IndexController="+body);
        return "index";
    }
}

哈哈。。。。。。。。。。。

結果：