/** * <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); } } }