获取Servlet的响应并修改其内容

获取Servlet的响应并修改其内容

---

• 有的时候我们需要批量的修改Servlet中的响应的内容，这时候我们就可以通过HttpServletResponseWrapper这个类，在Filter中先获取响应的内容，接着修改响应的内容，最后再把修改完之后的内容作为响应返回给客户端。代码如下演示：

1.创建一个继承自HttpServletResponseWrapper的类

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import java.io.CharArrayWriter;
import java.io.PrintWriter;

/**
* Created on 2016/3/29 19:54.
*/
public class ResponseWrapper extends HttpServletResponseWrapper {
private PrintWriter cachedWriter;
private CharArrayWriter bufferedWriter;

public ResponseWrapper(HttpServletResponse response) {
super(response);

// 设置响应体格式
response.setHeader("Content-type", "text/json;charset=UTF-8");
// 这个是我们保存返回结果的地方
bufferedWriter = new CharArrayWriter();
// 这个是包装PrintWriter的，让所有结果通过这个PrintWriter写入到bufferedWriter中
cachedWriter = new PrintWriter(bufferedWriter);
}

@Override
public PrintWriter getWriter() {
return cachedWriter;
}

/**
* 获取原始的HTML页面内容。
*
* @return
*/
public String getResult() {
return bufferedWriter.toString();
}
}

2.在过滤器中修改响应的内容

package cn.iduoka.util.format;

import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

/**
* Created on 2016/3/29 19:55.
*/
@WebFilter(filterName = "FormatFilter", urlPatterns = ("/"))
public class FormatFilter implements Filter {

@Override
public void destroy() {
// TODO Auto-generated method stub

}

@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
// 使用我们自定义的响应包装器来包装原始的ServletResponse
ResponseWrapper wrapper = new ResponseWrapper((HttpServletResponse) response);
// 这句话非常重要，注意看到第二个参数是我们的包装器而不是response
chain.doFilter(request, wrapper);
// 获取到我们所截获的值 --> 做处理之前所获取到的值
String result = wrapper.getResult();
// 对返回的结果做处理操作
result = result.replace("</html>", "<div style='display: none;'>hello world</div>\n</html>");
// 重置响应输出的内容长度
response.setContentLength(-1);
// 输出最终的结果 --> 做完处理之后再把这个值返回回去
PrintWriter out = response.getWriter();
out.write(result);
out.flush();
out.close();
}

@Override
public void init(FilterConfig filterConfig) throws ServletException {
// TODO Auto-generated method stub

}

}