com.google.cloud.bigtable.hbase.wrappers.veneer.BigtableVeneerApi 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.7
/*
 * 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.InternalApi;
import com.google.api.gax.grpc.InstantiatingGrpcChannelProvider;
import com.google.cloud.bigtable.admin.v2.BigtableInstanceAdminClient;
import com.google.cloud.bigtable.admin.v2.BigtableInstanceAdminSettings;
import com.google.cloud.bigtable.admin.v2.BigtableTableAdminClient;
import com.google.cloud.bigtable.data.v2.BigtableDataClient;
import com.google.cloud.bigtable.data.v2.stub.EnhancedBigtableStubSettings;
import com.google.cloud.bigtable.hbase.wrappers.AdminClientWrapper;
import com.google.cloud.bigtable.hbase.wrappers.BigtableApi;
import com.google.cloud.bigtable.hbase.wrappers.DataClientWrapper;
import com.google.cloud.bigtable.metrics.BigtableClientMetrics;
import com.google.cloud.bigtable.metrics.BigtableClientMetrics.MetricLevel;
import com.google.cloud.bigtable.metrics.Counter;
import java.io.IOException;

/** For internal use only - public for technical reasons. */
@InternalApi("For internal usage only")
public class BigtableVeneerApi extends BigtableApi {

  private final Counter activeSessions =
      BigtableClientMetrics.counter(MetricLevel.Info, "session.active");

  private static final SharedDataClientWrapperFactory sharedClientFactory =
      new SharedDataClientWrapperFactory();
  private final DataClientWrapper dataClientWrapper;
  private final AdminClientWrapper adminClientWrapper;
  private final int channelPoolSize;

  public BigtableVeneerApi(BigtableHBaseVeneerSettings settings) throws IOException {
    super(settings);

    // active channel count is hard coded at client creation time based on the setting. If
    // transportChannelProvider in the data setting is not InstantiatingGrpcChannelProvider, this
    // count wil not be present. If channel pool caching is enabled, channel pool size is calculated
    // in SharedDataClientWrapperFactory to avoid incrementing/decrementing the same channel
    // multiple times for the same key.
    if (settings.isChannelPoolCachingEnabled()) {
      dataClientWrapper = sharedClientFactory.createDataClient(settings);
      channelPoolSize = 0;
    } else {
      dataClientWrapper =
          new DataClientVeneerApi(
              BigtableDataClient.create(settings.getDataSettings()), settings.getClientTimeouts());
      channelPoolSize = getChannelPoolSize(settings.getDataSettings().getStubSettings());
      for (int i = 0; i < channelPoolSize; i++) {
        BigtableClientMetrics.counter(MetricLevel.Info, "grpc.channel.active").inc();
      }
    }
    BigtableInstanceAdminSettings instanceAdminSettings = settings.getInstanceAdminSettings();
    adminClientWrapper =
        new AdminClientVeneerApi(
            BigtableTableAdminClient.create(settings.getTableAdminSettings()),
            BigtableInstanceAdminClient.create(instanceAdminSettings));
    activeSessions.inc();
  }

  @Override
  public AdminClientWrapper getAdminClient() {
    return adminClientWrapper;
  }

  @Override
  public DataClientWrapper getDataClient() {
    return dataClientWrapper;
  }

  @Override
  public void close() throws IOException {
    dataClientWrapper.close();
    adminClientWrapper.close();
    activeSessions.dec();
    for (int i = 0; i < channelPoolSize; i++) {
      BigtableClientMetrics.counter(MetricLevel.Info, "grpc.channel.active").dec();
    }
  }

  static int getChannelPoolSize(EnhancedBigtableStubSettings stubSettings) {
    if (stubSettings.getTransportChannelProvider() instanceof InstantiatingGrpcChannelProvider) {
      InstantiatingGrpcChannelProvider channelProvider =
          (InstantiatingGrpcChannelProvider) stubSettings.getTransportChannelProvider();
      return channelProvider.toBuilder().getPoolSize();
    }
    return 0;
  }
}