How to use

toTableNameStr

method

in

com.google.cloud.bigtable.grpc.BigtableInstanceName

/**
 * <p>toBigtableName.</p>
 *
 * @param tableName a {@link org.apache.hadoop.hbase.TableName} object.
 * @return a {@link java.lang.String} object.
 */
protected String toBigtableName(TableName tableName) {
 return bigtableInstanceName.toTableNameStr(tableName.getNameAsString());
}

/**
 * <p>toBigtableName.</p>
 *
 * @param tableName a {@link org.apache.hadoop.hbase.TableName} object.
 * @return a {@link java.lang.String} object.
 */
private String toBigtableName(TableName tableName) {
 return bigtableInstanceName.toTableNameStr(tableName.getNameAsString());
}

/**
 * <p>toBigtableName.</p>
 *
 * @param tableName a {@link org.apache.hadoop.hbase.TableName} object.
 * @return a {@link java.lang.String} object.
 */
protected String toBigtableName(TableName tableName) {
 return bigtableInstanceName.toTableNameStr(tableName.getNameAsString());
}

/**
 * <p>toTableName.</p>
 *
 * @param tableId a {@link java.lang.String} object.
 * @return a {@link com.google.cloud.bigtable.grpc.BigtableTableName} object.
 */
public BigtableTableName toTableName(String tableId) {
 return new BigtableTableName(toTableNameStr(tableId));
}

/**
 * <p>toBigtableName.</p>
 *
 * @param tableName a {@link org.apache.hadoop.hbase.TableName} object.
 * @return a {@link java.lang.String} object.
 */
private String toBigtableName(TableName tableName) {
 return bigtableInstanceName.toTableNameStr(tableName.getNameAsString());
}

/** {@inheritDoc} */
@Override
public ListenableFuture<Void> deleteTableAsync(String tableId) {
 DeleteTableRequest request = DeleteTableRequest.newBuilder()
   .setName(instanceName.toTableNameStr(tableId))
   .build();
 return Futures.transform(adminClient.deleteTableAsync(request), new Function<Empty, Void>() {
  @Override
  public Void apply(Empty empty) {
   return null;
  }
 }, MoreExecutors.directExecutor());
}

/** {@inheritDoc} */
@Override
public ListenableFuture<Table> getTableAsync(String tableId) {
 GetTableRequest requestProto = GetTableRequest.newBuilder()
   .setName(instanceName.toTableNameStr(tableId))
   .build();
 return Futures.transform(adminClient.getTableAsync(requestProto),
   new Function<com.google.bigtable.admin.v2.Table, Table>() {
    @Override
    public Table apply(com.google.bigtable.admin.v2.Table tableProto) {
     return Table.fromProto(tableProto);
    }
   }, MoreExecutors.directExecutor());
}

/** {@inheritDoc} */
@Override
public CompletableFuture<Void> deleteTable(TableName tableName) {
 DeleteTableRequest request = DeleteTableRequest.newBuilder()
   .setName(bigtableInstanceName.toTableNameStr(tableName.getNameAsString()))
   .build();
 return bigtableTableAdminClient.deleteTableAsync(request)
   .thenAccept(r -> disabledTables.remove(tableName));
}

/** {@inheritDoc} */
@Override
public CompletableFuture<Void> deleteTable(TableName tableName) {
 DeleteTableRequest request = DeleteTableRequest.newBuilder()
   .setName(bigtableInstanceName.toTableNameStr(tableName.getNameAsString()))
   .build();
 return bigtableTableAdminClient.deleteTableAsync(request)
   .thenAccept(r -> disabledTables.remove(tableName));
}

 private DropRowRangeRequest buildDropRowRangeRequest(String tableId, String rowKeyPrefix) {
  DropRowRangeRequest.Builder dropRequestProtoBuiler =
    DropRowRangeRequest.newBuilder()
      .setName(instanceName.toTableNameStr(tableId));

  if (!Strings.isNullOrEmpty(rowKeyPrefix)) {
   dropRequestProtoBuiler
     .setDeleteAllDataFromTable(false)
     .setRowKeyPrefix(ByteString.copyFromUtf8(rowKeyPrefix));
  } else {
   dropRequestProtoBuiler.setDeleteAllDataFromTable(true);
  }

  return dropRequestProtoBuiler.build();
 }
}

/** {@inheritDoc} */
@Override
public void deleteTable(String tableId) {
 DeleteTableRequest request = DeleteTableRequest.newBuilder()
   .setName(instanceName.toTableNameStr(tableId))
   .build();
 adminClient.deleteTable(request);
}

