springboot 获取body内数据并赋值给servlet, 同时保留body数据

最新推荐文章于 2024-09-01 21:49:00 发布

zzy_阳阳

最新推荐文章于 2024-09-01 21:49:00 发布

阅读量5.6k

点赞数 1

分类专栏： SpringBoot 遇到问题

本文链接：https://blog.csdn.net/ZZY_Yang/article/details/80346012

版权

本文介绍如何在SpringBoot应用中使用过滤器（Filter）捕获HTTP请求体的数据，并通过自定义的`ParameterRequestWrapper`类重新封装请求，以确保请求体在后续处理中仍可读。配置过滤器的方法包括直接在启动类或@Configuration注解的类中添加代码。

摘要由CSDN通过智能技术生成

在此我们通过过滤器来完成本功能

@WebFilter(filterName = "RequestBodyFilter", urlPatterns = "/**")
public class RequestBodyFilter implements Filter {
    public static final String LOGTHREAD_ID="logthreadId";
    private static final Logger LOGGER = LoggerFactory.getLogger(RequestBodyFilter.class);

    public static final String URI_ENCODING = NsConfiguration.getInstance()
            .getConf(NsConfiguration.KEY_URI_ENCODING);

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        //日志打印唯一标识
        MDC.put(LOGTHREAD_ID,java.util.UUID.randomUUID().toString().replace("-",""));

        HttpServletRequest request = (HttpServletRequest) servletRequest;
        HttpServletResponse response = (HttpServletResponse) servletResponse;
        request.setCharacterEncoding("UTF-8");
        //获取body数据
        BufferedReader reader = new BufferedReader(new InputStreamReader(request.getInputStream()));
        String line = null;
        StringBuilder sb = new StringBuilder();
        while((line = reader.readLine())!=null){
            sb.append(line);
        }
        String postContent = sb.toString();

        //如果body中存在数据放入HttpServletRequest
        if (!RequestBodyFilter.isEmpty(postContent) ){
            request = new ParameterRequestWrapper(request);
            LOGGER.debug(postContent);
            String[] params = postContent.split("&");
            for (String param : params) {
                String[] pkv = param.split("=");
                String data = URLDecoder.decode(pkv[1], URI_ENCODING);
                ((ParameterRequestWrapper)request).addParameter(pkv[0],data);
            }
        }
        filterChain.doFilter(request,response);

    }

    @Override
    public void destroy() {

    }

    public static boolean isEmpty(Object str) {
        return str == null || "".equals(str) || "{}".equals(str);
    }