com.tigrisdata.db.client.StandardTigrisAsyncDatabase Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2022 Tigris Data, Inc.
 *
 * 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 com.tigrisdata.db.client;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.util.concurrent.ListenableFuture;
import com.tigrisdata.db.api.v1.grpc.Api;
import com.tigrisdata.db.api.v1.grpc.TigrisDBGrpc;
import static com.tigrisdata.db.client.Messages.BEGIN_TRANSACTION_FAILED;
import static com.tigrisdata.db.client.Messages.DESCRIBE_DB_FAILED;
import static com.tigrisdata.db.client.Messages.DROP_COLLECTION_FAILED;
import static com.tigrisdata.db.client.Messages.LIST_COLLECTION_FAILED;
import static com.tigrisdata.db.client.TypeConverter.toBeginTransactionRequest;
import static com.tigrisdata.db.client.TypeConverter.toDatabaseDescription;
import static com.tigrisdata.db.client.TypeConverter.toDropCollectionRequest;
import com.tigrisdata.db.client.error.TigrisException;
import com.tigrisdata.db.type.TigrisCollectionType;
import com.tigrisdata.tools.schema.core.ModelToJsonSchema;
import io.grpc.ManagedChannel;

import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.Executor;
import java.util.function.Predicate;
import java.util.stream.Collectors;

/** Async implementation of TigrisDB */
public class StandardTigrisAsyncDatabase implements TigrisAsyncDatabase {
  private final String databaseName;
  private final TigrisDBGrpc.TigrisDBFutureStub stub;
  private final ManagedChannel channel;
  private final Executor executor;
  private final ObjectMapper objectMapper;
  private final ModelToJsonSchema modelToJsonSchema;

  StandardTigrisAsyncDatabase(
      String databaseName,
      TigrisDBGrpc.TigrisDBFutureStub stub,
      ManagedChannel channel,
      Executor executor,
      ObjectMapper objectMapper,
      ModelToJsonSchema modelToJsonSchema) {
    this.stub = stub;
    this.channel = channel;
    this.databaseName = databaseName;
    this.executor = executor;
    this.objectMapper = objectMapper;
    this.modelToJsonSchema = modelToJsonSchema;
  }

  @Override
  public CompletableFuture> listCollections() {
    ListenableFuture listListenableFuture =
        stub.listCollections(
            Api.ListCollectionsRequest.newBuilder().setDb(this.databaseName).build());
    return Utilities.transformFuture(
        listListenableFuture,
        listCollectionsResponse ->
            listCollectionsResponse.getCollectionsList().stream()
                .map(TypeConverter::toCollectionInfo)
                .collect(Collectors.toList()),
        executor,
        LIST_COLLECTION_FAILED);
  }

  @Override
  public CompletableFuture createOrUpdateCollections(
      Class... collectionModelTypes) {
    CompletableFuture result = new CompletableFuture<>();

    CompletableFuture transactionResponseCompletableFuture =
        beginTransaction(TransactionOptions.DEFAULT_INSTANCE);

    transactionResponseCompletableFuture.whenComplete(
        ((transactionSession, throwable) -> {
          // pass on the error
          if (throwable != null) {
            result.completeExceptionally(throwable);
          }
          CreateOrUpdateCollectionsResponse response = null;
          for (Class collectionModel : collectionModelTypes) {
            try {
              String schemaContent = modelToJsonSchema.toJsonSchema(collectionModel).toString();

              response =
                  transactionSession.createOrUpdateCollections(
                      new TigrisJSONSchema(schemaContent), CollectionOptions.DEFAULT_INSTANCE);
            } catch (Exception ex) {
              result.completeExceptionally(ex);
            }
          }
          if (response != null) {
            result.complete(response);
          }
        }));
    return result;
  }

  @Override
  public CompletableFuture createOrUpdateCollections(
      String[] packagesToScan, Optional>> filter) {
    return this.createOrUpdateCollections(
        Utilities.scanTigrisDBCollectionModels(packagesToScan, filter));
  }

  @Override
  public CompletableFuture dropCollection(String collectionName) {
    // TODO: pull CollectionsOut in API
    ListenableFuture dropCollectionResponseListenableFuture =
        stub.dropCollection(
            toDropCollectionRequest(
                databaseName,
                collectionName,
                CollectionOptions.DEFAULT_INSTANCE,
                Optional.empty()));
    return Utilities.transformFuture(
        dropCollectionResponseListenableFuture,
        response -> new DropCollectionResponse(response.getStatus(), response.getMessage()),
        executor,
        DROP_COLLECTION_FAILED);
  }

  @Override
  public  TigrisAsyncCollection getCollection(
      Class collectionTypeClass) {
    return new StandardTigrisAsyncCollection<>(
        databaseName, collectionTypeClass, channel, executor, objectMapper);
  }

  @Override
  public CompletableFuture beginTransaction(
      TransactionOptions transactionOptions) {
    ListenableFuture beginTransactionResponseListenableFuture =
        stub.beginTransaction(toBeginTransactionRequest(databaseName, transactionOptions));
    return Utilities.transformFuture(
        beginTransactionResponseListenableFuture,
        response ->
            new StandardTransactionSession(
                databaseName, response.getTxCtx(), channel, objectMapper),
        executor,
        BEGIN_TRANSACTION_FAILED);
  }

  @Override
  public CompletableFuture describe() throws TigrisException {
    ListenableFuture describeDatabaseResponseListenableFuture =
        stub.describeDatabase(Api.DescribeDatabaseRequest.newBuilder().setDb(databaseName).build());

    return Utilities.transformFuture(
        describeDatabaseResponseListenableFuture,
        response -> {
          try {
            return toDatabaseDescription(response);
          } catch (TigrisException e) {
            throw new IllegalArgumentException(e);
          }
        },
        executor,
        DESCRIBE_DB_FAILED);
  }

  @Override
  public String name() {
    return databaseName;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;

    StandardTigrisAsyncDatabase that = (StandardTigrisAsyncDatabase) o;

    return Objects.equals(databaseName, that.databaseName);
  }

  @Override
  public int hashCode() {
    return databaseName != null ? databaseName.hashCode() : 0;
  }

  @Override
  public String toString() {
    return "StandardTigrisAsyncDatabase{" + "databaseName='" + databaseName + '\'' + '}';
  }
}