子类化HttpServletResponseWrapper

HttpServletResponseWrapper为我们实现对response对象的后处理提供了帮助——你只需编写一个HttpServletResponseWrapper的子类,加入自己的功能实现(修饰器模式)。那么子类化HttpServletResponseWrapper都需要重写那些方法呢?

1、获取response对象,并将输出存放在 自定义的流里面,那么关于输出流(outputStream、writer)的操作都是需要重写的了:
   1)以流的方式获取输出——重写getOutputStream()
   2)以字符方式获取输出——重写getWriter()
   3)刷新流——重写flushBuffer()
   4)重置流——重写reset()
然后加入新增的获取输出数据的方法就ok了。
2、定义response包装器WapperedResponse继承HttpServletResponseWrapper
public class WapperedResponse extends HttpServletResponseWrapper {
	private ByteArrayOutputStream buffer=null;
	private ServletOutputStream out=null;
	private PrintWriter writer=null;
	
	public GZipResponse(HttpServletResponse resp) throws IOException{
		super(resp);
		buffer=new ByteArrayOutputStream();//真正存储数据的流
		out=new WapperedOutputStream(buffer);
		writer=new PrintWriter(new OutputStreamWriter(buffer,this.getCharacterEncoding()));
	}
	//重载父类获取outputstream的方法
	@Override
	public ServletOutputStream getOutputStream()throws IOException{
		return out;
	}
	//重载父类获取writer的方法
	@Override
	public PrintWriter getWriter() throws UnsupportedEncodingException{
		return writer;
	}
	//重载父类获取flushBuffer的方法
	@Override
	public void flushBuffer()throws IOException{
		if(out!=null){
			out.flush();
		}
		if(writer!=null){
			writer.flush();
		}
	}
	@Override
	public void reset(){
		buffer.reset();
	}
	public byte[] getResponseData()throws IOException{
		flushBuffer();//将out、writer中的数据强制输出到WapperedResponse的buffer里面,否则取不到数据
		return buffer.toByteArray();
	}
	
	//内部类,对ServletOutputStream进行包装
	private class WapperedOutputStream extends ServletOutputStream{
		private ByteArrayOutputStream bos=null;
		public WapperedOutputStream(ByteArrayOutputStream stream) throws IOException{
			bos=stream;
		}
		@Override
		public void write(int b) throws IOException{
			bos.write(b);
		}
	}
}

那么Servlet容器通过调用getOutputStream()方法获得的输出流将是我们自定义的包装流WapperedOutputStream。

3、现在就可以在过滤器中使用WapperedResponse进行对response包装、处理了。
public void doFilter(ServletRequest arg0, ServletResponse arg1,
			FilterChain arg2) throws IOException, ServletException {
		HttpServletResponse resp=(HttpServletResponse)arg1;
		WapperedResponse wapper=new WapperedResponse(resp);
		arg2.doFilter(arg0, wapper);
		byte[] b1=wapper.getResponseData();
 		//do something with b1 here
		byte[] b2=...;
		//输出处理后的数据
		ServletOutputStream output=arg1.getOutputStream();
		output.write(b2);
		output.flush();
	}

你可能感兴趣的:(servlet)