解决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);
    }
}

总结

---

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