adjustments to the newer TableResult interface

AtScaleInc · Oct 29, 2024 · 9665247 · 9665247
1 parent 928a0dc
commit 9665247
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/BigQueryImpl.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/BigQueryImpl.java
@@ -1529,12 +1529,14 @@ public SyncTableResult syncQuery(QueryJobConfiguration configuration, QueryResul
     TableResult tableResult = null;
     if (response.getJobComplete()) {
       Map<BigQueryRpc.Option, ?> optionMap = optionMap(options);
-      tableResult = new TableResult(Schema.fromPb(response.getSchema()),
-              response.getTotalRows().longValue(),
-              new PageImpl<>(
+      tableResult = TableResult.newBuilder()
+              .setSchema(Schema.fromPb(response.getSchema()))
+              .setTotalRows(response.getTotalRows().longValue())
+              .setPageNoSchema(new PageImpl<>(
                       new QueryResponsePageFetcher(jobId, getOptions(), response.getPageToken(), optionMap),
                       response.getPageToken(),
-                      transformTableData(response.getRows(), Schema.fromPb(response.getSchema()))));
+                      transformTableData(response.getRows(), Schema.fromPb(response.getSchema()))))
+              .build();
     }
 
     return new SyncTableResult(response.getJobComplete(), jobId, tableResult);