解决request.getInputStream()与request.getReader()只能调用一次的问题

4 篇文章 0 订阅

request.getInputStream()与request.getReader()只能调用一次,因为是流,读取完就没了。

 

然而,有时我们可能会需要读取多次的场景,如在前端过滤器中,拦截请求数据并记录日志,记录日志后,请求继续向后传递,在后面的业务代码中,仍然可以继续取得请求的数据。

 

怎么办,办法总是有的,重写HttpServletRequestWrapper,自己实现一个Wrapper。

 

    package com.inspur.cloud.am;  
      
    import java.io.BufferedReader;  
    import java.io.ByteArrayInputStream;  
    import java.io.IOException;  
    import java.io.InputStreamReader;  
    import javax.servlet.ServletInputStream;  
    import javax.servlet.http.HttpServletRequest;  
    import javax.servlet.http.HttpServletRequestWrapper;  
    import org.springframework.util.StreamUtils;  
      
    public class MyHttpServletRequestWrapper extends HttpServletRequestWrapper{  
      
        private byte[] requestBody = null;   
          
        public MyHttpServletRequestWrapper(HttpServletRequest request) {  
              
            super(request);  
              
            //缓存请求body  
            try {  
                requestBody = StreamUtils.copyToByteArray(request.getInputStream());  
            } catch (IOException e) {  
                e.printStackTrace();  
            }    
        }  
          
        /** 
         * 重写 getInputStream() 
         */  
        @Override  
        public ServletInputStream getInputStream() throws IOException {  
            if(requestBody == null){  
                requestBody= new byte[0];  
            }  
            final ByteArrayInputStream bais = new ByteArrayInputStream(requestBody);    
            return new ServletInputStream() {    
                @Override    
                public int read() throws IOException {    
                    return bais.read();    
                }    
            };    
        }  
      
        /** 
         * 重写 getReader() 
         */  
        @Override  
        public BufferedReader getReader() throws IOException {  
            return new BufferedReader(new InputStreamReader(getInputStream()));    
        }  
          
      
    }  

在最前端的filter中,new一个自己的wrapper,并doChain继续向后传递即可。
 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值