fix: Configure defualt keep alive pings for BQ Storage Read API RPCs (#12037)

lqiu96 · web-flow · commit a808bbef93e7 · 2026-03-02T12:14:23.000-08:00
See b/424828238 for more information
diff --git a/java-bigquerystorage/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/stub/EnhancedBigQueryReadStubSettings.java b/java-bigquerystorage/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/stub/EnhancedBigQueryReadStubSettings.java
@@ -36,6 +36,7 @@
 import com.google.cloud.bigquery.storage.v1.SplitReadStreamResponse;
 import com.google.cloud.bigquery.storage.v1.stub.readrows.ReadRowsResumptionStrategy;
 import com.google.common.collect.ImmutableList;
+import java.time.Duration;
 import java.util.List;
 
 /**
@@ -111,7 +112,12 @@ public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilde
 
   /** Returns a builder for the default ChannelProvider for this service. */
   public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() {
-    return BigQueryReadStubSettings.defaultGrpcTransportProviderBuilder();
+    // Configure a default keep-alive value to prevent idle connections from dropping.
+    // Some CreateReadSession calls may be long-running (10+ min).
+    return BigQueryReadStubSettings.defaultGrpcTransportProviderBuilder()
+        .setKeepAliveTimeDuration(Duration.ofMinutes(1))
+        .setKeepAliveTimeoutDuration(Duration.ofMinutes(1))
+        .setKeepAliveWithoutCalls(true);
   }
 
   public static TransportChannelProvider defaultTransportChannelProvider() {
