|
15 | 15 | */ |
16 | 16 | package com.google.cloud.bigtable.admin.v2; |
17 | 17 |
|
| 18 | +import com.google.api.core.ApiClock; |
18 | 19 | import com.google.api.core.ApiFunction; |
19 | 20 | import com.google.api.core.ApiFuture; |
20 | 21 | import com.google.api.core.ApiFutures; |
| 22 | +import com.google.api.gax.grpc.GrpcCallContext; |
21 | 23 | import com.google.api.gax.grpc.GrpcCallSettings; |
22 | 24 | import com.google.api.gax.grpc.GrpcCallableFactory; |
23 | 25 | import com.google.api.gax.grpc.ProtoOperationTransformers.MetadataTransformer; |
|
26 | 28 | import com.google.api.gax.longrunning.OperationTimedPollAlgorithm; |
27 | 29 | import com.google.api.gax.retrying.RetrySettings; |
28 | 30 | import com.google.api.gax.rpc.ApiExceptions; |
| 31 | +import com.google.api.gax.rpc.ClientContext; |
29 | 32 | import com.google.api.gax.rpc.OperationCallSettings; |
30 | 33 | import com.google.api.gax.rpc.OperationCallable; |
31 | 34 | import com.google.api.gax.rpc.UnaryCallSettings; |
@@ -132,7 +135,7 @@ protected BigtableTableAdminClientV2( |
132 | 135 | private static AwaitConsistencyCallableV2 createAwaitConsistencyCallable( |
133 | 136 | GrpcBigtableTableAdminStub stub, |
134 | 137 | BigtableTableAdminStubSettings settings, |
135 | | - com.google.api.core.ApiClock clock, |
| 138 | + ApiClock clock, |
136 | 139 | ScheduledExecutorService executor) { |
137 | 140 | // TODO(igorbernstein2): expose polling settings |
138 | 141 | RetrySettings pollingSettings = |
@@ -195,8 +198,8 @@ public Void parse(InputStream stream) { |
195 | 198 | final MetadataTransformer<OptimizeRestoredTableMetadata> protoMetadataTransformer = |
196 | 199 | MetadataTransformer.create(OptimizeRestoredTableMetadata.class); |
197 | 200 |
|
198 | | - final ResponseTransformer<com.google.protobuf.Empty> protoResponseTransformer = |
199 | | - ResponseTransformer.create(com.google.protobuf.Empty.class); |
| 201 | + final ResponseTransformer<Empty> protoResponseTransformer = |
| 202 | + ResponseTransformer.create(Empty.class); |
200 | 203 |
|
201 | 204 | OperationCallSettings<Void, Empty, OptimizeRestoredTableMetadata> operationCallSettings = |
202 | 205 | OperationCallSettings.<Void, Empty, OptimizeRestoredTableMetadata>newBuilder() |
@@ -229,11 +232,11 @@ public Empty apply(OperationSnapshot input) { |
229 | 232 | // already encapsulates the fully-configured default call context (including channels, |
230 | 233 | // credentials, |
231 | 234 | // and headers) for executing the polling RPCs. |
232 | | - com.google.api.gax.rpc.ClientContext clientContext = |
233 | | - com.google.api.gax.rpc.ClientContext.newBuilder() |
| 235 | + ClientContext clientContext = |
| 236 | + ClientContext.newBuilder() |
234 | 237 | .setClock(settings.getStubSettings().getClock()) |
235 | 238 | .setExecutor(backgroundExecutor) |
236 | | - .setDefaultCallContext(com.google.api.gax.grpc.GrpcCallContext.createDefault()) |
| 239 | + .setDefaultCallContext(GrpcCallContext.createDefault()) |
237 | 240 | .build(); |
238 | 241 |
|
239 | 242 | return GrpcCallableFactory.createOperationCallable( |
|
0 commit comments