3535import com .google .cloud .bigtable .admin .v2 .models .ConsistencyRequest ;
3636import com .google .cloud .bigtable .admin .v2 .models .OptimizeRestoredTableOperationToken ;
3737import com .google .cloud .bigtable .admin .v2 .models .RestoredTableResult ;
38- import com .google .cloud .bigtable .admin .v2 .stub .AwaitConsistencyCallable ;
3938import com .google .cloud .bigtable .admin .v2 .stub .BigtableTableAdminStub ;
4039import com .google .cloud .bigtable .admin .v2 .stub .BigtableTableAdminStubSettings ;
41- import com .google .cloud .bigtable .admin .v2 .stub .EnhancedBigtableTableAdminStub ;
4240import com .google .common .base .Strings ;
4341import com .google .longrunning .Operation ;
4442import com .google .protobuf .Empty ;
5856 * generator cannot handle natively (e.g., chained Long Running Operations, Consistency Polling).
5957 */
6058public class BigtableTableAdminClientV2 extends BaseBigtableTableAdminClient {
61- private final AwaitConsistencyCallable awaitConsistencyCallable ;
59+ private final com . google . cloud . bigtable . admin . v2 . stub . AwaitConsistencyCallable awaitConsistencyCallable ;
6260 private final OperationCallable <Void , Empty , OptimizeRestoredTableMetadata >
6361 optimizeRestoredTableOperationBaseCallable ;
6462
@@ -80,15 +78,15 @@ protected BigtableTableAdminClientV2(BigtableTableAdminStub stub) {
8078 @ com .google .common .annotations .VisibleForTesting
8179 BigtableTableAdminClientV2 (
8280 BigtableTableAdminStub stub ,
83- AwaitConsistencyCallable awaitConsistencyCallable ,
81+ com . google . cloud . bigtable . admin . v2 . stub . AwaitConsistencyCallable awaitConsistencyCallable ,
8482 OperationCallable <Void , Empty , OptimizeRestoredTableMetadata >
8583 optimizeRestoredTableOperationBaseCallable ) {
8684 super (stub );
8785 this .awaitConsistencyCallable = awaitConsistencyCallable ;
8886 this .optimizeRestoredTableOperationBaseCallable = optimizeRestoredTableOperationBaseCallable ;
8987 }
9088
91- private AwaitConsistencyCallable createAwaitConsistencyCallable (
89+ private com . google . cloud . bigtable . admin . v2 . stub . AwaitConsistencyCallable createAwaitConsistencyCallable (
9290 BigtableTableAdminStubSettings settings ) throws IOException {
9391 ClientContext clientContext = ClientContext .create (settings );
9492 // TODO(igorbernstein2): expose polling settings
@@ -104,7 +102,7 @@ private AwaitConsistencyCallable createAwaitConsistencyCallable(
104102 .setRpcTimeoutMultiplier (1.0 )
105103 .build ();
106104
107- return AwaitConsistencyCallable .create (
105+ return com . google . cloud . bigtable . admin . v2 . stub . AwaitConsistencyCallable .create (
108106 getStub ().generateConsistencyTokenCallable (),
109107 getStub ().checkConsistencyCallable (),
110108 clientContext ,
@@ -200,18 +198,6 @@ public static final BigtableTableAdminClientV2 create(BaseBigtableTableAdminSett
200198 return new BigtableTableAdminClientV2 (settings );
201199 }
202200
203- /** Constructs an instance of BigtableTableAdminClientV2 with the given stub. */
204- public static final BigtableTableAdminClientV2 create (BigtableTableAdminStub stub ) {
205- if (stub instanceof EnhancedBigtableTableAdminStub ) {
206- EnhancedBigtableTableAdminStub enhancedStub = (EnhancedBigtableTableAdminStub ) stub ;
207- return new BigtableTableAdminClientV2 (
208- stub ,
209- (AwaitConsistencyCallable ) enhancedStub .awaitConsistencyCallable (),
210- enhancedStub .awaitOptimizeRestoredTableCallable ());
211- }
212- return new BigtableTableAdminClientV2 (stub );
213- }
214-
215201 /**
216202 * Awaits the completion of the "Optimize Restored Table" operation.
217203 *
@@ -330,7 +316,7 @@ private UnaryCallable<ConsistencyRequest, Void> getAwaitConsistencyCallable() {
330316 return awaitConsistencyCallable ;
331317 }
332318 throw new IllegalStateException (
333- "AwaitConsistencyCallable not initialized. BigtableTableAdminClientV2 must be "
319+ "com.google.cloud.bigtable.admin.v2.stub. AwaitConsistencyCallable not initialized. BigtableTableAdminClientV2 must be "
334320 + "initialized via settings to use this functionality." );
335321 }
336322
0 commit comments