多次读取请求request里数据

最新推荐文章于 2023-09-19 08:43:42 发布

活在戏中吧

最新推荐文章于 2023-09-19 08:43:42 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： Java 学

本文链接：https://blog.youkuaiyun.com/qq_21856521/article/details/89315027

学同时被 2 个专栏收录

39 篇文章

订阅专栏

Java

25 篇文章

订阅专栏

本文介绍了一种在WebService中解决重复读取POST请求数据的问题方法。通过使用HttpServletRequestWrapper类重写相关方法，使得filter和controller都能正确读取content-type为“text/xml”的数据，避免了数据流只能被读取一次的限制。

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

问题：
webService通信时，在filter中拦截请求做用户权限验证。拦截到content-type的内容是“text/xml”，只能用传入的数据流中取数据了，而request数据流里的数据只能读取一次，filter里读取了，到了controller里就会读取失败。
解决方法：

采用ThreadLocal，在filter里把读取到的post参数存入ThreadLocal里，而controller就可以再从ThreadLocal里把请求参数读取出来
使用servlet提供的HttpServletRequestWrapper类，重写相关ServletRequest方法，实现多次读取的能力。

这里只记录重写HttpServletRequestWrapper的方法。

import org.apache.commons.io.IOUtils;
 
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;
 
/**
 * Created by Endstart on 16/11/30.
 */
public class WrappedHttpServletRequest extends HttpServletRequestWrapper {
 
    private byte[] bytes;
    private WrappedServletInputStream wrappedServletInputStream;
 
    public WrappedHttpServletRequest(HttpServletRequest request) throws IOException {
        super(request);
        // 读取输入流里的请求参数，并保存到bytes里
        bytes = IOUtils.toByteArray(request.getInputStream());
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
        this.wrappedServletInputStream = new WrappedServletInputStream(byteArrayInputStream);

　　　　　// 很重要，把post参数重新写入请求流
　　　　　reWriteInputStream();
 
    }
 
    /**
     * 把参数重新写进请求里
     */
    public void reWriteInputStream() {
        wrappedServletInputStream.setStream(new ByteArrayInputStream(bytes != null ? bytes : new byte[0]));
    }
 
    @Override
    public ServletInputStream getInputStream() throws IOException {
        return wrappedServletInputStream;
    }
 
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(wrappedServletInputStream));
    }
 
    /**
     * 获取post参数，可以自己再转为相应格式
     */
    public String getRequestParams() throws IOException {
        return new String(bytes, this.getCharacterEncoding());
    }
 
    private class WrappedServletInputStream extends ServletInputStream {
 
        public void setStream(InputStream stream) {
            this.stream = stream;
        }
 
        private InputStream stream;
 
        public WrappedServletInputStream(InputStream stream) {
            this.stream = stream;
        }
 
        @Override
        public int read() throws IOException {
            return stream.read();
        }
 
        @Override
        public boolean isFinished() {
            return true;
        }
 
        @Override
        public boolean isReady() {
            return true;
        }
 
        @Override
        public void setReadListener(ReadListener readListener) {
 
        }
    }
}

filter中使用重写的HttpServletRequestWrapper来获取数据(需要根据自己的实际情况判断哪些请求需要拦截)

WrappedHttpServletRequest requestWrapper = new WrappedHttpServletRequest((HttpServletRequest) servletRequest);
String params = requestWrapper.getRequestParams();

当然放行的时候，参数也要改写为新的requestWrapper

filterChain.doFilter(requestWrapper, servletResponse);

参考文章