解决request.getInputStream()只能读取一次的问题

上海好程序员

已于 2024-02-27 16:21:34 修改

阅读量295

点赞数 1

文章标签： java

于 2024-02-27 16:16:54 首次发布

本文链接：https://blog.csdn.net/fromfire2/article/details/136325420

版权

文章介绍了如何使用Lombok创建一个名为NewRequest的类，扩展HttpServletRequest以支持在Content-Type为application/json时重载输入流，以及如何在RequestFilter中应用此新请求处理方式。

摘要由CSDN通过智能技术生成

class NewRequest:

import lombok.SneakyThrows;
import lombok.var;

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.InputStreamReader;

public class NewRequest extends HttpServletRequestWrapper {

    byte[] body;

    @SneakyThrows
    public NewRequest(HttpServletRequest request) {
        super(request);
        body = BinaryUtil.getBytes(request.getInputStream());
    }

    @Override
    public ServletInputStream getInputStream() {
        var input = new ByteArrayInputStream(body);
        return new ServletInputStream() {
            @Override
            public int read() {
                return input.read();
            }

            @Override
            public boolean isFinished() {
                return input.available() == 0;
            }

            @Override
            public boolean isReady() {
                return input.available() == 1;
            }

            @Override
            public void setReadListener(ReadListener readListener) {
            }
        };
    }

    @Override
    public BufferedReader getReader() {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }

}

class RequestFilter:

import org.springframework.context.annotation.Configuration;
import org.springframework.web.filter.OncePerRequestFilter;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

@Configuration
public class RequestFilter extends OncePerRequestFilter {

    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain)
            throws ServletException, IOException {
        if (StringUtil.contains(request.getContentType(), "application/json"))
            request = new NewRequest(request);//异常日志需要, json要支持重复获取
        chain.doFilter(request, response);
    }

}

放任意位置即可, 只替换了contentType包含application/json的请求