获取@RequestBody请求的参数信息报错:java.io.IOException Stream closed

问题描述

项目中采用spring aop进行日志记录,在切面类通知方法中编写日志逻辑时,需要获取 HttpSevletRequest 中的请求参数;对于普通参数来说,没有任何问题,但是当请求方式为 POST/PUT 并并且是 @RequestBody 标记的请求,在获取JSON参数时,会出现 java.io.IOException: Stream closed 异常;

原因

HttpServletReqeust获取输入流时仅允许读取一次,spring已经对 @ReqeustBody 提前进行了处理,通过断点调试发现,aop代码中获取 request.getInputStream() 时,输入流已经关闭,因此出现流已经关闭的异常;

异常信息:

java.io.IOException: Stream closed
at org.apache.catalina.connector.InputBuffer.read(InputBuffer.java:372)
at org.apache.catalina.connector.CoyoteInputStream.read(CoyoteInputStream.java:190)
at sun.nio.cs.StreamDecoder.readBytes(StreamDecoder.java:284)
at sun.nio.cs.StreamDecoder.implRead(StreamDecoder.java:326)
at sun.nio.cs.StreamDecoder.read(StreamDecoder.java:178)
at java.io.InputStreamReader.read(InputStreamReader.java:184)
at java.io.BufferedReader.fill(BufferedReader.java:161)
at java.io.BufferedReader.readLine(BufferedReader.java:324)
at java.io.BufferedReader.readLine(BufferedReader.java:389)
at com.dongzz.cms.common.utils.WebUtil.getParamsJson(WebUtil.java:151)
at com.dongzz.cms.common.annotation.aspect.LogHandlerAdvice.recordSysLogs(LogHandlerAdvice.java:83)
at com.dongzz.cms.common.annotation.aspect.LogHandlerAdvice.handleSysLogs(LogHandlerAdvice.java:44)

解决思路

重新构建 ServletRequest,读取输入流后进行缓存,然后重写进流里面,使请求输入流支持二次读取;

  1. 自定义 HttpServletRequestWrapper,重新构建请求对象;

    package com.dongzz.cms.common.filter;
    
    import javax.servlet.ServletInputStream;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletRequestWrapper;
    import java.io.*;
    import java.nio.charset.StandardCharsets;
    
    /**
     * 重新构建 ServletRequest,读取输入流后进行缓存,然后重写进流里面,使请求输入流支持二次读取
     */
    public class RequestWrapper extends HttpServletRequestWrapper {
    
        private final byte[] body;
    
        public RequestWrapper(HttpServletRequest request) throws IOException {
            super(request);
            body = read(request.getInputStream()).getBytes(StandardCharsets.UTF_8);
        }
    
        @Override
        public BufferedReader getReader() throws IOException {
            return new BufferedReader(new InputStreamReader(getInputStream()));
    
        }
    
        @Override
        public ServletInputStream getInputStream() throws IOException {
            ByteArrayInputStream bis = new ByteArrayInputStream(body);
            return new ServletInputStream() {
    
                @Override
                public int read() throws IOException {
                    return bis.read();
                }
    
            };
    
        }
    
        /**
         * 读取输入流中的参数,转化为字符串
         *
         * @param is
         * @return
         */
        private String read(InputStream is) {
            StringBuilder sb = new StringBuilder();
            BufferedReader reader = new BufferedReader(new InputStreamReader(is, StandardCharsets.UTF_8));
            try {
                String line;
                while ((line = reader.readLine()) != null) {
                    sb.append(line);
                }
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                if (null != reader) {
                    try {
                        reader.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }
            return sb.toString();
        }
    }
    
    
  2. 自定义过滤器,用自定义的 ServletRequest 处理 @RequestBody 特殊请求;

    package com.dongzz.cms.common.filter;
    
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.http.HttpMethod;
    import org.springframework.http.MediaType;
    
    import javax.servlet.*;
    import javax.servlet.annotation.WebFilter;
    import javax.servlet.http.HttpServletRequest;
    import java.io.IOException;
    
    /**
     * 解决读取 @RequestBody 相关参数时,HttpServletRequest的输入流只能读取一次的问题
     * 问题表现: java.io.IOException: Stream closed
     */
    @WebFilter(
            filterName = "bodyFilter",
            urlPatterns = "/*"
    )
    public class RequestBodyFilter implements Filter {
    
        public static final Logger logger = LoggerFactory.getLogger(RequestBodyFilter.class);
    
        @Override
        public void init(FilterConfig config) throws ServletException {
            logger.debug("init request boy filter.");
        }
    
        @Override
        public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain chain) throws IOException, ServletException {
            ServletRequest requestWrapper = null;
    
            if (servletRequest instanceof HttpServletRequest) {
                HttpServletRequest request = (HttpServletRequest) servletRequest;
                String mehtod = request.getMethod();
                String contentType = request.getContentType();
                // 特殊处理 POST/PUT请求,忽略上传请求
                if ((HttpMethod.POST.name().equals(mehtod) || HttpMethod.PUT.name().equals(mehtod)) && (!MediaType.MULTIPART_FORM_DATA_VALUE.equals(contentType))) {
                    requestWrapper = new RequestWrapper(request); // 重新构建请求,新的对象读取输入流后进行缓存,然后重写进流,因此支持二次读取
                }
    
            }
    
            if (null == requestWrapper) {
                chain.doFilter(servletRequest, servletResponse);
            } else {
                chain.doFilter(requestWrapper, servletResponse);
            }
        }
    
        @Override
        public void destroy() {
            logger.debug("destroy request body filter.");
        }
    }
    
    

此时在 aop 通知方法中通过 HttpServletRequest 获取 @RequestBody 参数时能够取得相关的参数值,不会出现流已关闭的异常;

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

暴走编程

您的鼓励将是我创作最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值