diff --git a/bigtable-dataflow-parent/bigtable-hbase-beam/src/main/java/com/google/cloud/bigtable/batch/common/CloudBigtableServiceImpl.java b/bigtable-dataflow-parent/bigtable-hbase-beam/src/main/java/com/google/cloud/bigtable/batch/common/CloudBigtableServiceImpl.java index 45defc8404..7bf036335b 100755 --- a/bigtable-dataflow-parent/bigtable-hbase-beam/src/main/java/com/google/cloud/bigtable/batch/common/CloudBigtableServiceImpl.java +++ b/bigtable-dataflow-parent/bigtable-hbase-beam/src/main/java/com/google/cloud/bigtable/batch/common/CloudBigtableServiceImpl.java @@ -24,6 +24,7 @@ import com.google.cloud.bigtable.beam.CloudBigtableTableConfiguration; import com.google.cloud.bigtable.hbase.wrappers.veneer.BigtableHBaseVeneerSettings; import java.io.IOException; +import java.time.Duration; import java.util.List; /** For internal use only - public for technical reasons. */ @@ -52,6 +53,15 @@ public List getSampleRowKeys(CloudBigtableTableConfiguration config) .setPreemptiveRefreshEnabled(false) .build()) .build()); + + builder + .stubSettings() + .sampleRowKeysSettings() + .retrySettings() + .setInitialRpcTimeoutDuration(Duration.ofMinutes(20)) + .setMaxRpcTimeoutDuration(Duration.ofMinutes(20)) + .setTotalTimeoutDuration(Duration.ofMinutes(20)); + try (BigtableDataClient client = BigtableDataClient.create(builder.build())) { return client.sampleRowKeys(config.getTableId()); }