diff --git a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminClient.java b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminClient.java index 0e5a9155a0..e9f057e73e 100644 --- a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminClient.java +++ b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableInstanceAdminClient.java @@ -125,7 +125,7 @@ public final class BigtableInstanceAdminClient implements AutoCloseable { private final String projectId; private final BigtableInstanceAdminStub stub; - private final BaseBigtableInstanceAdminClient baseClient; + private final BigtableInstanceAdminClientV2 baseClient; /** Constructs an instance of BigtableInstanceAdminClient with the given project ID. */ public static BigtableInstanceAdminClient create(@Nonnull String projectId) throws IOException { @@ -148,7 +148,7 @@ private BigtableInstanceAdminClient( @Nonnull String projectId, @Nonnull BigtableInstanceAdminStub stub) { this.projectId = projectId; this.stub = stub; - this.baseClient = BaseBigtableInstanceAdminClient.create(stub); + this.baseClient = BigtableInstanceAdminClientV2.create(stub); } /** Gets the project ID this client is associated with. */ @@ -160,7 +160,7 @@ public String getProjectId() { * Returns the modern autogenerated client. This provides access to the newest features and * proto-based methods. */ - public BaseBigtableInstanceAdminClient getBaseClient() { + public BigtableInstanceAdminClientV2 getBaseClient() { return baseClient; } diff --git a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClient.java b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClient.java index 3cd7fef840..be636f6305 100644 --- a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClient.java +++ b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClient.java @@ -169,7 +169,7 @@ public final class BigtableTableAdminClient implements AutoCloseable { private final EnhancedBigtableTableAdminStub stub; private final String projectId; private final String instanceId; - private final BaseBigtableTableAdminClient baseClient; + private final BigtableTableAdminClientV2 baseClient; /** Constructs an instance of BigtableTableAdminClient with the given project and instance IDs. */ public static BigtableTableAdminClient create( @@ -209,7 +209,12 @@ private BigtableTableAdminClient( this.projectId = projectId; this.instanceId = instanceId; this.stub = stub; - this.baseClient = BaseBigtableTableAdminClient.create(stub); + this.baseClient = + new BigtableTableAdminClientV2( + stub, + (com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable) + stub.awaitConsistencyCallable(), + stub.awaitOptimizeRestoredTableCallable()); } /** Gets the project ID of the instance whose tables this client manages. */ @@ -226,7 +231,7 @@ public String getInstanceId() { * Returns the modern autogenerated client. This provides access to the newest features and * proto-based methods. */ - public BaseBigtableTableAdminClient getBaseClient() { + public BigtableTableAdminClientV2 getBaseClient() { return baseClient; } diff --git a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2.java b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2.java index ce04ed16e5..5a4e2a3df1 100644 --- a/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2.java +++ b/google-cloud-bigtable/src/main/java/com/google/cloud/bigtable/admin/v2/BigtableTableAdminClientV2.java @@ -35,7 +35,6 @@ import com.google.cloud.bigtable.admin.v2.models.ConsistencyRequest; import com.google.cloud.bigtable.admin.v2.models.OptimizeRestoredTableOperationToken; import com.google.cloud.bigtable.admin.v2.models.RestoredTableResult; -import com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable; import com.google.cloud.bigtable.admin.v2.stub.BigtableTableAdminStub; import com.google.cloud.bigtable.admin.v2.stub.BigtableTableAdminStubSettings; import com.google.common.base.Strings; @@ -58,7 +57,8 @@ * built-in, automated polling for consistency tokens. */ public class BigtableTableAdminClientV2 extends BaseBigtableTableAdminClient { - private final AwaitConsistencyCallable awaitConsistencyCallable; + private final com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable + awaitConsistencyCallable; private final OperationCallable optimizeRestoredTableOperationBaseCallable; @@ -80,7 +80,7 @@ protected BigtableTableAdminClientV2(BigtableTableAdminStub stub) { @com.google.common.annotations.VisibleForTesting BigtableTableAdminClientV2( BigtableTableAdminStub stub, - AwaitConsistencyCallable awaitConsistencyCallable, + com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable awaitConsistencyCallable, OperationCallable optimizeRestoredTableOperationBaseCallable) { super(stub); @@ -88,8 +88,8 @@ protected BigtableTableAdminClientV2(BigtableTableAdminStub stub) { this.optimizeRestoredTableOperationBaseCallable = optimizeRestoredTableOperationBaseCallable; } - private AwaitConsistencyCallable createAwaitConsistencyCallable( - BigtableTableAdminStubSettings settings) throws IOException { + private com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable + createAwaitConsistencyCallable(BigtableTableAdminStubSettings settings) throws IOException { ClientContext clientContext = ClientContext.create(settings); // TODO(igorbernstein2): expose polling settings RetrySettings pollingSettings = @@ -104,7 +104,7 @@ private AwaitConsistencyCallable createAwaitConsistencyCallable( .setRpcTimeoutMultiplier(1.0) .build(); - return AwaitConsistencyCallable.create( + return com.google.cloud.bigtable.admin.v2.stub.AwaitConsistencyCallable.create( getStub().generateConsistencyTokenCallable(), getStub().checkConsistencyCallable(), clientContext, @@ -200,11 +200,6 @@ public static final BigtableTableAdminClientV2 create(BaseBigtableTableAdminSett return new BigtableTableAdminClientV2(settings); } - /** Constructs an instance of BigtableTableAdminClientV2 with the given stub. */ - public static final BigtableTableAdminClientV2 create(BigtableTableAdminStub stub) { - return new BigtableTableAdminClientV2(stub); - } - /** * Awaits the completion of the "Optimize Restored Table" operation. *