filter,HttpServletResponseWrapper拦截http请求,获取响应消息

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

1、获取response对象,并将输出存放在 自定义的流里面,那么关于输出流(outputStream、writer)的操作都是需要重写的了:
   1)以流的方式获取输出——重写getOutputStream()
   2)以字符方式获取输出——重写getWriter()
   3)刷新流——重写flushBuffer()
   4)重置流——重写reset()
然后加入新增的获取输出数据的方法就ok了。 2、定义response包装器WapperedResponse继承HttpServletResponseWrapper
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.UnsupportedEncodingException;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;

public class MonitorResponseWrapper extends HttpServletResponseWrapper {

	private ByteArrayOutputStream buffer=null;
	private ServletOutputStream out=null;
	private PrintWriter writer=null;
	
	public MonitorResponseWrapper(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{
		//将out、writer中的数据强制输出到WapperedResponse的buffer里面,否则取不到数据
		flushBuffer();
		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();
	}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值