用过滤器对页面数据进行压缩

package cn.lfd.web.filter;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.zip.GZIPOutputStream;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
/*
 * 对HttpServletResponse进行增强，让他的getOutputStream和getWriter方法不直接把数组打给浏览器，而是存到内存中，压缩后再打给浏览器
 */
public class GzipFilter implements Filter{

	@Override
	public void destroy() {
		
	}

	@Override
	public void doFilter(ServletRequest req, ServletResponse resp,
			FilterChain chain) throws IOException, ServletException {
		HttpServletRequest request = (HttpServletRequest) req;
		HttpServletResponse response = (HttpServletResponse) resp;
		
		BufferResponse bufferRequest = new BufferResponse(response);
		
		chain.doFilter(request, bufferRequest);
		
		byte buffer[] = bufferRequest.getBuffer();//得到response中ByteArrayOutputStream的数据
		System.out.println("压缩前"+buffer.length);
		
		ByteArrayOutputStream bout = new ByteArrayOutputStream();
		GZIPOutputStream gout = new GZIPOutputStream(bout);//用GZIPOutputStream对数据进行压缩
		gout.write(buffer);
		gout.close();//把压缩后的数据弄进bout
		
		byte flush[] = bout.toByteArray();
		System.out.println("压缩后"+flush.length);
		response.setHeader("content-encoding", "gzip");//告诉页面数据是gzip格式
		response.setContentLength(flush.length);//告诉页面数据的长度
		response.getOutputStream().write(flush);//通过字节输出流把含有数据的字节数组打给浏览器
	}

	@Override
	public void init(FilterConfig filterConfig) throws ServletException {
		
	}
}

//对HttpServletResponse的getWriter方法和getOutputStream进行增强
class BufferResponse extends HttpServletResponseWrapper {
	
	private PrintWriter pw;//打印流
	private ByteArrayOutputStream baos = new ByteArrayOutputStream();//定义一个字节数组输出流接收数据
	public BufferResponse(HttpServletResponse response) {
		super(response);
	}

	@Override
	public ServletOutputStream getOutputStream() throws IOException {
		// TODO Auto-generated method stub
		return new MyServletOutputStream(baos);//new 出一个我自己的ServletOutputStream给它,我自己的ServletOutputStream已被增强
	}

	@Override
	public PrintWriter getWriter() throws IOException {
		pw = new PrintWriter(new OutputStreamWriter(baos,super.getResponse().getCharacterEncoding()));//通过一个转化流把数据读到ByteArrayOutputStream
		return pw;
	}
	
	public byte[] getBuffer() {//得到含有数据的字节数组
		try {
			if(pw!=null) {//PrintWriter中有一个缓冲区域,需要调用close方法把其中的数据全部读到ByteArrayOutputStream
				pw.close();
			}
			if(baos!=null) {
				baos.flush();
				return baos.toByteArray();//把ByteArrayOutputStream中的数据弄进一个字节数组
			}
			return null;
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}
}

//对ServletOutputStream的write方法进行增强</span>
class MyServletOutputStream extends ServletOutputStream{
	private ByteArrayOutputStream baos;
	public MyServletOutputStream(ByteArrayOutputStream baos) {
		this.baos = baos;
	}
	@Override
	public void write(int b) throws IOException {
		this.baos.write(b);//把数据读到ByteArrayOutputStream中
	}
}