/** {@inheritDoc} */
@Override
public Table getTable(String tableId) {
 GetTableRequest requestProto = GetTableRequest.newBuilder()
   .setName(instanceName.toTableNameStr(tableId))
   .build();
 return Table.fromProto(adminClient.getTable(requestProto));
}

@Override
public List<HRegionInfo> getAllRegionInfos(TableName tableName) throws IOException {
 ServerName serverName = ServerName.valueOf(options.getDataHost(), options.getPort(), 0);
 SampleRowKeysRequest.Builder request = SampleRowKeysRequest.newBuilder();
 request.setTableName(options.getInstanceName().toTableNameStr(tableName.getNameAsString()));
 List<SampleRowKeysResponse> sampleRowKeyResponse = this.session.getDataClient().sampleRowKeys(request.build());
 return getSampledRowKeysAdapter(tableName, serverName).adaptResponse(sampleRowKeyResponse)
   .stream()
   .map(location -> location.getRegionInfo())
   .collect(Collectors.toCollection(CopyOnWriteArrayList::new));
}

@Override
public CompletableFuture<Void> truncateTable(TableName tableName, boolean preserveSplits) {
 if (!preserveSplits) {
  LOG.info("truncate will preserveSplits. The passed in variable is ignored.");
 }
 DropRowRangeRequest request = DropRowRangeRequest
   .newBuilder()
   .setDeleteAllDataFromTable(true)
   .setName(bigtableInstanceName.toTableNameStr(tableName.getNameAsString()))
   .build();
 return bigtableTableAdminClient.dropRowRangeAsync(request).thenApply(r -> null);
}

@Override
public List<HRegionInfo> getAllRegionInfos(TableName tableName) throws IOException {
 ServerName serverName = ServerName.valueOf(options.getDataHost(), options.getPort(), 0);
 SampleRowKeysRequest.Builder request = SampleRowKeysRequest.newBuilder();
 request.setTableName(options.getInstanceName().toTableNameStr(tableName.getNameAsString()));
 List<SampleRowKeysResponse> sampleRowKeyResponse = this.session.getDataClient().sampleRowKeys(request.build());
 return getSampledRowKeysAdapter(tableName, serverName).adaptResponse(sampleRowKeyResponse)
   .stream()
   .map(location -> location.getRegionInfo())
   .collect(Collectors.toCollection(CopyOnWriteArrayList::new));
}

@Override
public CompletableFuture<Void> truncateTable(TableName tableName, boolean preserveSplits) {
 if (!preserveSplits) {
  LOG.info("truncate will preserveSplits. The passed in variable is ignored.");
 }
 DropRowRangeRequest request = DropRowRangeRequest
   .newBuilder()
   .setDeleteAllDataFromTable(true)
   .setName(bigtableInstanceName.toTableNameStr(tableName.getNameAsString()))
   .build();
 return bigtableTableAdminClient.dropRowRangeAsync(request).thenApply(r -> null);
}

@Override
public CompletableFuture<Void> snapshot(String snapshotName, TableName tableName) {
 return CompletableFuture.supplyAsync(() -> {
   try {
    return SnapshotTableRequest.newBuilder()
        .setCluster(getSnapshotClusterName().toString())
        .setSnapshotId(snapshotName)
        .setName(options.getInstanceName().toTableNameStr(tableName.getNameAsString())).build();
   } catch (IOException e) {
    throw new CompletionException(e);
   }
  }).thenCompose(
    c -> bigtableTableAdminClient.snapshotTableAsync(c).thenApply(r -> null));
}

@Override
public CompletableFuture<Void> snapshot(String snapshotName, TableName tableName) {
 return CompletableFuture.supplyAsync(() -> {
   try {
    return SnapshotTableRequest.newBuilder()
        .setCluster(getSnapshotClusterName().toString())
        .setSnapshotId(snapshotName)
        .setName(options.getInstanceName().toTableNameStr(tableName.getNameAsString())).build();
   } catch (IOException e) {
    throw new CompletionException(e);
   }
  }).thenCompose(
    c -> bigtableTableAdminClient.snapshotTableAsync(c).thenApply(r -> null));
}

@After
public void tearDown() throws Exception {
 final String tableName = bigtableOptions.getInstanceName().toTableNameStr(tableId);
 deleteTable(tableName);
 session.close();
}

 @Override
 public List<SampleRowKeysResponse> getSampleRowKeys(BigtableSource source) throws IOException {
  try (BigtableSession session = new BigtableSession(options)) {
   SampleRowKeysRequest request =
     SampleRowKeysRequest.newBuilder()
       .setTableName(options.getInstanceName().toTableNameStr(source.getTableId().get()))
       .build();
   return session.getDataClient().sampleRowKeys(request);
  }
 }
}