POST请求获取body里面参数工具类,解决request.getInputStream()只能读取一次的问题

AuLuo-

已于 2025-01-02 16:03:48 修改

阅读量7.0k

点赞数 1

CC 4.0 BY-SA版权

文章标签： servlet java 开发语言

于 2019-06-25 15:47:50 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_39184713/article/details/93624012

本文介绍了一种通过自定义RequestWrapper类来读取和处理HTTP POST请求中的body参数的方法，适用于需要深入操作请求数据的场景，如权限校验等。

获取post请求中的参数

public class RequestWrapper extends HttpServletRequestWrapper{
    private final String body;
    public RequestWrapper(HttpServletRequest request) {
        super(request);
        StringBuilder stringBuilder = new StringBuilder();
        BufferedReader bufferedReader = null;
        InputStream inputStream = null;
        try {
            inputStream = request.getInputStream();
            if (inputStream != null) {
                bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
                char[] charBuffer = new char[128];
                int bytesRead = -1;
                while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
                    stringBuilder.append(charBuffer, 0, bytesRead);
                }
            } else {
                stringBuilder.append("");
            }
        } catch (IOException ex) {

        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (bufferedReader != null) {
                try {
                    bufferedReader.close();
                }
                catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        this.body = stringBuilder.toString();
    }
    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
        ServletInputStream servletInputStream = new ServletInputStream() {
            @Override
            public boolean isFinished() {
                return false;
            }
            @Override
            public boolean isReady() {
                return false;
            }
            @Override
            public void setReadListener(ReadListener readListener) {
            }
            @Override
            public int read() throws IOException {
                return byteArrayInputStream.read();
            }
        };
        return servletInputStream;

    }

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

    public String getBody() {
        return this.body;
    }
}

zull路由需要在ZuulFilter中做权限校检

在鉴权完成后，需要set参数传递如下
RequestContext ctx= RequestContext.getCurrentContext();
 // 一定要get一下,requestQueryParams才能取到值
  req.getParameterMap();
  Map<String, List<String>> requestQueryParams = ctx.getRequestQueryParams();
  ArrayList<String> paramsList = new ArrayList<>();
  requestQueryParams.put("XXX", paramsList);
  ctx.setRequestQueryParams(requestQueryParams);
  
POST需要在过滤其中获取body中的参数如下
   RequestContext ctx= RequestContext.getCurrentContext();
   HttpServletRequest req=ctx.getRequest();
   RequestWrapper requestWrapper = new RequestWrapper(req);
   String body = requestWrapper.getBody();