kvd.client.KvdContains Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kvd-client Show documentation
key value database
The newest version!
/*
 * Copyright 2021 Andre Gebers
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package kvd.client;

import java.util.concurrent.CompletableFuture;
import java.util.function.Consumer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import kvd.common.KvdException;
import kvd.common.packet.Packets;
import kvd.common.packet.proto.Packet;
import kvd.common.packet.proto.PacketType;

class KvdContains implements Abortable {

  private static final Logger log = LoggerFactory.getLogger(KvdContains.class);

  private CompletableFuture future = new CompletableFuture();

  private ClientBackend backend;

  private byte[] key;

  private int channelId;

  private Consumer closeListener;

  private int txId;

  public KvdContains(ClientBackend backend, int txId, byte[] key, Consumer closeListener) {
    this.backend = backend;
    this.txId = txId;
    this.key = key;
    this.closeListener = closeListener;
  }

  public void start() {
    channelId = backend.createChannel(this::receive);
    try {
      backend.sendAsync(Packets.packet(PacketType.CONTAINS_REQUEST, channelId, txId, key));
    } catch(Exception e) {
      try {
        close();
      } catch(Exception e2) {
        // ignore
      }
      throw new KvdException("contains failed", e);
    }
  }

  @Override
  public void abort() {
    future.completeExceptionally(new KvdException("aborted"));
    close();
  }

  private void close() {
    backend.closeChannel(channelId);
    this.closeListener.accept(this);
  }

  public void receive(Packet packet) {
    try {
      close();
    } catch(Exception e) {
      log.error("contains close failed", e);
    }
    if(PacketType.CONTAINS_RESPONSE.equals(packet.getType())) {
      byte[] buf = packet.getByteBody().toByteArray();
      if((buf != null) && (buf.length >= 1)) {
        future.complete((buf[0] == 1));
      } else {
        log.error("invalid response");
        future.completeExceptionally(new KvdException("invalid response"));
      }
    } else if(PacketType.CONTAINS_ABORT.equals(packet.getType())) {
      future.completeExceptionally(new KvdException("server abort"));
    } else {
      log.error("received unexpected packet '{}'", packet.getType());
      future.completeExceptionally(new KvdException("received unexpected packet " + packet.getType()));
    }
  }

  public CompletableFuture getFuture() {
    return future;
  }

  @Override
  public String toString() {
    return "CONTAINS " + key;
  }

}