001/* 002 * 003 * Licensed to the Apache Software Foundation (ASF) under one 004 * or more contributor license agreements. See the NOTICE file 005 * distributed with this work for additional information 006 * regarding copyright ownership. The ASF licenses this file 007 * to you under the Apache License, Version 2.0 (the 008 * "License"); you may not use this file except in compliance 009 * with the License. You may obtain a copy of the License at 010 * 011 * http://www.apache.org/licenses/LICENSE-2.0 012 * 013 * Unless required by applicable law or agreed to in writing, software 014 * distributed under the License is distributed on an "AS IS" BASIS, 015 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 016 * See the License for the specific language governing permissions and 017 * limitations under the License. 018 */ 019 020package org.apache.hadoop.hbase.rest.filter; 021 022import java.io.IOException; 023import java.io.OutputStream; 024import java.util.HashSet; 025import java.util.Locale; 026import java.util.Set; 027import java.util.StringTokenizer; 028 029import javax.servlet.Filter; 030import javax.servlet.FilterChain; 031import javax.servlet.FilterConfig; 032import javax.servlet.ServletException; 033import javax.servlet.ServletRequest; 034import javax.servlet.ServletResponse; 035import javax.servlet.http.HttpServletRequest; 036import javax.servlet.http.HttpServletResponse; 037 038import org.apache.yetus.audience.InterfaceAudience; 039import org.apache.hadoop.hbase.HBaseInterfaceAudience; 040 041@InterfaceAudience.LimitedPrivate(HBaseInterfaceAudience.CONFIG) 042public class GzipFilter implements Filter { 043 private Set<String> mimeTypes = new HashSet<>(); 044 045 @Override 046 public void init(FilterConfig filterConfig) throws ServletException { 047 String s = filterConfig.getInitParameter("mimeTypes"); 048 if (s != null) { 049 StringTokenizer tok = new StringTokenizer(s, ",", false); 050 while (tok.hasMoreTokens()) { 051 mimeTypes.add(tok.nextToken()); 052 } 053 } 054 } 055 056 @Override 057 public void destroy() { 058 } 059 060 @Override 061 public void doFilter(ServletRequest req, ServletResponse rsp, 062 FilterChain chain) throws IOException, ServletException { 063 HttpServletRequest request = (HttpServletRequest)req; 064 HttpServletResponse response = (HttpServletResponse)rsp; 065 String contentEncoding = request.getHeader("content-encoding"); 066 String acceptEncoding = request.getHeader("accept-encoding"); 067 String contentType = request.getHeader("content-type"); 068 if ((contentEncoding != null) && 069 (contentEncoding.toLowerCase(Locale.ROOT).indexOf("gzip") > -1)) { 070 request = new GZIPRequestWrapper(request); 071 } 072 if (((acceptEncoding != null) && 073 (acceptEncoding.toLowerCase(Locale.ROOT).indexOf("gzip") > -1)) || 074 ((contentType != null) && mimeTypes.contains(contentType))) { 075 response = new GZIPResponseWrapper(response); 076 } 077 chain.doFilter(request, response); 078 if (response instanceof GZIPResponseWrapper) { 079 OutputStream os = response.getOutputStream(); 080 if (os instanceof GZIPResponseStream) { 081 ((GZIPResponseStream)os).finish(); 082 } 083 } 084 } 085 086}