refactor(samples): query pagination (#1008)

googleapis · Dec 9, 2020 · 7f2e5ca · 7f2e5ca
1 parent 9c97f4a
commit 7f2e5ca
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/samples/snippets/src/main/java/com/example/bigquery/QueryPagination.java b/samples/snippets/src/main/java/com/example/bigquery/QueryPagination.java
@@ -57,11 +57,19 @@ public static void queryPagination(String datasetName, String tableName, String
       TableResult results =
           bigquery.listTableData(tableId, BigQuery.TableDataListOption.pageSize(20));
 
+      // First Page
       results
-          .getNextPage()
           .iterateAll()
           .forEach(row -> row.forEach(val -> System.out.printf("%s,", val.toString())));
 
+      if (results.hasNextPage()) {
+        // Next Page
+        results
+            .getNextPage()
+            .iterateAll()
+            .forEach(row -> row.forEach(val -> System.out.printf("%s,", val.toString())));
+      }
+
       System.out.println("Query pagination performed successfully.");
     } catch (BigQueryException | InterruptedException e) {
       System.out.println("Query not performed \n" + e.toString());