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