SpringMVC自定义HttpServletRequestWrapper实现 body内容重复读取时,form-data格式读取失败 @RequestParam

场景

过滤器中需要对Controller中接口请求参数处理,getInputStream()方法只能调用一次,所以需要自定义HttpServletRequest

使用继承 HttpServletRequestWrapper 类的方法,读取body内容并存储,在Filter中将使用自定义的HttpServletRequestWrapper替换。

问题:

application/json格式没有问题,form-data的时候,使用@RequestParam获取的参数会提示 参数xx不存在

解决

form-data解析参数时,会判断有没有调用过getInputStream, 如果先调用了不会再重复获取。

在包装类中,调用getInputStream之前调用一下request.getParameterMap(); 会先解析body内容并将处理结果缓存起来,再调用自己的getInputStream方法
伪代码如下:

public class MockHttpServletRequestWrapper extends HttpServletRequestWrapper {
    private final byte[] body;
    
    public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        // 不用存储结果
        request.getParameterMap();
        // form-data时, 返回空字符串, 如果使用时需处理
        body = StreamUtils.copyToByteArray(request.getInputStream());
    }
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {

        final ByteArrayInputStream in = new ByteArrayInputStream(body);

        return new ServletInputStream() {

            @Override
            public int read() throws IOException {
                return in.read();
            }

            @Override
            public boolean isFinished() {
                try {
                    return bais.available() == 0;
                } catch (Exception e) {
                    // log here
                }
                return false;
            }
            @Override
            public boolean isReady() {
                return true;
            }
            @Override
            public void setReadListener(ReadListener readListener) {
            }
        };
    }
}

end


写法有误call call call 我

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值