|
26 | 26 | import com.google.api.gax.rpc.ServerStreamingCallSettings; |
27 | 27 | import com.google.api.gax.rpc.StatusCode.Code; |
28 | 28 | import com.google.api.gax.rpc.StubSettings; |
| 29 | +import com.google.api.gax.rpc.TransportChannelProvider; |
29 | 30 | import com.google.api.gax.rpc.UnaryCallSettings; |
30 | 31 | import com.google.cloud.bigtable.data.v2.internal.DummyBatchingDescriptor; |
31 | 32 | import com.google.cloud.bigtable.data.v2.models.ConditionalRowMutation; |
@@ -176,6 +177,19 @@ public String getAppProfileId() { |
176 | 177 | return appProfileId; |
177 | 178 | } |
178 | 179 |
|
| 180 | + /** Returns a builder for the default ChannelProvider for this service. */ |
| 181 | + public static InstantiatingGrpcChannelProvider.Builder defaultGrpcTransportProviderBuilder() { |
| 182 | + return BigtableStubSettings.defaultGrpcTransportProviderBuilder() |
| 183 | + // TODO: tune channels |
| 184 | + .setChannelsPerCpu(2) |
| 185 | + .setMaxInboundMessageSize(MAX_MESSAGE_SIZE); |
| 186 | + } |
| 187 | + |
| 188 | + @SuppressWarnings("WeakerAccess") |
| 189 | + public static TransportChannelProvider defaultTransportChannelProvider() { |
| 190 | + return defaultGrpcTransportProviderBuilder().build(); |
| 191 | + } |
| 192 | + |
179 | 193 | /** Returns a builder for the default credentials for this service. */ |
180 | 194 | public static GoogleCredentialsProvider.Builder defaultCredentialsProviderBuilder() { |
181 | 195 | return BigtableStubSettings.defaultCredentialsProviderBuilder() |
@@ -260,12 +274,7 @@ private Builder() { |
260 | 274 | BigtableStubSettings.Builder baseDefaults = BigtableStubSettings.newBuilder(); |
261 | 275 |
|
262 | 276 | setEndpoint(baseDefaults.getEndpoint()); |
263 | | - setTransportChannelProvider( |
264 | | - InstantiatingGrpcChannelProvider.newBuilder() |
265 | | - // TODO: tune channels |
266 | | - .setChannelsPerCpu(2) |
267 | | - .setMaxInboundMessageSize(MAX_MESSAGE_SIZE) |
268 | | - .build()); |
| 277 | + setTransportChannelProvider(defaultTransportChannelProvider()); |
269 | 278 | setStreamWatchdogCheckInterval(baseDefaults.getStreamWatchdogCheckInterval()); |
270 | 279 | setStreamWatchdogProvider(baseDefaults.getStreamWatchdogProvider()); |
271 | 280 |
|
|
0 commit comments