Skip to content
This repository was archived by the owner on Nov 11, 2022. It is now read-only.

Commit 5305b35

Browse files
committed
Use a retrying Dataflow Client for DataflowPipelineJob
1 parent 32d7342 commit 5305b35

File tree

2 files changed

+1
-12
lines changed

2 files changed

+1
-12
lines changed

sdk/src/main/java/com/google/cloud/dataflow/sdk/runners/DataflowPipelineRunner.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -625,7 +625,7 @@ public DataflowPipelineJob run(Pipeline pipeline) {
625625
// regularly and need not be retried automatically.
626626
DataflowPipelineJob dataflowPipelineJob =
627627
new DataflowPipelineJob(options.getProject(), jobResult.getId(),
628-
Transport.newRawDataflowClient(options).build(), aggregatorTransforms);
628+
Transport.newDataflowClient(options).build(), aggregatorTransforms);
629629

630630
// If the service returned client request id, the SDK needs to compare it
631631
// with the original id generated in the request, if they are not the same

sdk/src/main/java/com/google/cloud/dataflow/sdk/util/Transport.java

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -157,17 +157,6 @@ public static Clouddebugger.Builder newClouddebuggerClient(DataflowPipelineOptio
157157
.setGoogleClientRequestInitializer(options.getGoogleApiTrace());
158158
}
159159

160-
/**
161-
* Returns a Dataflow client that does not automatically retry failed
162-
* requests.
163-
*/
164-
public static Dataflow.Builder
165-
newRawDataflowClient(DataflowPipelineOptions options) {
166-
return newDataflowClient(options)
167-
.setHttpRequestInitializer(options.getGcpCredential())
168-
.setGoogleClientRequestInitializer(options.getGoogleApiTrace());
169-
}
170-
171160
/**
172161
* Returns a Cloud Storage client builder.
173162
*

0 commit comments

Comments
 (0)