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

问题场景

有一个传递json参数的post接口,笔者增加了一个拦截器。在拦截器中需要获取参数然后进行一些操作。这里的获取是通过流来获取的参数。测试都ok,发上去以后就报了这个错误:Cannot call getInputStream(), 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);
    }
}

总结

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

猜你喜欢

转载自blog.csdn.net/m0_46897923/article/details/128417442
今日推荐