diff --git a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/util/common/worker/WorkProgressUpdater.java b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/util/common/worker/WorkProgressUpdater.java index 8ed15373ffac..f04e9a47afe1 100644 --- a/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/util/common/worker/WorkProgressUpdater.java +++ b/runners/google-cloud-dataflow-java/worker/src/main/java/org/apache/beam/runners/dataflow/worker/util/common/worker/WorkProgressUpdater.java @@ -309,7 +309,7 @@ private void reportProgress() { LOG.info("Cancelling workitem execution: {}", workString(), e); worker.abort(); } catch (Throwable e) { - LOG.warn("Error reporting workitem progress update to Dataflow service: ", e); + LOG.error("Error reporting workitem progress update to Dataflow service: ", e); } } diff --git a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryStorageStreamSource.java b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryStorageStreamSource.java index adc0933defed..5dbebc7fb79d 100644 --- a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryStorageStreamSource.java +++ b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryStorageStreamSource.java @@ -368,8 +368,13 @@ public synchronized BigQueryStorageStreamSource getCurrentSource() { .setName(source.readStream.getName()) .setFraction((float) fraction) .build(); - - SplitReadStreamResponse splitResponse = storageClient.splitReadStream(splitRequest); + SplitReadStreamResponse splitResponse; + try { + splitResponse = storageClient.splitReadStream(splitRequest); + } catch (Exception e) { + LOG.warn("Skip split read stream due to failed request: ", e); + return null; + } if (!splitResponse.hasPrimaryStream() || !splitResponse.hasRemainderStream()) { // No more splits are possible! impossibleSplitPointCalls.inc();