解决HttpServletRequest中的流无法重复读取的问题

简介: 解决HttpServletRequest中的流无法重复读取的问题

问题场景



有一个传递json参数的post接口,笔者增加了一个拦截器。在拦截器中需要获取参数然后进行一些操作。这里的获取是通过流来获取的参数。测试都ok,发上去以后就报了这个错误:Cannot callgetInputStream(), getReader() already called。看报错信息其实很明了,就是不能重复获取流。什么?没有重复获取?其实rest接口接收json参数(使用@RequestBody)也是通过获取流来实现的,其实获取json就这一个方法,就是通过流来获取。无论穿上什么外衣,在哪里你终究还是你。


问题思考



1.既然是我在拦截器中提前将流获取了,导致了接口中无法获取流,那能不能将流获取后再塞回去呢?

如果可以塞回去,那接口也就可以正常获取了。

经测试,无法正常塞回去,这个方向告吹(测试了requst的所有方法,没有能将流再塞回去的)。


(网上有位道友使用了类似的方法:request.setAttribute(data,json);,不过他的例子不是纯json数据,这个json外层还有一个data,其实和这个场景略有不同)


2.上网一通查,发现大家用的都是包装类,也就是将现有的http请求进行包装。怎么个包装法呢?既然流无法重复获取,那我就把流中的数据取出来自己存,自己定义的存储结构,那还不想怎么获取就怎么获取,也不会有获取次数限制了?是不是?


问题解决



所以笔者就新增了一个过滤器,然后在过滤器中使用包装类进行传递请求,这样就解决了问题,代码如下所示:


1.包装类代码


package cn.com.sunacwy.mdm.filter;
import lombok.extern.slf4j.Slf4j;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;
@Slf4j
public class MdmHttpServletRequestWrapper extends HttpServletRequestWrapper {
    private String body;
  //方便再拦截器中直接使用body,无需再从流中获取,当然从流中获取也是ok的
    public String getBody(){
        return this.body;
    }
    public MdmHttpServletRequestWrapper(HttpServletRequest request) {
        super(request);
        //将请求中的流取出来放到body里,后面都只操作body就行
        this.body = RequestReadUtils.read(request);
    }
    @Override
    public ServletInputStream getInputStream()  {
      //返回body的流信息即可
        try(final ByteArrayInputStream bais = new ByteArrayInputStream(body.getBytes());ServletInputStream servletInputStream = new ServletInputStream() {
            @Override
            public boolean isFinished() {
                return false;
            }
            @Override
            public boolean isReady() {
                return false;
            }
            @Override
            public void setReadListener(ReadListener readListener) {
            }
            @Override
            public int read() {
                return bais.read();
            }
        }){
            return servletInputStream;
        }catch(IOException e){
            log.info("#### MdmHttpServletRequestWrapper,发生异常,异常信息:{} ####",e.getMessage());
            return null;
        }
    }
    @Override
    public BufferedReader getReader(){
        return new BufferedReader(new InputStreamReader(this.getInputStream()));
    }
    static class RequestReadUtils {
        private static final int BUFFER_SIZE = 1024 * 8;
        public static String read(HttpServletRequest request) {
            try(BufferedReader bufferedReader = request.getReader();StringWriter writer = new StringWriter()){
                int read;
                char[] buf = new char[BUFFER_SIZE];
                while( ( read = bufferedReader.read(buf) ) != -1 ) {
                    writer.write(buf, 0, read);
                }
                return writer.getBuffer().toString();
            }catch (Exception e){
                log.info("#### MdmHttpServletRequestWrapper,发生异常,异常信息:{} ####",e.getMessage());
                return "";
            }
        }
    }
}


2.过滤器代码


@WebFilter(filterName = "RewriteQequestFilter", urlPatterns = "/mdmcustomer/commit")
@Order(1)
public class RewriteQequestFilter implements Filter {
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        MdmHttpServletRequestWrapper requestWrapper = new MdmHttpServletRequestWrapper((HttpServletRequest)request);
        chain.doFilter(requestWrapper,response);
    }
}


总结



问题解决其实不难,难点在于知识面能不能覆盖到包装类这块,当然了技术都是在解决问题中进度的,在此记录,加深印象。

相关文章
|
8月前
|
Java 程序员
Stream流操作-结果收集终止方法-最终篇
Stream流操作-结果收集终止方法-最终篇
78 0
|
10月前
|
存储
Stream流中各阶段方法说明及组合示例
Stream流中各阶段方法说明及组合示例
75 1
|
9月前
|
缓存
缓存流
缓存流
22 0
|
9月前
|
JSON Java Maven
自定义starter(实现HttpServletRequest重复读取)
自定义starter(实现HttpServletRequest重复读取)
90 0
|
10月前
|
存储 Java
流及其相关操作
流及其相关操作
|
12月前
HttpServletRequest共享数据、请求转发、请求包含、获取流对象及中文乱码问题
HttpServletRequest共享数据、请求转发、请求包含、获取流对象及中文乱码问题
59 0
|
JSON 缓存 前端开发
【并发技术系列】「Web请求读取系列」如何构建一个可重复读取的Request的流机制
【并发技术系列】「Web请求读取系列」如何构建一个可重复读取的Request的流机制
211 0
【并发技术系列】「Web请求读取系列」如何构建一个可重复读取的Request的流机制
|
SQL Java 程序员
Stream流的操作
Stream流的操作
56 0
|
存储 分布式计算 Java
记一次InputStream流读取不完整留下的惨痛教训
记一次InputStream流读取不完整留下的惨痛教训
284 0
记一次InputStream流读取不完整留下的惨痛教训
面试官:IO 操作必须要手动关闭吗?关闭流方法是否有顺序?
面试官:IO 操作必须要手动关闭吗?关闭流方法是否有顺序?
142 0