解决request.getInputStream()只能读取一次的方案

 最近做的一个项目中,需要在拦截器中校验参数是否被篡改,所以需要在json字符串反序列化前,提前获取request中的字节数据,但这样会在后续反序列化时报错,原因是request.getInputStream()只能被调用一次,解决办法就是在过滤器中对request请求进行包装,创建一个body[]数组,将流中数据读出保存到body中,以便后续可以反复读取,在重写getInputStream()方法,下附解决步骤:

1.创建request请求包装类的对象

//request请求的包装类
public class RequestWrapper extends HttpServletRequestWrapper {
    String default_charset = "utf-8";

    private final byte[] body;

    public RequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        ServletInputStream inputStream = request.getInputStream();
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        byte[] buf = new byte[1024];
        int len=0;
        while ((len = inputStream.read(buf))!=-1){
            out.write(buf,0,len);
        }
        out.flush();
        this.body = out.toByteArray();
        out.close();
    }

    public byte[] getBody() {
        return body;
    }

    public String getBodyString() {
       return new String(body, Charset.forName(default_charset));
    }

    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }

    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream inputStream = new ByteArrayInputStream(this.body);
        return new ServletInputStream() {
            public int read() throws IOException {
                return inputStream.read();
            }

            public boolean isFinished() {
                return false;
            }

            public boolean isReady() {
                return false;
            }
            public void setReadListener(ReadListener readListener) {}
        };
    }

}

2.在过滤器中进行用自己创建的包装类对象替换

public class ReplaceStreamFilter implements Filter {
    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
    }
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        RequestWrapper wrapper = new RequestWrapper((HttpServletRequest)request);
        log.info("request请求对象被替换。。。。。。。。。。。。。。。。。。");
        chain.doFilter(wrapper,response);
    }
    @Override
    public void destroy() {
    }
}

3.在后续校验中直接获取即可

Java中,request.getInputStream()方法可以用来读取HTTP请求中的请求体内容。如果请求体中是一个文件,可以通过该方法获取文件的二进制数据并进行处理。下面是一个示例代码片段,可以从HTTP请求中读取文件内容并将其保存到本地磁盘上: ```java import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; public class FileUploadServlet extends HttpServlet { @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { InputStream inputStream = request.getInputStream(); String saveFilePath = "/path/to/save/uploaded/file"; OutputStream outputStream = new FileOutputStream(saveFilePath); int bytesRead = -1; byte[] buffer = new byte[4096]; while ((bytesRead = inputStream.read(buffer)) != -1) { outputStream.write(buffer, 0, bytesRead); } outputStream.close(); inputStream.close(); System.out.println("文件已成功上传!"); } } ``` 在这个示例中,我们在Servlet中实现了doPost()方法,用于处理POST请求。该方法调用request.getInputStream()方法来获取HTTP请求的输入流,然后使用Java IO类将其写入本地文件中。可以通过修改saveFilePath变量来指定保存文件的路径。最后,我们关闭输入流和输出流,并打印一条消息,指示文件已成功上传。 需要注意的是,在处理文件上传时,还需要在HTTP请求头中设置Content-Type属性为multipart/form-data,以便服务器能够正确地解析请求体。否则,请求体内容可能会被服务器解析为普通的文本数据。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值