Merge branch 'escape-hatch' into samples-update

Author: jinseopkim0

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminClientV2.java

@@ -21,9 +21,8 @@
 /**
  * Modern Cloud Bigtable Instance Admin Client.
  *
- * <p>This client extends the auto-generated {@link BaseBigtableInstanceAdminClient} to provide
- * manual overrides and additional convenience methods for Critical User Journeys (CUJs) that the
- * GAPIC generator cannot handle natively.
+ * <p>This client extends the {@link BaseBigtableInstanceAdminClient} to provide a simplified and
+ * enhanced API surface for managing Cloud Bigtable instances and clusters.
  */
 public class BigtableInstanceAdminClientV2 extends BaseBigtableInstanceAdminClient {
 

google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2.java

@@ -45,15 +45,16 @@
 import io.grpc.MethodDescriptor.MethodType;
 import java.io.IOException;
 import java.io.InputStream;
+import java.time.Duration;
 import java.util.concurrent.ExecutionException;
-import org.threeten.bp.Duration;
 
 /**
  * Modern Cloud Bigtable Table Admin Client.
  *
- * <p>This client extends the auto-generated {@link BaseBigtableTableAdminClient} to provide manual
- * overrides and additional convenience methods for Critical User Journeys (CUJs) that the GAPIC
- * generator cannot handle natively (e.g., chained Long Running Operations, Consistency Polling).
+ * <p>This client extends the {@link BaseBigtableTableAdminClient} to provide enhanced convenience
+ * methods for table administration. It improves the user experience by handling chained Long
+ * Running Operations (such as seamlessly restoring and then optimizing a table) and provides
+ * built-in, automated polling for consistency tokens.
  */
 public class BigtableTableAdminClientV2 extends BaseBigtableTableAdminClient {
   private final com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable
@@ -95,11 +96,11 @@ protected BigtableTableAdminClientV2(BigtableTableAdminStub stub) {
         RetrySettings.newBuilder()
             .setTotalTimeout(
                 settings.checkConsistencySettings().getRetrySettings().getTotalTimeout())
-            .setInitialRetryDelay(Duration.ofSeconds(10))
+            .setInitialRetryDelayDuration(Duration.ofSeconds(10))
             .setRetryDelayMultiplier(1.0)
-            .setMaxRetryDelay(Duration.ofSeconds(10))
-            .setInitialRpcTimeout(Duration.ZERO)
-            .setMaxRpcTimeout(Duration.ZERO)
+            .setMaxRetryDelayDuration(Duration.ofSeconds(10))
+            .setInitialRpcTimeoutDuration(Duration.ZERO)
+            .setMaxRpcTimeoutDuration(Duration.ZERO)
             .setRpcTimeoutMultiplier(1.0)
             .build();
 
@@ -157,7 +158,7 @@ public Operation parse(InputStream stream) {
         OperationCallSettings.<Void, Empty, OptimizeRestoredTableMetadata>newBuilder()
             .setInitialCallSettings(
                 UnaryCallSettings.<Void, OperationSnapshot>newUnaryCallSettingsBuilder()
-                    .setSimpleTimeoutNoRetries(Duration.ZERO)
+                    .setSimpleTimeoutNoRetriesDuration(Duration.ZERO)
                     .build())
             .setMetadataTransformer(
                 new ApiFunction<OperationSnapshot, OptimizeRestoredTableMetadata>() {
@@ -176,13 +177,13 @@ public Empty apply(OperationSnapshot input) {
             .setPollingAlgorithm(
                 OperationTimedPollAlgorithm.create(
                     RetrySettings.newBuilder()
-                        .setInitialRetryDelay(Duration.ofMillis(500L))
+                        .setInitialRetryDelayDuration(Duration.ofMillis(500L))
                         .setRetryDelayMultiplier(1.5)
-                        .setMaxRetryDelay(Duration.ofMillis(5000L))
-                        .setInitialRpcTimeout(Duration.ZERO)
+                        .setMaxRetryDelayDuration(Duration.ofMillis(5000L))
+                        .setInitialRpcTimeoutDuration(Duration.ZERO)
                         .setRpcTimeoutMultiplier(1.0)
-                        .setMaxRpcTimeout(Duration.ZERO)
-                        .setTotalTimeout(Duration.ofMillis(600000L))
+                        .setMaxRpcTimeoutDuration(Duration.ZERO)
+                        .setTotalTimeoutDuration(Duration.ofMillis(600000L))
                         .build()))
             .build();
 

google-cloud-bigtable/src/test/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2Test.java

@@ -155,7 +155,7 @@ public void testCreateClientWithSettings() throws Exception {
 
   @Test
   public void testAwaitConsistency_ThrowsWhenNotInitialized() {
-    BigtableTableAdminClientV2 uninitializedClient = new BigtableTableAdminClientV2(mockStub);
+    BigtableTableAdminClientV2 uninitializedClient = BigtableTableAdminClientV2.create(mockStub);
 
     try {
       uninitializedClient.waitForConsistency(TABLE_NAME, "token");
@@ -168,7 +168,7 @@ public void testAwaitConsistency_ThrowsWhenNotInitialized() {
 
   @Test
   public void testOptimizeRestoredTable_ThrowsWhenNotInitialized() {
-    BigtableTableAdminClientV2 uninitializedClient = new BigtableTableAdminClientV2(mockStub);
+    BigtableTableAdminClientV2 uninitializedClient = BigtableTableAdminClientV2.create(mockStub);
 
     OptimizeRestoredTableOperationToken mockToken =
         Mockito.mock(OptimizeRestoredTableOperationToken.class);