Servlet 过滤器和响应包装器答案

【问题标题】：Servlet filters and response wrapperServlet 过滤器和响应包装器
【发布时间】：2014-04-17 14:25:22
【问题描述】：

试图弄乱我尝试过的响应内容this filter tutorial

所以我将我的类定义为

public class FilterServletOutputStream extends ServletOutputStream
{
    //=======================================================================
    private DataOutputStream stream; 
    //=======================================================================
    public FilterServletOutputStream(OutputStream output) { 
      stream = new DataOutputStream(output); 
    }
    //=======================================================================
    @Override
    public void write(int b) throws IOException  {
        stream.write(b); 
    }
    //=======================================================================
    @Override
    public void write(byte[] b) throws IOException  { 
        stream.write(b); 
    }
    //=======================================================================
    @Override
    public void write(byte[] b, int off, int len) throws IOException  {
        stream.write(b,off,len); 
    }
    //=======================================================================
}




public class GenericResponseWrapper extends HttpServletResponseWrapper 
{ 
  private ByteArrayOutputStream output;
  private int contentLength;
  private String contentType;
  public GenericResponseWrapper(HttpServletResponse response) { 
    super(response);
    output=new ByteArrayOutputStream();
  } 
  public byte[] getData() { 
    return output.toByteArray(); 
  } 
  @Override
  public ServletOutputStream getOutputStream() 
  { 
    return new FilterServletOutputStream(output); 
  } 

  @Override
  public PrintWriter getWriter() 
  { 
    return new PrintWriter(getOutputStream(),true); 
  } 

  @Override
  public void setContentLength(int length) { 
    this.contentLength = length;
    super.setContentLength(length); 
  } 

  public int getContentLength() { 
    return contentLength; 
  } 
  @Override
  public void setContentType(String type) { 
    this.contentType = type;
    super.setContentType(type); 
  } 
  @Override
  public String getContentType() { 
    return contentType; 
  } 
}

然后是我的过滤器。

public void  doFilter(ServletRequest request, ServletResponse response, FilterChain   chain) throws IOException, ServletException
{
    //===================================================================
    GenericResponseWrapper wrapper = new GenericResponseWrapper((HttpServletResponse) response);
    chain.doFilter(request, wrapper);
    OutputStream out = response.getOutputStream();
    out.write(wrapper.getData());
    out.write("test content".getBytes());
    out.close();
    //===================================================================
}

过滤器似乎没问题。我可以阅读“测试内容”，但似乎 wrapper.getData() 返回 0 个字节。当我调用chain.doFilter(..) 时，通常是一个servlet，有时它会分派到一个jsp 页面。尝试了一个 JSP 普通页面的 URL，但它似乎没有写入数据。过滤器被配置为捕获所有请求并且它做得很好。

我做错了什么？唯一的区别是我添加了@override。但我也尝试过不使用它们。

添加了一个 JSP。

  <%@page contentType="text/html" pageEncoding="UTF-8" session="false"%>
  <!DOCTYPE html>
  <html>
  <head>
  <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
  <style>
  <%-- @include file="/static/css/divs.css" --%>
  <%-- @include file="/static/css/links.css" --%>
  </style>
  <title>title</title>
  </head>
  <body style="margin-top: 0px; margin-left: 0px; margin-right: 0px">
  <%--@include file="publicheader.jsp" --%>
  <div class="desktopwebpagecontainer">
  main page.
  </div>
  <%--@include file="publicfooter.jsp" --%>
  </body>
  </html>

【问题讨论】：

你怎么知道wrapper.getData()返回0字节？
我不能这就是为什么我说“看起来”。我能说的是它不为空。
打印出getData().length。
我打印到控制台。为 0。之前无法打印到响应中。
我无法重现这个。给我们看一个Servlet，它会重现这个。尽量减少示例。

标签： java jsp servlets servlet-filters

【解决方案1】：

解决这个问题的原因是使 pwriter 和 outpstrm 类成员（字段）成为可能。

private PrintWriter pwriter = null;
private ServletOutputStream outpstrm = null;

然后改变获取方法

@Override
public ServletOutputStream getOutputStream() 
{ 
    if (outpstrm == null) outpstrm = new AppServletOutputStream(output);
    return outpstrm;
} 
@Override
public PrintWriter getWriter() 
{ 
    if (pwriter == null) pwriter = new PrintWriter(getOutputStream(),true); 
    return pwriter;
}

这似乎解决了我的问题。由于此更改，包装器工作正常。顺便一提。谁说RequestDispatcher类的方法forward(req, res)方法不能调用？

【讨论】：

是AppServletOutputStream 与FilterServletOutputStream 相同。其次，我无法让request.getRequestDispatcher("abcd.html").forward(request, response); 工作。我在过滤器映射中添加了<dispatcher>FORWARD</dispatcher> <dispatcher>REQUEST</dispatcher>，因此过滤器在转发期间也会被调用，但不知何故修改后的响应不会发送到浏览器。浏览器仍然显示 html 文件中的 abcd.html。