com.signalfx.shaded.apache.http.client.protocol.ResponseContentEncoding Maven / Gradle / Ivy
Show all versions of signalfx-codahale Show documentation
/*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
* .
*
*/
package com.signalfx.shaded.apache.http.client.protocol;
import java.io.IOException;
import java.util.Locale;
import com.signalfx.shaded.apache.http.Header;
import com.signalfx.shaded.apache.http.HeaderElement;
import com.signalfx.shaded.apache.http.HttpEntity;
import com.signalfx.shaded.apache.http.HttpException;
import com.signalfx.shaded.apache.http.HttpResponse;
import com.signalfx.shaded.apache.http.HttpResponseInterceptor;
import com.signalfx.shaded.apache.http.annotation.Contract;
import com.signalfx.shaded.apache.http.annotation.ThreadingBehavior;
import com.signalfx.shaded.apache.http.client.config.RequestConfig;
import com.signalfx.shaded.apache.http.client.entity.DecompressingEntity;
import com.signalfx.shaded.apache.http.client.entity.DeflateInputStreamFactory;
import com.signalfx.shaded.apache.http.client.entity.GZIPInputStreamFactory;
import com.signalfx.shaded.apache.http.client.entity.InputStreamFactory;
import com.signalfx.shaded.apache.http.config.Lookup;
import com.signalfx.shaded.apache.http.config.RegistryBuilder;
import com.signalfx.shaded.apache.http.protocol.HttpContext;
/**
* {@link HttpResponseInterceptor} responsible for processing Content-Encoding
* responses.
*
* Instances of this class are stateless and immutable, therefore threadsafe.
*
* @since 4.1
*
*/
@Contract(threading = ThreadingBehavior.IMMUTABLE_CONDITIONAL)
public class ResponseContentEncoding implements HttpResponseInterceptor {
public static final String UNCOMPRESSED = "http.client.response.uncompressed";
private final Lookup decoderRegistry;
private final boolean ignoreUnknown;
/**
* @since 4.5
*/
public ResponseContentEncoding(final Lookup decoderRegistry, final boolean ignoreUnknown) {
this.decoderRegistry = decoderRegistry != null ? decoderRegistry :
RegistryBuilder.create()
.register("gzip", GZIPInputStreamFactory.getInstance())
.register("x-gzip", GZIPInputStreamFactory.getInstance())
.register("deflate", DeflateInputStreamFactory.getInstance())
.build();
this.ignoreUnknown = ignoreUnknown;
}
/**
* @since 4.5
*/
public ResponseContentEncoding(final boolean ignoreUnknown) {
this(null, ignoreUnknown);
}
/**
* @since 4.4
*/
public ResponseContentEncoding(final Lookup decoderRegistry) {
this(decoderRegistry, true);
}
/**
* Handles {@code gzip} and {@code deflate} compressed entities by using the following
* decoders:
*
* - gzip - see {@link java.util.zip.GZIPInputStream}
* - deflate - see {@link com.signalfx.shaded.apache.http.client.entity.DeflateInputStream}
*
*/
public ResponseContentEncoding() {
this(null);
}
@Override
public void process(
final HttpResponse response,
final HttpContext context) throws HttpException, IOException {
final HttpEntity entity = response.getEntity();
final HttpClientContext clientContext = HttpClientContext.adapt(context);
final RequestConfig requestConfig = clientContext.getRequestConfig();
// entity can be null in case of 304 Not Modified, 204 No Content or similar
// check for zero length entity.
if (requestConfig.isContentCompressionEnabled() && entity != null && entity.getContentLength() != 0) {
final Header ceheader = entity.getContentEncoding();
if (ceheader != null) {
final HeaderElement[] codecs = ceheader.getElements();
for (final HeaderElement codec : codecs) {
final String codecname = codec.getName().toLowerCase(Locale.ROOT);
final InputStreamFactory decoderFactory = decoderRegistry.lookup(codecname);
if (decoderFactory != null) {
response.setEntity(new DecompressingEntity(response.getEntity(), decoderFactory));
response.removeHeaders("Content-Length");
response.removeHeaders("Content-Encoding");
response.removeHeaders("Content-MD5");
} else {
if (!"identity".equals(codecname) && !ignoreUnknown) {
throw new HttpException("Unsupported Content-Encoding: " + codec.getName());
}
}
}
}
}
}
}