chore: Use long for expectedRowCount

Author: lqiu96

samples/snippets/src/main/java/com/example/bigquerystorage/ExportOpenTelemetry.java

@@ -105,11 +105,11 @@ public static void exportToOpenTelemetry(String projectId, String datasetName, S
 
     // Final cleanup for the stream during worker teardown.
     writer.cleanup();
-    verifyExpectedRowCount(parentTable, 12);
+    verifyExpectedRowCount(parentTable, 12L);
     System.out.println("Appended records successfully.");
   }
 
-  private static void verifyExpectedRowCount(TableName parentTable, int expectedRowCount)
+  private static void verifyExpectedRowCount(TableName parentTable, long expectedRowCount)
       throws InterruptedException {
     String queryRowCount =
         "SELECT COUNT(*) FROM `"
@@ -122,8 +122,8 @@ private static void verifyExpectedRowCount(TableName parentTable, int expectedRo
     QueryJobConfiguration queryConfig = QueryJobConfiguration.newBuilder(queryRowCount).build();
     BigQuery bigquery = BigQueryOptions.getDefaultInstance().getService();
     TableResult results = bigquery.query(queryConfig);
-    int countRowsActual =
-        Integer.parseInt(results.getValues().iterator().next().get("f0_").getStringValue());
+    long countRowsActual =
+        Long.parseLong(results.getValues().iterator().next().get("f0_").getStringValue());
     if (countRowsActual != expectedRowCount) {
       throw new RuntimeException(
           "Unexpected row count. Expected: " + expectedRowCount + ". Actual: " + countRowsActual);

samples/snippets/src/main/java/com/example/bigquerystorage/WriteToDefaultStream.java

@@ -106,11 +106,11 @@ public static void writeToDefaultStream(String projectId, String datasetName, St
 
     // Final cleanup for the stream during worker teardown.
     writer.cleanup();
-    verifyExpectedRowCount(parentTable, 12);
+    verifyExpectedRowCount(parentTable, 12L);
     System.out.println("Appended records successfully.");
   }
 
-  private static void verifyExpectedRowCount(TableName parentTable, int expectedRowCount)
+  private static void verifyExpectedRowCount(TableName parentTable, long expectedRowCount)
       throws InterruptedException {
     String queryRowCount =
         "SELECT COUNT(*) FROM `"
@@ -123,8 +123,8 @@ private static void verifyExpectedRowCount(TableName parentTable, int expectedRo
     QueryJobConfiguration queryConfig = QueryJobConfiguration.newBuilder(queryRowCount).build();
     BigQuery bigquery = BigQueryOptions.getDefaultInstance().getService();
     TableResult results = bigquery.query(queryConfig);
-    int countRowsActual =
-        Integer.parseInt(results.getValues().iterator().next().get("f0_").getStringValue());
+    long countRowsActual =
+        Long.parseLong(results.getValues().iterator().next().get("f0_").getStringValue());
     if (countRowsActual != expectedRowCount) {
       throw new RuntimeException(
           "Unexpected row count. Expected: " + expectedRowCount + ". Actual: " + countRowsActual);

samples/snippets/src/main/java/com/example/bigquerystorage/WriteToDefaultStreamTimestampJson.java

@@ -88,11 +88,11 @@ public static void writeToDefaultStream(String projectId, String datasetName, St
 
     // Final cleanup for the stream during worker teardown.
     writer.cleanup();
-    verifyExpectedRowCount(parentTable, 20);
+    verifyExpectedRowCount(parentTable, 20L);
     System.out.println("Appended records successfully.");
   }
 
-  private static void verifyExpectedRowCount(TableName parentTable, int expectedRowCount)
+  private static void verifyExpectedRowCount(TableName parentTable, long expectedRowCount)
       throws InterruptedException {
     String queryRowCount =
         "SELECT COUNT(*) FROM `"
@@ -105,8 +105,8 @@ private static void verifyExpectedRowCount(TableName parentTable, int expectedRo
     QueryJobConfiguration queryConfig = QueryJobConfiguration.newBuilder(queryRowCount).build();
     BigQuery bigquery = BigQueryOptions.getDefaultInstance().getService();
     TableResult results = bigquery.query(queryConfig);
-    int countRowsActual =
-        Integer.parseInt(results.getValues().iterator().next().get("f0_").getStringValue());
+    long countRowsActual =
+        Long.parseLong(results.getValues().iterator().next().get("f0_").getStringValue());
     if (countRowsActual != expectedRowCount) {
       throw new RuntimeException(
           "Unexpected row count. Expected: " + expectedRowCount + ". Actual: " + countRowsActual);

samples/snippets/src/main/java/com/example/bigquerystorage/WriteToDefaultStreamWithArrow.java

@@ -180,8 +180,8 @@ private static void verifyExpectedRowCount(TableName parentTable, long expectedR
     BigQuery bigquery =
         BigQueryOptions.newBuilder().setProjectId(parentTable.getProject()).build().getService();
     TableResult results = bigquery.query(queryConfig);
-    int countRowsActual =
-        Integer.parseInt(results.getValues().iterator().next().get("f0_").getStringValue());
+    long countRowsActual =
+        Long.parseLong(results.getValues().iterator().next().get("f0_").getStringValue());
     if (countRowsActual != expectedRowCount) {
       throw new RuntimeException(
           "Unexpected row count. Expected: " + expectedRowCount + ". Actual: " + countRowsActual);
@@ -216,8 +216,7 @@ private static class DataWriter {
 
     private final AtomicInteger recreateCount = new AtomicInteger(0);
 
-    private StreamWriter createStreamWriter(String streamName, Schema arrowSchema)
-        throws DescriptorValidationException, IOException, InterruptedException {
+    private StreamWriter createStreamWriter(String streamName, Schema arrowSchema) throws IOException {
       // Configure in-stream automatic retry settings.
       // Error codes that are immediately retried:
       // * ABORTED, UNAVAILABLE, CANCELLED, INTERNAL, DEADLINE_EXCEEDED