de.datexis.encoder.AbstractRESTAdapter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of texoo-encoder-api Show documentation
TeXoo REST encoder interface
There is a newer version: 1.3.3
package de.datexis.encoder;

import de.datexis.encoder.serialization.DeserializationProvider;
import de.datexis.encoder.serialization.SerializationProvider;
import org.nd4j.shade.jackson.annotation.JsonIgnore;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * Outline for REST API Adapter.
 */
public abstract class AbstractRESTAdapter implements RESTAdapter {
  
  private static final Logger log = LoggerFactory.getLogger(AbstractRESTAdapter.class);

  public static final String HTTP_REQUEST_METHOD = "POST";
  public static final String HTTP_CONTENT_TYPE_NAME = "Content-Type";
  public static final String HTTP_ACCEPT_TYPE_NAME = "Accept";

  private long embeddingVectorSize;

  private int connectTimeout;
  private int readTimeout;

  protected AbstractRESTAdapter() {}
  
  public AbstractRESTAdapter(long embeddingVectorSize, int connectTimeout, int readTimeout) {
    this.connectTimeout = connectTimeout;
    this.readTimeout = readTimeout;
    this.embeddingVectorSize = embeddingVectorSize;
  }

  @Override
  public long getEmbeddingVectorSize() {
    return embeddingVectorSize;
  }
  
  public void setEmbeddingVectorSize(long embeddingVectorSize) {
    this.embeddingVectorSize = embeddingVectorSize;
  }
  
  public int getConnectTimeout() {
    return connectTimeout;
  }
  
  public void setConnectTimeout(int connectTimeout) {
    this.connectTimeout = connectTimeout;
  }
  
  public int getReadTimeout() {
    return readTimeout;
  }
  
  public void setReadTimeout(int readTimeout) {
    this.readTimeout = readTimeout;
  }
  
  @Override
  public double[] encode(String data) throws IOException {
    try {
      return encodeImpl(data);
    } catch (IOException e) {
      log.error("IO error while encoding: {}", data, e);
      throw e;
    }
  }

  public abstract double[] encodeImpl(String data) throws IOException;

  @Override
  public double[][] encode(String[] data) throws IOException {
    try {
      return encodeImpl(data);
    } catch (IOException e) {
      log.error("IO error while encoding: {}", data, e);
      throw e;
    }
  }

  public abstract double[][] encodeImpl(String[] data) throws IOException;

  @Override
  public double[][][] encode(String[][] data) throws IOException {
    try {
      return encodeImpl(data);
    } catch (IOException e) {
      log.error("IO error while encoding: {}", data, e);
      throw e;
    }
  }

  public abstract double[][][] encodeImpl(String[][] data) throws IOException;

  public  O request(I input, Class classOfO, URL url) throws IOException{
    log.debug("building request");
    HttpURLConnection httpConnection = configureConnection(url);

    log.debug("connect to: {}", httpConnection.getURL());
    httpConnection.connect();

    log.debug("writing to: {}", httpConnection.getURL());
    getSerializationProvider().serialize(input, httpConnection.getOutputStream());

    log.debug("reading from: {}", httpConnection.getURL());
    O output = getDeserializationProvider().deserialize(httpConnection.getInputStream(), classOfO);
    log.debug("response read from: {}", httpConnection.getURL());

    httpConnection.disconnect();
    log.debug("disconnected from: {}", httpConnection.getURL());

    return output;
  }

  public HttpURLConnection configureConnection(URL url) throws IOException{
    HttpURLConnection httpConnection = getConnection(url);
    httpConnection.setRequestMethod(HTTP_REQUEST_METHOD);
    httpConnection.setRequestProperty(HTTP_CONTENT_TYPE_NAME, getSerializationProvider().getContentType());
    httpConnection.setRequestProperty(HTTP_ACCEPT_TYPE_NAME, getDeserializationProvider().getAcceptType());
    httpConnection.setConnectTimeout(connectTimeout);
    httpConnection.setReadTimeout(readTimeout);
    httpConnection.setDoOutput(true);
    httpConnection.setDoInput(true);

    return httpConnection;
  }

  @JsonIgnore
  public abstract SerializationProvider getSerializationProvider();
  
  @JsonIgnore
  public abstract DeserializationProvider getDeserializationProvider();

  public HttpURLConnection getConnection(URL url) throws IOException {
    return (HttpURLConnection) url.openConnection();
  }

}