Skip to content

Commit 9c59d78

Browse files
committed
fixups
1 parent 5625ffb commit 9c59d78

File tree

5 files changed

+8
-10
lines changed

5 files changed

+8
-10
lines changed

sdk/src/main/java/com/google/cloud/dataflow/sdk/io/BigQueryIO.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1075,7 +1075,7 @@ public long getEstimatedSizeBytes(PipelineOptions options) throws Exception {
10751075
public BoundedReader<TableRow> createReader(PipelineOptions options) throws IOException {
10761076
BigQueryOptions bqOptions = options.as(BigQueryOptions.class);
10771077
return new BigQueryReader(this, bqServices.getReaderFromQuery(
1078-
bqOptions, createBasicQueryConfig(), executingProject.get()));
1078+
bqOptions, executingProject.get(), createBasicQueryConfig()));
10791079
}
10801080

10811081
@Override

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ public interface BigQueryServices extends Serializable {
5858
* Returns a real, mock, or fake {@link BigQueryJsonReader} to query tables.
5959
*/
6060
BigQueryJsonReader getReaderFromQuery(
61-
BigQueryOptions bqOptions, JobConfigurationQuery queryConfig, String projectId);
61+
BigQueryOptions bqOptions, String projectId, JobConfigurationQuery queryConfig);
6262

6363
/**
6464
* An interface for the Cloud BigQuery load service.

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

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ public BigQueryJsonReader getReaderFromTable(BigQueryOptions bqOptions, TableRef
8080

8181
@Override
8282
public BigQueryJsonReader getReaderFromQuery(
83-
BigQueryOptions bqOptions, JobConfigurationQuery queryConfig, String projectId) {
84-
return BigQueryJsonReaderImpl.fromQuery(bqOptions, queryConfig, projectId);
83+
BigQueryOptions bqOptions, String projectId, JobConfigurationQuery queryConfig) {
84+
return BigQueryJsonReaderImpl.fromQuery(bqOptions, projectId, queryConfig);
8585
}
8686

8787
@VisibleForTesting
@@ -516,9 +516,7 @@ private BigQueryJsonReaderImpl(BigQueryTableRowIterator iterator) {
516516
}
517517

518518
private static BigQueryJsonReader fromQuery(
519-
BigQueryOptions bqOptions,
520-
JobConfigurationQuery queryConfig,
521-
String projectId) {
519+
BigQueryOptions bqOptions, String projectId, JobConfigurationQuery queryConfig) {
522520
return new BigQueryJsonReaderImpl(
523521
BigQueryTableRowIterator.fromQuery(
524522
queryConfig, projectId, Transport.newBigQueryClient(bqOptions).build()));

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -408,8 +408,8 @@ private void deleteDataset(String datasetId) throws IOException, InterruptedExce
408408
*/
409409
private TableReference executeQueryAndWaitForCompletion()
410410
throws IOException, InterruptedException {
411-
checkState(projectId != null, "Cannot dryRun a query in unknown (null) project");
412-
checkState(queryConfig != null, "Cannot dryRun a null query");
411+
checkState(projectId != null, "Unable to execute a query without a configured project id");
412+
checkState(queryConfig != null, "Unable to execute a query without a configured query");
413413
// Dry run query to get source table location
414414
Job dryRunJob = new Job()
415415
.setConfiguration(new JobConfiguration()

sdk/src/test/java/com/google/cloud/dataflow/sdk/io/BigQueryIOTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ public BigQueryJsonReader getReaderFromTable(
183183

184184
@Override
185185
public BigQueryJsonReader getReaderFromQuery(
186-
BigQueryOptions bqOptions, JobConfigurationQuery queryConfig, String projectId) {
186+
BigQueryOptions bqOptions, String projectId, JobConfigurationQuery queryConfig) {
187187
return new FakeBigQueryReader(jsonTableRowReturns);
188188
}
189189

0 commit comments

Comments
 (0)