拦截请求(request)和响应(response)信息

最新推荐文章于 2024-08-12 14:19:51 发布

gitlee

最新推荐文章于 2024-08-12 14:19:51 发布

阅读量1.1w

点赞数 2

分类专栏： java SpringBoot 文章标签： response filter request interceptor

本文链接：https://blog.csdn.net/llll20000/article/details/80576924

版权

SpringBoot 同时被 2 个专栏收录

5 篇文章 0 订阅

订阅专栏

java

2 篇文章 0 订阅

订阅专栏

思路

使用拦截器；
使用Filter；

实现

使用拦截器时不能获取到返回的信息(response里的content为空)；
使用Filter，成功获取到内容：

@WebFilter(urlPatterns = "/*")
public class GlobalFilter implements Filter {
    private Logger logger = LoggerFactory.getLogger(GlobalFilter.class);

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

    }

    /**
     * 拦截请求和相应信息，并保存到数据库中
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        WkResponseWrapper wrapper = new WkResponseWrapper((HttpServletResponse)response);
        ContentCachingRequestWrapper requestWrapper = new ContentCachingRequestWrapper((HttpServletRequest) request);
        HttpServletRequest req = (HttpServletRequest) request;
        String params = "";

        try {

            chain.doFilter(requestWrapper, wrapper);
            String responseContent = new String(wrapper.getDataStream());

            String method = req.getMethod();
            if ("POST".equalsIgnoreCase(method)) {
                params = new String(requestWrapper.getContentAsByteArray());
            } else if ("GET".equalsIgnoreCase(method)) {
                params = JSON.toJSONString(request.getParameterMap());
            }

            response.getWriter().write(responseContent);


        } catch (Exception e) {
            logger.error("保存请求和相应信息时出现异常：", e);
        }
    }

    @Override
    public void destroy() {

    }


}

public class WkResponseWrapper  extends HttpServletResponseWrapper {
    ByteArrayOutputStream output;
    FilterServletOutputStream filterOutput;

    public WkResponseWrapper(HttpServletResponse response) {
        super(response);
        output = new ByteArrayOutputStream();
    }

    @Override
    public ServletOutputStream getOutputStream() throws IOException {
        if (filterOutput == null) {
            filterOutput = new FilterServletOutputStream(output);
        }
        return filterOutput;
    }

    public byte[] getDataStream() {
        return output.toByteArray();
    }


    class FilterServletOutputStream extends ServletOutputStream {
        DataOutputStream output;

        public FilterServletOutputStream(OutputStream output) {
            this.output = new DataOutputStream(output);
        }

        @Override
        public void write(int arg0) throws IOException {
            output.write(arg0);
        }

        @Override
        public void write(byte[] arg0, int arg1, int arg2) throws IOException {
            output.write(arg0, arg1, arg2);
        }

        @Override
        public void write(byte[] arg0) throws IOException {
            output.write(arg0);
        }

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

        @Override
        public void setWriteListener(WriteListener writeListener) {

        }
    }
}