Java Servlet 过滤器 - 根据状态修改响应标头

Question

我正在尝试编写一个 servlet 过滤器，该过滤器将根据请求的状态将标头添加到响应中。我知道我必须在传递给chain.doFilter 之前用 HttpServletResponseWrapper 包装响应，但标头永远不会被发送，所以我显然遗漏了一些非常明显的东西。

代码看起来像：

@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException
{
   HttpServletResponse httpServletResponse = (HttpServletResponse) response;
   HttpServletResponseWrapper responseWrapper = new HttpServletResponseWrapper(httpServletResponse);

   chain.doFilter(request, responseWrapper);

   if(responseWrapper.getStatus() < 400)
   {
      responseWrapper.addHeader("X-Custom-Foobar", "abc");
   }
}

在检查完成之前，我必须在包装器中捕获一些内容以防止响应发送到客户端吗？

Answer 1

所以这个规范令人沮丧的部分是你必须完全拦截ServletOutputStream 并缓冲它。我最终在这里遵循了这个例子 :: https://stackoverflow.com/a/11027170/76343

基类HttpServletResponseWrapper 是一个完整的传递，一旦输出流关闭，对响应的所有进一步修改都将被静音。

不幸的是，似乎没有更优雅的方式来实现这一点。

Answer 2

您需要扩展 HttpResponseWrapper() 并覆盖适当的方法。仅仅使用香草HttpResponseWrapper 本身并没有任何作用。

Answer 3

这段代码的顺序颠倒了：

   chain.doFilter(request, responseWrapper);

   if(responseWrapper.getStatus() < 400)
   {
      responseWrapper.addHeader("X-Custom-Foobar", "abc");
   }

试试这个：

   if(responseWrapper.getStatus() < 400)
   {
      responseWrapper.addHeader("X-Custom-Foobar", "abc");
   }

   chain.doFilter(request, responseWrapper);

doFilter 方法直到在线发送响应后才会返回到您的方法。

Answer 4

这实际上是可能的。但是因为在调用chain.doFilter(request, response)之后响应已经提交了，所以我们必须在收到状态码之后，但是在响应提交之前设置headers。这是一个例子：

public class HeadersFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) {
    }

    @Override
    public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain) throws IOException,
                                                                                             ServletException {

        HttpServletRequest request = (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) resp;

        chain.doFilter(request, new ResponseWrapper(response));
    }

    public static class ResponseWrapper extends HttpServletResponseWrapper {
        public ResponseWrapper(HttpServletResponse response) {
            super(response);
        }

        @Override
        public void setStatus(int sc) {
            super.setStatus(sc);

            // SET YOUR HEADERS HERE
            // setHeader("Cache-Control", "no-cache, no-store, must-revalidate");
        }
    }

    @Override
    public void destroy() {
    }
}