feat: provide the default stream name (#2444)

googleapis · Mar 13, 2024 · fe9c3ae · fe9c3ae
1 parent 6e267fb
commit fe9c3ae
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/...loud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/StreamWriter.java b/...loud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/StreamWriter.java
@@ -561,6 +561,11 @@ public static void setMaxRequestCallbackWaitTime(Duration waitTime) {
     ConnectionWorker.MAXIMUM_REQUEST_CALLBACK_WAIT_TIME = waitTime;
   }
 
+  /** @return the default stream name associated with tableName */
+  public static String getDefaultStreamName(TableName tableName) {
+    return tableName + defaultStreamMatching;
+  }
+
   long getCreationTimestamp() {
     return creationTimestamp;
   }

diff --git a/...-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java b/...-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java
@@ -2224,4 +2224,14 @@ public void testExclusiveAppendSuccessAndNonRetryableError() throws Exception {
         Status.Code.INVALID_ARGUMENT,
         ((StatusRuntimeException) ex.getCause()).getStatus().getCode());
   }
+
+  @Test
+  public void testGetDefaultStreamName() {
+    TableName tableName = TableName.of("projectId", "datasetId", "tableId");
+
+    String actualDefaultName = StreamWriter.getDefaultStreamName(tableName);
+
+    assertEquals(
+        "projects/projectId/datasets/datasetId/tables/tableId/_default", actualDefaultName);
+  }
 }