org.postgresql.adba.operations.PgValidationOperation 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 java.time.Duration;
import java.util.function.Consumer;
import java.util.stream.Collector;
import jdk.incubator.sql2.Operation;
import jdk.incubator.sql2.Session;
import jdk.incubator.sql2.Submission;
import org.postgresql.adba.PgSession;
import org.postgresql.adba.operations.helpers.ParameterHolder;
import org.postgresql.adba.submissions.BaseSubmission;
import org.postgresql.adba.submissions.RowSubmission;

public class PgValidationOperation implements Operation {
  private final PgSession connection;
  private final Session.Validation depth;
  private Consumer errorHandler;

  /**
   * This operation validates that the connection is still valid.
   * @param connection connection to validate
   * @param depth to what depth the validation should happen
   */
  public PgValidationOperation(PgSession connection, Session.Validation depth) {
    this.connection = connection;
    this.depth = depth;
  }

  @Override
  public Operation 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 Operation timeout(Duration minTime) {
    return this;
  }

  @Override
  public Submission submit() {
    switch (depth) {
      case NONE:
      case LOCAL:
        if (connection.isConnectionClosed()) {
          BaseSubmission submission = new BaseSubmission<>(this::cancel, BaseSubmission.Types.VOID, errorHandler,
              null, null, null);
          submission.getCompletionStage().toCompletableFuture().completeExceptionally(new IllegalStateException());
          return submission;
        } else {
          BaseSubmission submission = new BaseSubmission<>(this::cancel, BaseSubmission.Types.VOID, errorHandler,
              null, null, null);
          submission.getCompletionStage().toCompletableFuture().complete(null);
          return submission;
        }
      case SOCKET:
        break;
      case NETWORK:
        break;
      case SERVER:
        break;
      case COMPLETE:
        RowSubmission submission = new RowSubmission<>(this::cancel, errorHandler, new ParameterHolder(),
            null, "select 1");
        submission.setCollector(Collector.of(
            () -> null,
            (a, v) -> {
            },
            (a, b) -> null,
            a -> null));
        connection.submit(submission);
        return submission;
      default:
        throw new IllegalStateException("not all enum values implemented in switch statement");
    }
    throw new IllegalArgumentException();
  }

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