.

googleapis · Apr 3, 2023 · 63cb115 · 63cb115
1 parent 3fea877
commit 63cb115
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/samples/snippets/src/main/java/com/example/bigquerystorage/WriteToDefaultStream.java b/samples/snippets/src/main/java/com/example/bigquerystorage/WriteToDefaultStream.java
@@ -151,7 +151,8 @@ public void initialize(TableName parentTable)
       // For more information about JsonStreamWriter, see:
       // https://googleapis.dev/java/google-cloud-bigquerystorage/latest/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.html
       streamWriter =
-          JsonStreamWriter.newBuilder(parentTable.toString(), BigQueryWriteClient.create()).build();
+          JsonStreamWriter.newBuilder(streamWriter.getStreamName(),
+              BigQueryWriteClient.create()).build();
     }
 
     public void append(AppendContext appendContext)
@@ -163,7 +164,7 @@ public void append(AppendContext appendContext)
         // If stream writer is premenantly failed, try recreate a stream writer.
         if (streamWriter.isClosed() && recreateCount.getAndIncrement() < MAX_RECREATE_COUNT) {
           streamWriter =
-              JsonStreamWriter.newBuilder(parentTable.toString(),
+              JsonStreamWriter.newBuilder(streamWriter.get,
                   BigQueryWriteClient.create()).build();
           this.error = null;
         }