如何在 Java 中捕获并记录 HTTP Servlet 响应内容？

如何捕获和记录 HTTP Servlet 响应内容

简介

记录请求和响应内容对于全面的服务器端调试和监控至关重要。虽然通过过滤器访问请求参数很简单，但检索响应内容可能更具挑战性。本文介绍如何创建一个过滤器来拦截和复制 servlet 响应以进行日志记录。

创建过滤器以捕获响应内容

要捕获响应内容，我们需要创建一个自定义 Filter 来包装 HttpServletResponse 对象并重写 getOutputStream() 和 getWriter() 方法。这些方法负责提供流来写入响应数据。通过覆盖它们，我们可以创建一个自定义流来拦截数据并将其复制到缓冲区中。

以下是完成此任务的过滤器示例：

@WebFilter("/*")
public class ResponseLogger implements Filter {

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) {
        HttpServletResponseCopier responseCopier = new HttpServletResponseCopier((HttpServletResponse) response);

        try {
            chain.doFilter(request, responseCopier);
            responseCopier.flushBuffer();
        } finally {
            byte[] copy = responseCopier.getCopy();
            System.out.println(new String(copy, response.getCharacterEncoding()));
        }
    }
}

自定义 Response Wrapper

HttpServletResponseCopier 类包装原始 HttpServletResponse 并创建一个拦截数据的自定义 ServletOutputStream。重写的 getOutputStream() 和 getWriter() 方法创建并返回 ServletOutputStreamCopier 实例。

public class HttpServletResponseCopier extends HttpServletResponseWrapper {

    private ServletOutputStreamCopier copier;

    @Override
    public ServletOutputStream getOutputStream() {
        if (copier == null) {
            copier = new ServletOutputStreamCopier(getResponse().getOutputStream());
        }
        return copier;
    }

    @Override
    public PrintWriter getWriter() {
        if (copier == null) {
            copier = new ServletOutputStreamCopier(getResponse().getOutputStream());
        }
        return new PrintWriter(new OutputStreamWriter(copier, getResponse().getCharacterEncoding()), true);
    }
}

拦截和复制响应数据

ServletOutputStreamCopier 类重写 write( int b) 方法拦截写入响应的数据。它将数据写入原始流，同时将其复制到 ByteArrayOutputStream 中以供以后检索。

public class ServletOutputStreamCopier extends ServletOutputStream {

    private ByteArrayOutputStream copy;

    @Override
    public void write(int b) {
        outputStream.write(b);
        copy.write(b);
    }

    public byte[] getCopy() {
        return copy.toByteArray();
    }
}

检索捕获的响应

请求-响应周期之后完成后，过滤器从 ServletOutputStreamCopier 的 getCopy() 方法检索捕获的响应内容。然后，该数据可用于记录或任何其他所需目的。

以上是如何在 Java 中捕获并记录 HTTP Servlet 响应内容？的详细内容。更多信息请关注PHP中文网其他相关文章！