HttpServletRequestWrapper记录

最新推荐文章于 2024-07-24 17:27:32 发布

qi23118

最新推荐文章于 2024-07-24 17:27:32 发布

阅读量136

点赞数

文章标签： java

本文链接：https://blog.csdn.net/qi23118/article/details/126367576

版权

HttpServletRequestWrapper 请求体缓存字节流 ServletInputStream

关键词由CSDN通过智能技术生成

public class WrapperRequest extends HttpServletRequestWrapper {

	private byte[] requestBody = null;// 用于将流保存下来
	private InputStream is = null;
	private HttpServletRequest request = null;

	public WrapperRequest(HttpServletRequest request) throws IOException {
		super(request);
		this.request = request;
	}

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

	@Override
	public ServletInputStream getInputStream() throws IOException {

		final ByteArrayInputStream bais = new ByteArrayInputStream(requestBody);

		return new ServletInputStream() {

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

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

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

			@Override
			public void setReadListener(ReadListener readListener) {

			}
		};
	}

    public byte[] getRequestBody() {
		try {
			is = this.request.getInputStream();
			String sessionStream = StreamUtils.copyToString(is, Charset.forName("UTF-8"));
			requestBody = sessionStream.getBytes("UTF-8");
		} catch (IOException e) {
			e.printStackTrace();
		}

        return requestBody;
    }
}