feat: Partial Projection of Table Metadata

googleapis · Jun 14, 2023 · 0985520 · 0985520
1 parent d178593
commit 0985520
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 4 deletions.
diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/Table.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/Table.java
@@ -49,10 +49,10 @@ public static class Builder extends TableInfo.Builder {
     private final BigQuery bigquery;
     private final TableInfo.BuilderImpl infoBuilder;
 
-    Builder(BigQuery bigquery, TableId tableId, TableDefinition defintion) {
+    Builder(BigQuery bigquery, TableId tableId, TableDefinition definition) {
       this.bigquery = bigquery;
       this.infoBuilder = new TableInfo.BuilderImpl();
-      this.infoBuilder.setTableId(tableId).setDefinition(defintion);
+      this.infoBuilder.setTableId(tableId).setDefinition(definition);
     }
 
     Builder(Table table) {

diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/BigQueryRpc.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/BigQueryRpc.java
@@ -40,7 +40,7 @@
 @InternalExtensionOnly
 public interface BigQueryRpc extends ServiceRpc {
 
-  // These options are part of the Google Cloud BigQuery query parameters
+  // These options are part of the Google Cloud BigQuery query parameters.
   enum Option {
     FIELDS("fields"),
     DELETE_CONTENTS("deleteContents"),
@@ -56,7 +56,8 @@ enum Option {
     START_INDEX("startIndex"),
     STATE_FILTER("stateFilter"),
     TIMEOUT("timeoutMs"),
-    REQUESTED_POLICY_VERSION("requestedPolicyVersion");
+    REQUESTED_POLICY_VERSION("requestedPolicyVersion"),
+    TABLE_METADATA_VIEW("view");
 
     private final String value;
 

diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java
@@ -294,6 +294,7 @@ public Table getTable(
           .get(projectId, datasetId, tableId)
           .setPrettyPrint(false)
           .setFields(Option.FIELDS.getString(options))
+          .setView("BASIC")
           .execute();
     } catch (IOException ex) {
       BigQueryException serviceException = translate(ex);
@@ -304,6 +305,13 @@ public Table getTable(
     }
   }
 
+  private String getTableMetadataOption(Map<Option, ?> options) {
+    if (options.containsKey("view")) {
+      return options.get("view").toString();
+    }
+    return "STORAGE_STATS";
+  }
+
   @Override
   public Tuple<String, Iterable<Table>> listTables(
       String projectId, String datasetId, Map<Option, ?> options) {

diff --git a/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java b/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java
@@ -1516,6 +1516,8 @@ public void testCreateAndGetTable() {
     assertEquals(DATASET, createdTable.getTableId().getDataset());
     assertEquals(tableName, createdTable.getTableId().getTable());
     Table remoteTable = bigquery.getTable(DATASET, tableName);
+    System.out.println("Penguins");
+    System.out.println(remoteTable);
     assertNotNull(remoteTable);
     assertTrue(remoteTable.getDefinition() instanceof StandardTableDefinition);
     assertEquals(createdTable.getTableId(), remoteTable.getTableId());