com.algolia.search.models.HttpRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of algoliasearch-core Show documentation
The newest version!
package com.algolia.search.models;

import com.algolia.search.models.common.CompressionType;
import java.io.InputStream;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Supplier;

public class HttpRequest {

  public HttpRequest(
      HttpMethod method,
      String methodPath,
      Map headers,
      int timeout,
      CompressionType compressionType) {
    this(method, methodPath, headers, timeout);
    this.compressionType = compressionType;
  }

  public HttpRequest(
      HttpMethod method, String methodPath, Map headers, int timeout) {
    this.method = method;
    this.methodPath = methodPath;
    this.headers = headers;
    this.timeout = timeout;
  }

  public HttpMethod getMethod() {
    return method;
  }

  public HttpRequest setMethod(HttpMethod method) {
    this.method = method;
    return this;
  }

  public URL getUri() {
    return uri;
  }

  public HttpRequest setUri(URL uri) {
    this.uri = uri;
    return this;
  }

  public String getMethodPath() {
    return methodPath;
  }

  public HttpRequest setMethodPath(String methodPath) {
    this.methodPath = methodPath;
    return this;
  }

  public Map getHeaders() {
    return headers;
  }

  public HttpRequest setHeaders(HashMap headers) {
    this.headers = headers;
    return this;
  }

  /** @deprecated produced input stream is not reusable, use {@link #getBodySupplier()} instead. */
  @Deprecated
  public InputStream getBody() {
    return body;
  }

  /** @deprecated use {@link #setBodySupplier(Supplier)} instead. */
  @Deprecated
  public HttpRequest setBody(InputStream body) {
    this.body = body;
    return this;
  }

  public int getTimeout() {
    return timeout;
  }

  public HttpRequest setTimeout(int timeout) {
    this.timeout = timeout;
    return this;
  }

  public CompressionType getCompressionType() {
    return compressionType;
  }

  public HttpRequest setCompressionType(CompressionType compressionType) {
    this.compressionType = compressionType;
    return this;
  }

  /**
   * Tells if any compression can be enabled for a request or not. Compression is enabled only for
   * POST/PUT methods on the Search API (not on Analytics and Insights).
   */
  public boolean canCompress() {
    if (this.compressionType == null || this.method == null) {
      return false;
    }

    boolean isMethodValid =
        this.method.equals(HttpMethod.POST) || this.method.equals(HttpMethod.PUT);
    boolean isCompressionEnabled = this.compressionType.equals(CompressionType.GZIP);

    return isMethodValid && isCompressionEnabled;
  }

  public void incrementTimeout(int retryCount) {
    // check for overflow
    if (timeout > 0 && retryCount > Integer.MAX_VALUE / timeout - 1) {
      timeout = Integer.MAX_VALUE;
    } else {
      timeout *= (retryCount + 1);
    }
  }

  public Supplier getBodySupplier() {
    return bodySupplier;
  }

  public HttpRequest setBodySupplier(Supplier bodySupplier) {
    this.bodySupplier = bodySupplier;
    return this;
  }

  private HttpMethod method;
  private URL uri;
  private String methodPath;
  private Map headers;
  private InputStream body; // deprecated since, not reusable
  private int timeout;
  private CompressionType compressionType;
  private Supplier bodySupplier;
}