From 5655186de7386afa9a3dae12b79d783da0301326 Mon Sep 17 00:00:00 2001 From: aIbrahiim Date: Thu, 9 Apr 2026 15:43:39 +0200 Subject: [PATCH] Retry transient NOT_FOUND on Storage API default stream --- .../sdk/io/gcp/bigquery/StorageApiWriteUnshardedRecords.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/StorageApiWriteUnshardedRecords.java b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/StorageApiWriteUnshardedRecords.java index bb37bd4c1735..2fe10453e256 100644 --- a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/StorageApiWriteUnshardedRecords.java +++ b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/StorageApiWriteUnshardedRecords.java @@ -850,7 +850,7 @@ long flush( boolean hasPersistentErrors = failedContext.getError() instanceof Exceptions.StreamFinalizedException || statusCode.equals(Status.Code.INVALID_ARGUMENT) - || statusCode.equals(Status.Code.NOT_FOUND) + || (!this.useDefaultStream && statusCode.equals(Status.Code.NOT_FOUND)) || statusCode.equals(Status.Code.FAILED_PRECONDITION); if (hasPersistentErrors) { throw new RuntimeException(