Merge pull request #489 from dpmills/dataflow-client

Use a retrying Dataflow Client for DataflowPipelineJob

Author: kennknowles

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

@@ -651,7 +651,7 @@ public DataflowPipelineJob run(Pipeline pipeline) {
     // regularly and need not be retried automatically.
     DataflowPipelineJob dataflowPipelineJob =
         new DataflowPipelineJob(options.getProject(), jobResult.getId(),
-            Transport.newRawDataflowClient(options).build(), aggregatorTransforms);
+            Transport.newDataflowClient(options).build(), aggregatorTransforms);
 
     // If the service returned client request id, the SDK needs to compare it
     // 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

@@ -161,17 +161,6 @@ public static Clouddebugger.Builder newClouddebuggerClient(DataflowPipelineOptio
         .setGoogleClientRequestInitializer(options.getGoogleApiTrace());
   }
 
-  /**
-   * Returns a Dataflow client that does not automatically retry failed
-   * requests.
-   */
-  public static Dataflow.Builder
-      newRawDataflowClient(DataflowPipelineOptions options) {
-    return newDataflowClient(options)
-        .setHttpRequestInitializer(options.getGcpCredential())
-        .setGoogleClientRequestInitializer(options.getGoogleApiTrace());
-  }
-
   /**
    * Returns a Cloud Storage client builder.
    *