org.postgresql.adba.submissions.OutSubmission Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pgadba Show documentation
ADBA implementation for PostgreSQL
The newest version!
package org.postgresql.adba.submissions;

import jdk.incubator.sql2.Result;
import jdk.incubator.sql2.SqlType;
import org.postgresql.adba.PgSubmission;
import org.postgresql.adba.communication.packets.DataRow;
import org.postgresql.adba.operations.helpers.ParameterHolder;

import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collector;

public class OutSubmission implements PgSubmission {
  private final Supplier cancel;
  private CompletableFuture publicStage;
  private Consumer errorHandler;
  private String sql;
  private ParameterHolder holder;
  private GroupSubmission groupSubmission;
  private final AtomicBoolean sendConsumed = new AtomicBoolean(false);
  private Function outParameterProcessor;
  private T outParameterValueHolder;
  private Map outParameterTypeMap;

  /**
   * Submission for an Out Operation.
   *
   * @param cancel cancel method
   * @param errorHandler error handler method
   * @param sql the query
   * @param outParameterTypes the types of the out parameters
   * @param processor the processor that will consume the lines
   * @param groupSubmission group submission this submission is a part of
   * @param holder holder for parameter values
   */
  public OutSubmission(Supplier cancel, Consumer errorHandler, String sql,
      Map outParameterTypes, Function processor,
      GroupSubmission groupSubmission, ParameterHolder holder) {
    this.cancel = cancel;
    this.errorHandler = errorHandler;
    this.sql = sql;
    this.outParameterTypeMap = outParameterTypes;
    this.outParameterProcessor = processor;
    this.groupSubmission = groupSubmission;
    this.holder = holder;
  }

  @Override
  public String getSql() {
    return sql;
  }

  @Override
  public AtomicBoolean getSendConsumed() {
    return sendConsumed;
  }

  @Override
  public ParameterHolder getHolder() {
    return holder;
  }

  @Override
  public Types getCompletionType() {
    return Types.OUT_PARAMETER;
  }

  @Override
  public void setCollector(Collector collector) {

  }

  @Override
  public Object finish(Object finishObject) {
    if (groupSubmission != null) {
      groupSubmission.addGroupResult(outParameterValueHolder);
    }
    ((CompletableFuture) getCompletionStage())
        .complete(outParameterValueHolder);
    return null;
  }

  @Override
  public void addRow(DataRow row) {
    outParameterValueHolder = outParameterProcessor.apply(row);
  }

  @Override
  public List getParamTypes() throws ExecutionException, InterruptedException {
    return holder.getParamTypes();
  }

  @Override
  public int numberOfQueryRepetitions() throws ExecutionException, InterruptedException {
    return 1;
  }

  @Override
  public Consumer getErrorHandler() {
    return errorHandler;
  }

  @Override
  public CompletionStage cancel() {
    return null;
  }

  @Override
  public CompletionStage getCompletionStage() {
    if (publicStage == null) {
      publicStage = new CompletableFuture<>();
    }

    return publicStage;
  }
}