SpringBoot拦截器读取流后不能再读取的问题

这篇文章主要介绍了SpringBoot拦截器读取流后不能再读取的问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

解决这个问题我能想到两种方式

1.通过修改标记的方式 ( inputstream.markSupported() 方法可以判断这个流是否支持 mark 和 reset 方法。他们分别是标记和重新定位流。)

2.将流赋值给一个 byte[] 数组，或者其他变量保存起来。下载读取流时就调用这个数组就行。

第一种方法

再回到问题上来我们可以先使用第一种方法判断 requet 中的inputStream 是否支持标记和重新定位。因为这种方式实现起来比较简单。无需考虑太多。

				?

									    @Override

									    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {

									         boolean b = request.getInputStream().markSupported();

									         System.out.println(b);

									    }

									// 输出结果为 false

上述代码会返回一个 false 那么很明显，request 中的 input 流是不支持标记和重新定位的。

第二种方法

我们再考虑第二种方法，我们需要一个变量保存这个流。并且还要保证再过滤器中和controller中都要拿到这个变量。直接定义一个全局变量获取修改传值方式，都是可以的。全局变量这种方式我就不演示了。

下面是改变传值方式的 demo

				?

									  public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {

									        ServletInputStream inputStream1 = request.getInputStream();

									        // 各种对 inputStream1 处理的操作...

									        Object obejct = inputStream1;

									        request.setAttribute("Params",obejct);

									}

这样就可以再controller那边就可以直接获取 Attribute 中的值。

但是！这样有很大的局限性，例如: 我已经写好了大多数的controller方法体。这时再改用这种方式传值。对于开发人员是一种莫大的痛苦 -_- 于是通过不屑的百度查询到另一种方法一一一改写HttpServletRequestWrapper方法。为什么是这个方法？因为他实现了 HttpServletRequest 这个接口。也是我们拦截器接收的 request 要求的类型。首先我们先看一张图。** 注意 filter 和 inteceptor 中间。 **

SpringBoot拦截器读取流后不能再读取的问题

通过上方这个图我们可以知道，在 Filter 和 Inteceptor 中间有一层Servlet。而Servlet就是提交request的地方。所以我们要重写HttpServletRequest方法只能在Servlet之前。也就是filter 中。下面就是直接上代码了

1.重写 HttpServletRequest

				?

									package com.xqw.kyg.util;

									import java.io.BufferedReader;

									import java.io.ByteArrayInputStream;

									import java.io.IOException;

									import java.io.InputStreamReader;

									import javax.servlet.ReadListener;

									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) throws IOException {

									        super(request);

									        requestBody = StreamUtils.copyToByteArray(request.getInputStream());

									    }

									    @Override

									    public ServletInputStream getInputStream() {

									        final ByteArrayInputStream bais = new ByteArrayInputStream(requestBody);

									        return new ServletInputStream() {

									            @Override

									            public int read(){

									                return bais.read();  // 读取 requestBody 中的数据

									            }

									            @Override

									            public boolean isFinished() {

									                return false;

									            }

									            @Override

									            public boolean isReady() {

									                return false;

									            }

									            @Override

									            public void setReadListener(ReadListener readListener) { }

									        };

									    }

									    @Override

									    public BufferedReader getReader() throws IOException{

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

									    }

									}

2.编写Filter

				?

									package com.xqw.kyg.filter;

									import com.xqw.kyg.util.MyHttpServletRequestWrapper;

									import org.springframework.stereotype.Component;

									import javax.servlet.*;

									import javax.servlet.http.HttpServletRequest;

									import java.io.IOException;

									@Component

									public class HttpServletRequestReplacedFilter implements Filter {

									    @Override

									    public void destroy() {}

									    @Override

									    public void doFilter(ServletRequest request, ServletResponse response,

									                         FilterChain chain) throws IOException, ServletException {

									        ServletRequest requestWrapper = null;

									        if(request instanceof HttpServletRequest) {

									            requestWrapper = new MyHttpServletRequestWrapper((HttpServletRequest) request);

									        }

									        if(requestWrapper == null) {

									            chain.doFilter(request, response);

									        } else {

									            chain.doFilter(requestWrapper, response);

									        }

									    }

									    @Override

									    public void init(FilterConfig arg0) throws ServletException {}

									}