多次读取Request数据流

最新推荐文章于 2024-02-22 16:24:52 发布

会飞的草帽1

最新推荐文章于 2024-02-22 16:24:52 发布

阅读量3k

点赞数

CC 4.0 BY-SA版权

分类专栏：服务器相关

本文链接：https://blog.youkuaiyun.com/qq_31301681/article/details/84140128

服务器相关专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种通过重写HttpServletRequestWrapper的getReader()和getInputStream()方法，实现多次读取HTTP请求中携带的数据流的技术。通过将输入流转换为字节数组并保存，再将字节数组转换回输入流，使得可以多次读取同一数据流，适用于需要多次解析请求数据的场景。

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

1.需要多次读取Request中携带的数据流，但是默认的只能读取一次；

2.如果需要多次读取Request中携带的数据流，需要重写 HttpServletRequestWrapper 的getReader() 和 getInputStream()方法；

3.代码:


import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

public class InputStreamReadRepeatableRequestWrapper extends HttpServletRequestWrapper {
	
	private final byte[] body; // 报文

	public InputStreamReadRepeatableRequestWrapper(HttpServletRequest request) throws IOException {
		super(request);
		body = InputStreamToByte(request.getInputStream());
	}
	
	/**
	 * 流转 字节数组 
	 * @param is
	 * @return
	 * @throws IOException
	 */
	private byte[] InputStreamToByte(InputStream is) throws IOException {
	   ByteArrayOutputStream bytestream = new ByteArrayOutputStream();
	   byte[] buffer=new byte[1024];
	   int ch;
	   while ((ch = is.read(buffer)) != -1) {
		   bytestream.write(buffer,0,ch);
	   }
	   byte data[] = bytestream.toByteArray();
	   bytestream.close();
	   return data;
	 }
	
	@Override
	public BufferedReader getReader() throws IOException {
		return new BufferedReader(new InputStreamReader(getInputStream()));
	}
	
	@Override
	public ServletInputStream getInputStream() throws IOException {
		final ByteArrayInputStream bais = new ByteArrayInputStream(body);
		return new ServletInputStream() {
			
			@Override
			public int read() throws IOException {
				return bais.read();
			}
		};
	}

}