Filter过滤器(gzip) 全站 压缩

package com.itheima.filter;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
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;

public class GzipFilter implements Filter {

	private FilterConfig config;
	
	@Override
	public void init(FilterConfig filterConfig) throws ServletException {

		this.config = filterConfig;
	}

	@Override
	public void doFilter(ServletRequest req, ServletResponse resp,
			FilterChain chain) throws IOException, ServletException {

		HttpServletRequest request = (HttpServletRequest) req;
		HttpServletResponse response = (HttpServletResponse) resp;
		
		String encoding = config.getInitParameter("encoding");
		request.setCharacterEncoding(encoding);
		response.setContentType("text/html;charset="+encoding);  
		//
		
//		response.setCharacterEncoding("")
		
		
		// HttpServletRequestWrapper  ---对request 进行 包装的包装类 ----   解决全站的 请求数据的乱码
		
		//HttpServletResponseWrapper ----对response 进行 包装的 包装类 ----  解决全站的 响应的数据的 压缩 
		
		MyHttpServletResponseWrapper  myresponse = new MyHttpServletResponseWrapper(response);
		 
		
		//放行 
		chain.doFilter(request, myresponse);  //  --- 目标 资源会得到 执行 
		
		
		System.out.println("之后 ...");
		
		// 将 写到   myresponse 里的数据 进行 压缩 , 然后  输出到 浏览器 去 
		
		//这里 报错的 原因是 因为 
		// 没有 原始的数据(压缩前的数据), 那么 如果 能够 将原始的 压缩前的数据 拿到 
		// 所有的问题就 都解决了... 
//		byte[] b = data.getBytes();
		byte[] b = myresponse.getOldData();
		
		
		System.out.println("压缩前: " + b.length);
		
		//底层流 
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		
		//采用gzip压缩 
		
		//这里 压缩的时候, 需要的是一个 输出流, 这里的输出流是一个底层流 
		GZIPOutputStream gout= new GZIPOutputStream(baos);
		gout.write(b);
		
		gout.close();  //确保 数据可以写到 底层流 baos中去 
		
		//这里 要注意: 由于 数据是写到 底层流 baos中去的,  gout 默认的是有 缓冲区的 .
		
		b = baos.toByteArray();
		
		System.out.println("压缩后 : " +b.length);
		
		//告诉浏览器 需要 解压缩 数据-- 通过 http的响应头 实现
		response.setHeader("content-encoding", "gzip");
		response.setContentLength(b.length);  //数据的长度 
		
		response.getOutputStream().write(b);
		
		
	}

	@Override
	public void destroy() {

	}

}

class MyHttpServletResponseWrapper extends HttpServletResponseWrapper{

	private HttpServletResponse response;

	public MyHttpServletResponseWrapper(HttpServletResponse response) {
		super(response);
		this.response= response;
	}
	
	ByteArrayOutputStream bout = new ByteArrayOutputStream();
	
	
	//通过 这个方法 可以 拿到 原始的 数据 
	public byte[] getOldData(){
		
		
		if(pw!=null){
			pw.close();
		}
		
		return bout.toByteArray();
	}
	
	PrintWriter pw = null;
	@Override
	public PrintWriter getWriter() throws IOException {
		
		//字符输出流, 有缓冲区 , 
		if(pw==null){
			
			//字符输出流  -----  bout (字节流)
			pw = new PrintWriter(new OutputStreamWriter(bout, response.getCharacterEncoding()));
		}
		return pw;
	}
	
	@Override
	public ServletOutputStream getOutputStream() throws IOException {
		
		//在 调用这个 方法的时候, 数据 写到 一个 俺们自己 定义的输出流去, 那么这样 数据就进入了 自己 定义的输出流中去 了  . 
		return new MyServletOutputStream(bout);
	}
}

class MyServletOutputStream extends ServletOutputStream{

	private OutputStream out;


	public MyServletOutputStream(OutputStream out){
		this.out = out;
	}
	
	
	//最终 是 调用到 这个 方法里 了 . 
	
	//所有的数据 都进入 到了     out  中去 了,而这里 out 又 是  MyHttpServletResponseWrapper 里的 bout 了, 所以 所有的数据 都 在 bout 中了 
	@Override
	public void write(int b) throws IOException {
		out.write(b);
	}
	
	// 这里 没有 一个 接受 字节 数组的write 方法
	
}