封装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()));
    }

}

<think>嗯,用户想了解如何通过继承HttpServletRequestWrapper来实现多次读取HTTP请求体。首先,我需要回忆一下HttpServletRequestWrapper的作用。它是javax.servlet包中的一个类,用于封装HttpServletRequest对象,允许我们覆盖其中的方法。通常,请求体(request body)只能被读取一次,因为输入流一旦被读取,指针就到了末尾,无法重置。这在需要多次处理请求体内容时会有问题,比如日志记录、参数解析或验证等场景。 用户提到的引用内容显示,可以通过继承HttpServletRequestWrapper,将请求体内容缓存到一个字节数组中,这样后续就可以多次读取。接下来需要思考如何具体实现这一点。可能需要覆盖getInputStream()和getReader()方法,使得每次调用时都返回一个基于缓存数据的流,而不是原始的流。 然后,示例代码的结构应该包括一个自定义的HttpServletRequestWrapper子类,比如CachedBodyHttpServletRequestWrapper。在这个类中,需要将原始的InputStream读取并保存到字节数组或ByteArrayInputStream中。这样,当多次调用getInputStream()时,每次都返回新的ByteArrayInputStream,从而支持多次读取。 另外,需要使用过滤器(Filter)来包装原始的HttpServletRequest对象,将其替换为自定义的Wrapper实例。这样,在后续的处理链中,其他组件(如控制器或拦截器)就可以通过这个Wrapper来多次获取请求体。 需要注意的地方包括正确处理字符编码,确保读取数据时不会出现乱码。此外,缓存数据可能会对内存产生影响,特别是处理大文件上传时,需要考虑内存限制,可能需要使用临时文件或其他流式处理方式,但用户的问题可能集中在一般情况下的多次读取,所以示例中使用字节数组是合适的。 用户可能需要具体的代码示例,因此需要展示如何编写Wrapper类以及如何配置过滤器。同时,要提醒用户注意在过滤器中正确传递包装后的请求对象,否则后续处理可能无法获取到缓存的body。 最后,验证示例代码的正确性,确保覆盖了所有必要的方法,如getInputStream()和getReader(),并且缓存逻辑正确无误。</think>### 通过 HttpServletRequestWrapper 实现请求体多次读取的步骤 #### 1. 创建自定义包装类 继承 `HttpServletRequestWrapper`,覆盖输入流方法,缓存请求体数据: ```java public class CachedBodyHttpServletRequestWrapper extends HttpServletRequestWrapper { private final byte[] cachedBody; public CachedBodyHttpServletRequestWrapper(HttpServletRequest request) throws IOException { super(request); InputStream requestInputStream = request.getInputStream(); this.cachedBody = IOUtils.toByteArray(requestInputStream); // 使用 Apache Commons IO } @Override public ServletInputStream getInputStream() { return new CachedBodyServletInputStream(cachedBody); } @Override public BufferedReader getReader() { ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(cachedBody); return new BufferedReader(new InputStreamReader(byteArrayInputStream, getCharacterEncoding())); } } ``` #### 2. 实现自定义 ServletInputStream ```java public class CachedBodyServletInputStream extends ServletInputStream { private final ByteArrayInputStream buffer; public CachedBodyServletInputStream(byte[] body) { this.buffer = new ByteArrayInputStream(body); } @Override public int read() { return buffer.read(); } @Override public boolean isFinished() { return buffer.available() == 0; } @Override public boolean isReady() { return true; } @Override public void setReadListener(ReadListener listener) { throw new UnsupportedOperationException("Not implemented"); } } ``` #### 3. 创建过滤器配置 ```java @WebFilter("/*") public class CachingRequestBodyFilter implements Filter { @Override public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { HttpServletRequest httpRequest = (HttpServletRequest) request; CachedBodyHttpServletRequestWrapper wrappedRequest = new CachedBodyHttpServletRequestWrapper(httpRequest); chain.doFilter(wrappedRequest, response); // 关键:传递包装后的请求 } } ``` ### 技术原理说明 1. **缓存机制**:将原始请求的 `InputStream` 转换为字节数组缓存,后续读取时始终返回基于缓存的流[^3] 2. **流重置**:通过 `ByteArrayInputStream` 实现多次读取能力 3. **过滤器拦截**:在请求处理链最前端完成请求体缓存[^2] ### 使用示例 ```java @PostMapping("/demo") public void processRequest(HttpServletRequest request) { // 第一次读取 String body1 = IOUtils.toString(request.getInputStream(), "UTF-8"); // 第二次读取(正常执行) String body2 = IOUtils.toString(request.getInputStream(), "UTF-8"); } ``` ### 注意事项 1. **性能影响**:大文件请求会导致内存压力,建议限制最大缓存大小 2. **编码兼容**:需正确处理请求的字符编码(通过 `getCharacterEncoding()`) 3. **依赖管理**:需要 Apache Commons IO 或类似工具处理流转换
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值