com.hubspot.horizon.apache.internal.SnappyContentEncodingResponseInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of HorizonApache Show documentation

The newest version!

package com.hubspot.horizon.apache.internal;

import com.google.common.base.Strings;
import com.google.common.collect.ImmutableSet;
import com.google.common.net.HttpHeaders;
import java.io.IOException;
import java.util.Locale;
import java.util.Set;
import org.apache.http.Header;
import org.apache.http.HttpException;
import org.apache.http.HttpResponse;
import org.apache.http.client.protocol.ResponseContentEncoding;
import org.apache.http.protocol.HttpContext;

public class SnappyContentEncodingResponseInterceptor extends ResponseContentEncoding {

  private static final Set IGNORED_ENCODINGS = ImmutableSet.of(
    "none",
    "utf8",
    "utf-8",
    "binary"
  );

  @Override
  public void process(HttpResponse response, HttpContext context)
    throws HttpException, IOException {
    Header encodingHeader = response.getFirstHeader(HttpHeaders.CONTENT_ENCODING);
    if (encodingHeader == null) {
      return;
    }

    String encoding = Strings
      .nullToEmpty(encodingHeader.getValue())
      .toLowerCase(Locale.ENGLISH);
    if ("snappy".equals(encoding) && response.getEntity() != null) {
      response.setEntity(new SnappyDecompressingEntity(response.getEntity()));
    } else if (!IGNORED_ENCODINGS.contains(encoding)) {
      super.process(response, context);
    }
  }
}