在过滤器中添加标头作为响应?

时间:2015-09-28 18:02:37

标签: java spring-mvc filter httpresponse servlet-filters

我需要在每个响应中添加标头。我打算在下面做

public class MyFilter extends OncePerRequestFilter {

    @Override
    protected void doFilterInternal(HttpServletRequest request,
                                    HttpServletResponse response, FilterChain filterChain)
            throws ServletException, IOException {

        filterChain.doFilter(request, response);
            response.addHeader("Access-Control-Allow-Origin", "*"); 
    }

}

我想在filterChain.doFilter(request, response)之后执行此操作,以便一旦控制器处理它,我只需在返回之前添加标头 对客户。这是对的吗?

但按照How to write response filter?

  

chain.doFilter返回后,做任何事都为时已晚   响应。此时,整个响应已经发送到了   客户端和您的代码无法访问它。

以上陈述对我来说不合适。我无法在filterChain.doFilter(request, response)之后添加标题吗?如果不是为什么?

我正在使用spring mvc。

4 个答案:

答案 0 :(得分:17)

在调用filterChain.doFilter之后,对响应做任何事都为时已晚。此时,整个响应已经发送给客户端。

您需要在自己的类中构建一个包装响应,将这些包装器传递给doFilter方法并处理包装器中的任何处理。

已经有一个响应包装器:HttpServletResponseWrapper可以扩展。例如:

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

您的过滤器:

@Override
protected void doFilterInternal(HttpServletRequest request,
                                HttpServletResponse response, FilterChain filterChain)
        throws ServletException, IOException {

    HttpServletResponse myResponse = (HttpServletResponse) response;
    MyResponseRequestWrapper responseWrapper = new MyResponseRequestWrapper(myResponse);
    responseWrapper.addHeader("Access-Control-Allow-Origin", "*");
    filterChain.doFilter(request, myResponse);
}

答案 1 :(得分:1)

我在Spring 3.0.x的项目中使用了它:

@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException
{
    response.addHeader("headerName", "headerValue");
    filterChain.doFilter(request, response);
}

很好。

答案 2 :(得分:0)

<强> From perfect article

修改响应的过滤器通常必须在响应返回到客户端之前捕获响应。为此,您将一个替代流传递给生成响应的servlet。替代流阻止servlet在完成时关闭原始响应流,并允许过滤器修改servlet的响应。

要将此替换流传递给servlet,过滤器会创建一个响应包装器,它会覆盖getWriter或getOutputStream方法以返回此替换流。包装器传递给过滤器链的doFilter方法。包装器方法默认调用包装的请求或响应对象。这种方法遵循设计模式中描述的众所周知的Wrapper或Decorator模式,

答案 3 :(得分:0)

这有点晚了,但以下内容可能会有所帮助 因此,如果您真的想将值附加到现有标头中,或将新值添加到现有标头中,则最好的方法是编写包装器并在包装器中设置值。

然后将响应链接到过滤器中

HttpServletResponse response = (HttpServletResponse) servletResponse;
ByteArrayPrinter pw = new ByteArrayPrinter();

// Create a wrapper
HttpServletResponse wrappedResp = new HttpServletResponseWrapper(response) {

    @Override
    public void setContentType(final String type) {
        super.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
    }

    @Override
    public PrintWriter getWriter() {
        return pw.getWriter();
    }

    // set the outputstream content type to JSON
    @Override
    public ServletOutputStream getOutputStream() throws IOException {
        ServletResponse response = this.getResponse();

        String ct = (response != null) ? response.getContentType() : null;
        if (ct != null && ct.contains(APPLICATION_XHTML)) {
            response.setContentType(ct + AppConstants.CONSTANT_COMMA + MediaType.APPLICATION_JSON_UTF8_VALUE);
        }
        return pw.getStream();
    }

};
chain.doFilter(httpRequest, wrappedResp);

这是ByteArrayPrinter.java

public class ByteArrayPrinter {

    private ByteArrayOutputStream baos = new ByteArrayOutputStream();

    private PrintWriter pw = new PrintWriter(baos);

    private ServletOutputStream sos = new ByteArrayServletStream(baos);

    public PrintWriter getWriter() {
        return pw;
    }

    public ServletOutputStream getStream() {
        return sos;
    }

    byte[] toByteArray() {
        return baos.toByteArray();
    }
}

这是ByteArrayServletOutputStream

public class ByteArrayServletStream extends ServletOutputStream {

    ByteArrayOutputStream baos;

    ByteArrayServletStream(ByteArrayOutputStream baos) {
        this.baos = baos;
    }

    @Override
    public void write(int param) throws IOException {
        baos.write(param);
    }

    @Override
    public boolean isReady() {
        // TODO Auto-generated method stub
        return false;
    }

    @Override
    public void setWriteListener(WriteListener listener) {
        // TODO Auto-generated method stub

    }

}