org.postgresql.adba.operations.PgArrayRowCountOperation 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.operations;

import jdk.incubator.sql2.ArrayRowCountOperation;
import jdk.incubator.sql2.Result;
import jdk.incubator.sql2.SqlType;
import jdk.incubator.sql2.Submission;
import org.postgresql.adba.PgSession;
import org.postgresql.adba.PgSubmission;
import org.postgresql.adba.operations.helpers.ArrayQueryParameter;
import org.postgresql.adba.operations.helpers.FutureArrayQueryParameter;
import org.postgresql.adba.operations.helpers.ParameterHolder;
import org.postgresql.adba.submissions.ArrayCountSubmission;
import org.postgresql.adba.submissions.GroupSubmission;

import java.time.Duration;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.CompletionStage;
import java.util.function.Consumer;
import java.util.stream.Collector;

public class PgArrayRowCountOperation implements ArrayRowCountOperation {
  private final PgSession connection;
  private final String sql;
  private ParameterHolder holder;
  private Consumer errorHandler;
  private Collector collector;
  private GroupSubmission groupSubmission;

  /**
   * Creates an operation that contains a number of executions of the same query.
   * @param connection connection that the queries should be part of
   * @param sql the query
   * @param groupSubmission the group that this execution should be part of
   */
  public PgArrayRowCountOperation(PgSession connection, String sql, GroupSubmission groupSubmission) {
    this.connection = connection;
    this.sql = sql;
    this.holder = new ParameterHolder();
    this.groupSubmission = groupSubmission;
  }

  @Override
  public ArrayRowCountOperation set(String id, List values, SqlType type) {
    holder.add(id, new ArrayQueryParameter(values, type));
    return this;
  }

  @Override
  public ArrayRowCountOperation set(String id, List values) {
    holder.add(id, new ArrayQueryParameter(values));
    return this;
  }

  @Override
  public  ArrayRowCountOperation set(String id, S[] values, SqlType type) {
    holder.add(id, new ArrayQueryParameter(Arrays.asList(values), type));
    return this;
  }

  @Override
  public  ArrayRowCountOperation set(String id, S[] values) {
    holder.add(id, new ArrayQueryParameter(Arrays.asList(values)));
    return this;
  }

  @Override
  public ArrayRowCountOperation set(String id, CompletionStage source, SqlType type) {
    holder.add(id, new FutureArrayQueryParameter(source, type));
    return this;
  }

  @Override
  public ArrayRowCountOperation set(String id, CompletionStage source) {
    holder.add(id, new FutureArrayQueryParameter(source));
    return this;
  }

  @Override
  public  ArrayRowCountOperation collect(Collector c) {
    this.collector = c;
    return this;
  }

  @Override
  public Submission submit() {
    PgSubmission submission = new ArrayCountSubmission<>(this::cancel, errorHandler, holder, sql, groupSubmission);
    if (collector != null) {
      submission.setCollector(collector);
    }
    connection.submit(submission);
    return submission;
  }

  @Override
  public ArrayRowCountOperation onError(Consumer errorHandler) {
    if (this.errorHandler != null) {
      throw new IllegalStateException("you are not allowed to call onError multiple times");
    }

    this.errorHandler = errorHandler;
    return this;
  }

  @Override
  public ArrayRowCountOperation timeout(Duration minTime) {
    return this;
  }

  private boolean cancel() {
    // todo set life cycle to canceled
    return true;
  }
}