封装Request,提供多次读取数据

本文介绍了一个自定义的HTTPServletRequestWrapper类,该类继承自HttpServletRequestWrapper,主要用于解决ServletInputStream只能被读取一次的问题。通过缓存请求体的内容,使得可以在同一个请求中多次读取输入流。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;

/**
 * 封装Request,提供InputStream重复读取支持
 */
public class HTTPServletRequestWrapper extends HttpServletRequestWrapper {

    private final String body;

    public HTTPServletRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);

        StringBuilder stringBuilder = new StringBuilder();

        InputStream inputStream = request.getInputStream();

        if (inputStream != null) {

            BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));

            char[] chunk = new char[128];

            int bytesRead;

            while ((bytesRead = bufferedReader.read(chunk)) > 0) {
                stringBuilder.append(chunk, 0, bytesRead);
            }

        } else {
            stringBuilder.append("");
        }

        body = stringBuilder.toString();

    }

    /**
     * 重新getInputStream,缓存Body数据
     *
     * @return
     * @throws IOException
     */
    @Override
    public ServletInputStream getInputStream() throws IOException {

        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());

        ServletInputStream servletInputStream = new ServletInputStream() {

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

            @Override
            public boolean isFinished() {
                return false;
            }

            @Override
            public boolean isReady() {
                return false;
            }

            @Override
            public void setReadListener(ReadListener listener) {

            }

        };

        return servletInputStream;

    }

    /**
     * 读取缓存数据
     *
     * @return
     * @throws IOException
     */
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(this.getInputStream()));
    }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值