org.postgresql.adba.communication.network.CloseResponse Maven / Gradle / Ivy

Go to download

package org.postgresql.adba.communication.network;

import java.io.IOException;
import jdk.incubator.sql2.Submission;
import org.postgresql.adba.communication.NetworkReadContext;
import org.postgresql.adba.communication.NetworkResponse;

public class CloseResponse implements NetworkResponse {

  private final Submission submission;

  public CloseResponse(Submission submission) {
    this.submission = submission;
  }

  @Override
  public NetworkResponse read(NetworkReadContext context) throws IOException {
    submission.getCompletionStage().toCompletableFuture().complete(null);
    return null;
  }

  @Override
  public NetworkResponse handleException(Throwable ex) {
    submission.getCompletionStage().toCompletableFuture().completeExceptionally(ex);
    return null;
  }
}