private void verifyBuilder( EnhancedBigQueryStorageStubSettings.Builder builder, String endpoint, CredentialsProvider credentialsProvider, Duration watchdogInterval, WatchdogProvider watchdogProvider) { assertThat(builder.getEndpoint()).isEqualTo(endpoint); assertThat(builder.getCredentialsProvider()).isEqualTo(credentialsProvider); assertThat(builder.getStreamWatchdogCheckInterval()).isEqualTo(watchdogInterval); assertThat(builder.getStreamWatchdogProvider()).isEqualTo(watchdogProvider); InstantiatingGrpcChannelProvider channelProvider = (InstantiatingGrpcChannelProvider) builder.getTransportChannelProvider(); assertThat(channelProvider.toBuilder().getMaxInboundMessageSize()) .isEqualTo(MAX_INBOUND_MESSAGE_SIZE); assertThat(channelProvider.toBuilder().getPoolSize()).isGreaterThan(1); }
private void verifySettings( EnhancedBigQueryStorageStubSettings settings, String endpoint, CredentialsProvider credentialsProvider, Duration watchdogInterval, WatchdogProvider watchdogProvider) { assertThat(settings.getEndpoint()).isEqualTo(endpoint); assertThat(settings.getCredentialsProvider()).isEqualTo(credentialsProvider); assertThat(settings.getStreamWatchdogCheckInterval()).isEqualTo(watchdogInterval); assertThat(settings.getStreamWatchdogProvider()).isEqualTo(watchdogProvider); InstantiatingGrpcChannelProvider channelProvider = (InstantiatingGrpcChannelProvider) settings.getTransportChannelProvider(); assertThat(channelProvider.toBuilder().getMaxInboundMessageSize()) .isEqualTo(MAX_INBOUND_MESSAGE_SIZE); assertThat(channelProvider.toBuilder().getPoolSize()).isGreaterThan(1); }
@Test public void multipleChannelsByDefaultTest() { String dummyProjectId = "my-project"; String dummyInstanceId = "my-instance"; EnhancedBigtableStubSettings.Builder builder = EnhancedBigtableStubSettings.newBuilder() .setProjectId(dummyProjectId) .setInstanceId(dummyInstanceId); InstantiatingGrpcChannelProvider provider = (InstantiatingGrpcChannelProvider) builder.getTransportChannelProvider(); assertThat(provider.toBuilder().getPoolSize()).isGreaterThan(1); }
@Test public void testCpuPoolSize() { // happy path Builder builder = InstantiatingGrpcChannelProvider.newBuilder().setProcessorCount(2); builder.setChannelsPerCpu(2.5); assertEquals(5, builder.getPoolSize()); // User specified max builder = builder.setProcessorCount(50); builder.setChannelsPerCpu(100, 10); assertEquals(10, builder.getPoolSize()); // Sane default maximum builder.setChannelsPerCpu(200); assertEquals(100, builder.getPoolSize()); }