chore: Add missing storage related fields to Table, TableInfo and Sta…

…ndardTableDefinition
googleapis · Apr 29, 2023 · df85341 · df85341
1 parent 909a574
commit df85341
Show file tree

Hide file tree

Showing 6 changed files with 402 additions and 0 deletions.
diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/StandardTableDefinition.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/StandardTableDefinition.java
@@ -132,6 +132,20 @@ public abstract static class Builder
 
     public abstract Builder setNumLongTermBytes(Long numLongTermBytes);
 
+    public abstract Builder setNumTimeTravelPhysicalBytes(Long numTimeTravelPhysicalBytes);
+
+    public abstract Builder setNumTotalLogicalBytes(Long numTotalLogicalBytes);
+
+    public abstract Builder setNumActiveLogicalBytes(Long numActiveLogicalBytes);
+
+    public abstract Builder setNumLongTermLogicalBytes(Long numLongTermLogicalBytes);
+
+    public abstract Builder setNumTotalPhysicalBytes(Long numTotalPhysicalBytes);
+
+    public abstract Builder setNumActivePhysicalBytes(Long numActivePhysicalBytes);
+
+    public abstract Builder setNumLongTermPhysicalBytes(Long numLongTermPhysicalBytes);
+
     public abstract Builder setNumRows(Long numRows);
 
     public abstract Builder setLocation(String location);
@@ -178,6 +192,69 @@ public abstract static class Builder
   @Nullable
   public abstract Long getNumLongTermBytes();
 
+  /**
+   * Returns the number of time travel physical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumTimeTravelPhysicalBytes();
+
+  /**
+   * Returns the number of total logical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumTotalLogicalBytes();
+
+  /**
+   * Returns the number of active logical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumActiveLogicalBytes();
+
+  /**
+   * Returns the number of long term logical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumLongTermLogicalBytes();
+
+  /**
+   * Returns the number of total physical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumTotalPhysicalBytes();
+
+  /**
+   * Returns the number of active physical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumActivePhysicalBytes();
+
+  /**
+   * Returns the number of long term physical bytes.
+   *
+   * @see <a href="https://cloud.google.com/bigquery/pricing#storage">Storage
+   *     Pricing</a>
+   */
+  @Nullable
+  public abstract Long getNumLongTermPhysicalBytes();
+
   /** Returns the number of rows in this table, excluding any data in the streaming buffer. */
   @Nullable
   public abstract Long getNumRows();
@@ -246,6 +323,13 @@ Table toPb() {
     }
     tablePb.setNumBytes(getNumBytes());
     tablePb.setNumLongTermBytes(getNumLongTermBytes());
+    tablePb.setNumTimeTravelPhysicalBytes(getNumTimeTravelPhysicalBytes());
+    tablePb.setNumTotalLogicalBytes(getNumTotalLogicalBytes());
+    tablePb.setNumActiveLogicalBytes(getNumActiveLogicalBytes());
+    tablePb.setNumLongTermLogicalBytes(getNumLongTermLogicalBytes());
+    tablePb.setNumTotalPhysicalBytes(getNumTotalPhysicalBytes());
+    tablePb.setNumActivePhysicalBytes(getNumActivePhysicalBytes());
+    tablePb.setNumLongTermPhysicalBytes(getNumLongTermPhysicalBytes());
     tablePb.setLocation(getLocation());
     if (getStreamingBuffer() != null) {
       tablePb.setStreamingBuffer(getStreamingBuffer().toPb());
@@ -296,6 +380,27 @@ static StandardTableDefinition fromPb(Table tablePb) {
     if (tablePb.getNumLongTermBytes() != null) {
       builder.setNumLongTermBytes(tablePb.getNumLongTermBytes());
     }
+    if (tablePb.getNumTimeTravelPhysicalBytes() != null) {
+      builder.setNumTimeTravelPhysicalBytes(tablePb.getNumTimeTravelPhysicalBytes());
+    }
+    if (tablePb.getNumTotalLogicalBytes() != null) {
+      builder.setNumTotalLogicalBytes(tablePb.getNumTotalLogicalBytes());
+    }
+    if (tablePb.getNumActiveLogicalBytes() != null) {
+      builder.setNumActiveLogicalBytes(tablePb.getNumActiveLogicalBytes());
+    }
+    if (tablePb.getNumLongTermLogicalBytes() != null) {
+      builder.setNumLongTermLogicalBytes(tablePb.getNumLongTermLogicalBytes());
+    }
+    if (tablePb.getNumTotalPhysicalBytes() != null) {
+      builder.setNumTotalPhysicalBytes(tablePb.getNumTotalPhysicalBytes());
+    }
+    if (tablePb.getNumActivePhysicalBytes() != null) {
+      builder.setNumActivePhysicalBytes(tablePb.getNumActivePhysicalBytes());
+    }
+    if (tablePb.getNumLongTermPhysicalBytes() != null) {
+      builder.setNumLongTermPhysicalBytes(tablePb.getNumLongTermPhysicalBytes());
+    }
     return builder.setNumBytes(tablePb.getNumBytes()).setLocation(tablePb.getLocation()).build();
   }
 }
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
@@ -114,6 +114,48 @@ Builder setNumLongTermBytes(Long numLongTermBytes) {
       return this;
     }
 
+    @Override
+    Builder setNumTimeTravelPhysicalBytes(Long numTimeTravelPhysicalBytes) {
+      infoBuilder.setNumTimeTravelPhysicalBytes(numTimeTravelPhysicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumTotalLogicalBytes(Long numTotalLogicalBytes) {
+      infoBuilder.setNumTotalLogicalBytes(numTotalLogicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumActiveLogicalBytes(Long numActiveLogicalBytes) {
+      infoBuilder.setNumActiveLogicalBytes(numActiveLogicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumLongTermLogicalBytes(Long numLongTermLogicalBytes) {
+      infoBuilder.setNumLongTermLogicalBytes(numLongTermLogicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumTotalPhysicalBytes(Long numTotalPhysicalBytes) {
+      infoBuilder.setNumTotalPhysicalBytes(numTotalPhysicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumActivePhysicalBytes(Long numActivePhysicalBytes) {
+      infoBuilder.setNumActivePhysicalBytes(numActivePhysicalBytes);
+      return this;
+    }
+
+    @Override
+    Builder setNumLongTermPhysicalBytes(Long numLongTermPhysicalBytes) {
+      infoBuilder.setNumLongTermPhysicalBytes(numLongTermPhysicalBytes);
+      return this;
+    }
+
     @Override
     Builder setNumRows(BigInteger numRows) {
       infoBuilder.setNumRows(numRows);