问题场景
有一个传递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); } }
总结
问题解决其实不难,难点在于知识面能不能覆盖到包装类这块,当然了技术都是在解决问题中进度的,在此记录,加深印象。