如何在 Java 中擷取並記錄 HTTP Servlet 回應內容？

如何擷取並記錄HTTP Servlet 回應內容

簡介

記錄與回應內容對於全面的伺服器端調試和監控至關重要。雖然透過篩選器存取請求參數很簡單，但檢索回應內容可能更具挑戰性。本文介紹如何建立一個過濾器來攔截和複製 servlet 回應以進行日誌記錄。

建立篩選器以擷取回應內容

要擷取回應內容，我們需要建立一個自訂Filter 來包裝HttpServletResponse 物件並重寫getWriter() 和OutgetWriter( ) 方法。這些方法負責提供流來寫入回應資料。透過覆蓋它們，我們可以建立一個自訂流來攔截資料並將其複製到緩衝區中。

以下是完成此任務的過濾器範例：

@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 類包裝了原來的類一個攔截資料的自訂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中文網其他相關文章！