com.google.cloud.bigtable.hbase.wrappers.veneer.BulkMutationVeneerApi Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bigtable-hbase Show documentation
This project contains artifacts that adapt bigtable client to work with hbase.
There is a newer version: 2.14.8
/*
 * Copyright 2020 Google LLC
 *
 * 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.google.cloud.bigtable.hbase.wrappers.veneer;

import com.google.api.core.ApiFuture;
import com.google.api.core.InternalApi;
import com.google.api.gax.batching.Batcher;
import com.google.cloud.bigtable.data.v2.models.RowMutationEntry;
import com.google.cloud.bigtable.hbase.wrappers.BulkMutationWrapper;
import com.google.cloud.bigtable.metrics.BigtableClientMetrics;
import com.google.cloud.bigtable.metrics.BigtableClientMetrics.MetricLevel;
import com.google.cloud.bigtable.metrics.Meter;
import com.google.common.base.Preconditions;
import java.io.IOException;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

/** For internal use only - public for technical reasons. */
@InternalApi("For internal usage only")
public class BulkMutationVeneerApi implements BulkMutationWrapper {

  private final Meter mutationAdded =
      BigtableClientMetrics.meter(MetricLevel.Info, "bulk-mutator.mutations.added");
  private final Batcher bulkMutateBatcher;

  // If set to 0, timeout is disabled. Negative value is not accepted.
  private final long closeTimeoutMilliseconds;

  BulkMutationVeneerApi(
      Batcher bulkMutateBatcher, long closeTimeoutMilliseconds) {
    this.bulkMutateBatcher = bulkMutateBatcher;
    Preconditions.checkArgument(closeTimeoutMilliseconds >= 0);
    this.closeTimeoutMilliseconds = closeTimeoutMilliseconds;
  }

  /** {@inheritDoc} */
  @Override
  public synchronized ApiFuture add(RowMutationEntry rowMutation) {
    Preconditions.checkNotNull(rowMutation, "mutation details cannot be null");
    mutationAdded.mark();
    return bulkMutateBatcher.add(rowMutation);
  }

  /** {@inheritDoc} */
  @Override
  public synchronized void sendUnsent() {
    bulkMutateBatcher.sendOutstanding();
  }

  /** {@inheritDoc} */
  @Override
  public synchronized void flush() {
    try {
      bulkMutateBatcher.flush();
    } catch (InterruptedException ex) {
      throw new RuntimeException("Could not complete RPC for current Batch", ex);
    }
  }

  @Override
  public synchronized void close() throws IOException {
    try {
      ApiFuture future = bulkMutateBatcher.closeAsync();
      if (closeTimeoutMilliseconds > 0) {
        future.get(closeTimeoutMilliseconds, TimeUnit.MILLISECONDS);
      } else {
        future.get();
      }
    } catch (InterruptedException | ExecutionException e) {
      throw new IOException("Could not close the bulk mutation Batcher", e);
    } catch (TimeoutException e) {
      throw new IOException("Cloud not close the bulk mutation Batcher, timed out in close()", e);
    }
  }
}