Skip to content

Commit 1891383

Browse files
Bigtable: be consistent and don't expose absolute table names
1 parent f977c8a commit 1891383

3 files changed

Lines changed: 20 additions & 23 deletions

File tree

google-cloud-clients/google-cloud-bigtable-admin/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClient.java

Lines changed: 15 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -383,38 +383,37 @@ public ApiFuture<Table> getTableAsync(String tableId) {
383383
}
384384

385385
/**
386-
* Lists all TableNames in the instance.
386+
* Lists all table ids in the instance.
387387
*
388388
* <p>Sample code:
389389
*
390390
* <pre>{@code
391-
* List<TableName> tableNames = client.listTables();
392-
* for(TableName name : tableNames) {
391+
* List<String> tableIds = client.listTables();
392+
* for(String tableId: tableIds) {
393393
* System.out.println(name.getTable());
394394
* }
395395
* }</pre>
396396
*/
397-
// TODO(igorbernstein2): consider changing this method to use relative table ids.
398397
@SuppressWarnings("WeakerAccess")
399-
public List<TableName> listTables() {
398+
public List<String> listTables() {
400399
return awaitFuture(listTablesAsync());
401400
}
402401

403402
/**
404-
* Asynchronously lists all TableNames in the instance.
403+
* Asynchronously lists all table ids in the instance.
405404
*
406405
* <p>Sample code:
407406
*
408407
* <pre>{@code
409-
* ApiFuture<List<TableName>> listFuture = client.listTables();
408+
* ApiFuture<List<String>> listFuture = client.listTables();
410409
*
411410
* ApiFutures.addCallback(
412411
* listFuture,
413-
* new ApiFutureCallback<List<TableName>>() {
414-
* public void onSuccess(List<TableName> tableNames) {
412+
* new ApiFutureCallback<List<String>>() {
413+
* public void onSuccess(List<String> tableIds) {
415414
* System.out.println("Got list of tables:");
416-
* for (TableName name : tableNames) {
417-
* System.out.println(name.getTable());
415+
* for (String tableId : tableIds) {
416+
* System.out.println(tableId);
418417
* }
419418
* }
420419
*
@@ -426,9 +425,8 @@ public List<TableName> listTables() {
426425
* );
427426
* }</pre>
428427
*/
429-
// TODO(igorbernstein2): consider changing this method to use relative table ids.
430428
@SuppressWarnings("WeakerAccess")
431-
public ApiFuture<List<TableName>> listTablesAsync() {
429+
public ApiFuture<List<String>> listTablesAsync() {
432430
ListTablesRequest request = ListTablesRequest.newBuilder().setParent(instanceName.toString())
433431
.build();
434432

@@ -479,12 +477,12 @@ public ApiFuture<List<com.google.bigtable.admin.v2.Table>> apply(
479477

480478
// Wrap all of the accumulated protos.
481479
return ApiFutures.transform(allProtos,
482-
new ApiFunction<List<com.google.bigtable.admin.v2.Table>, List<TableName>>() {
480+
new ApiFunction<List<com.google.bigtable.admin.v2.Table>, List<String>>() {
483481
@Override
484-
public List<TableName> apply(List<com.google.bigtable.admin.v2.Table> protos) {
485-
List<TableName> results = Lists.newArrayListWithCapacity(protos.size());
482+
public List<String> apply(List<com.google.bigtable.admin.v2.Table> protos) {
483+
List<String> results = Lists.newArrayListWithCapacity(protos.size());
486484
for (com.google.bigtable.admin.v2.Table proto : protos) {
487-
results.add(TableName.parse(proto.getName()));
485+
results.add(TableName.parse(proto.getName()).getTable());
488486
}
489487
return results;
490488
}

google-cloud-clients/google-cloud-bigtable-admin/src/test/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -241,12 +241,12 @@ public void testListTables() {
241241
);
242242

243243
// Execute
244-
List<TableName> actualResults = adminClient.listTables();
244+
List<String> actualResults = adminClient.listTables();
245245

246246
// Verify
247-
List<TableName> expectedResults = Lists.newArrayList();
247+
List<String> expectedResults = Lists.newArrayList();
248248
for (com.google.bigtable.admin.v2.Table expectedProto : expectedProtos) {
249-
expectedResults.add(TableName.parse(expectedProto.getName()));
249+
expectedResults.add(TableName.parse(expectedProto.getName()).getTable());
250250
}
251251

252252
assertThat(actualResults).containsExactlyElementsIn(expectedResults);

google-cloud-clients/google-cloud-bigtable-admin/src/test/java/com/google/cloud/bigtable/admin/v2/it/BigtableTableAdminClientIT.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import static org.junit.Assert.assertTrue;
2323

2424
import com.google.bigtable.admin.v2.InstanceName;
25-
import com.google.bigtable.admin.v2.TableName;
2625
import com.google.cloud.bigtable.admin.v2.BigtableTableAdminClient;
2726
import com.google.cloud.bigtable.admin.v2.models.ColumnFamily;
2827
import com.google.cloud.bigtable.admin.v2.models.CreateTableRequest;
@@ -204,7 +203,7 @@ public void listTables() {
204203

205204
try {
206205
tableAdmin.createTable(CreateTableRequest.of(tableId));
207-
List<TableName> tables = tableAdmin.listTables();
206+
List<String> tables = tableAdmin.listTables();
208207
assertNotNull(tables);
209208
assertFalse("List tables did not return any tables", tables.isEmpty());
210209
} finally {
@@ -218,7 +217,7 @@ public void listTablesAsync() throws Exception {
218217

219218
try {
220219
tableAdmin.createTable(CreateTableRequest.of(tableId));
221-
List<TableName> tables = tableAdmin.listTablesAsync().get();
220+
List<String> tables = tableAdmin.listTablesAsync().get();
222221
assertNotNull(tables);
223222
assertFalse("List tables did not return any tables", tables.isEmpty());
224223
} finally {

0 commit comments

Comments
 (0)