Skip to content

Commit f69e7fa

Browse files
BigQuery: Add support for retreiving labels (googleapis#3879)
1 parent 19fe6eb commit f69e7fa

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/BigQuery.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ enum TableField implements FieldSelector {
8989
EXTERNAL_DATA_CONFIGURATION("externalDataConfiguration"),
9090
FRIENDLY_NAME("friendlyName"),
9191
ID("id"),
92+
LABELS("labels"),
9293
LAST_MODIFIED_TIME("lastModifiedTime"),
9394
LOCATION("location"),
9495
NUM_BYTES("numBytes"),

google-cloud-clients/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -420,16 +420,19 @@ public void testCreateAndGetTableWithSelectedField() {
420420
String tableName = "test_create_and_get_selected_fields_table";
421421
TableId tableId = TableId.of(DATASET, tableName);
422422
StandardTableDefinition tableDefinition = StandardTableDefinition.of(TABLE_SCHEMA);
423-
Table createdTable = bigquery.create(TableInfo.of(tableId, tableDefinition));
423+
Table createdTable = bigquery.create(TableInfo.newBuilder(tableId, tableDefinition)
424+
.setLabels(Collections.singletonMap("a", "b"))
425+
.build());
424426
assertNotNull(createdTable);
425427
assertEquals(DATASET, createdTable.getTableId().getDataset());
426428
assertEquals(tableName, createdTable.getTableId().getTable());
427429
Table remoteTable = bigquery.getTable(DATASET, tableName,
428-
TableOption.fields(TableField.CREATION_TIME));
430+
TableOption.fields(TableField.CREATION_TIME, TableField.LABELS));
429431
assertNotNull(remoteTable);
430432
assertTrue(remoteTable.getDefinition() instanceof StandardTableDefinition);
431433
assertEquals(createdTable.getTableId(), remoteTable.getTableId());
432434
assertEquals(TableDefinition.Type.TABLE, remoteTable.getDefinition().getType());
435+
assertThat(remoteTable.getLabels()).containsExactly("a", "b");
433436
assertNotNull(remoteTable.getCreationTime());
434437
assertNull(remoteTable.getDefinition().getSchema());
435438
assertNull(remoteTable.getLastModifiedTime());

0 commit comments

Comments
 (0)