use of io.github.resilience4j.bulkhead.configure.threadpool.ThreadPoolBulkheadConfiguration in project resilience4j by resilience4j.
the class BulkHeadConfigurationTest method testCreateThreadPoolBulkHeadRegistryWithSharedConfigs.
@Test
public void testCreateThreadPoolBulkHeadRegistryWithSharedConfigs() {
// Given
ThreadPoolBulkheadConfigurationProperties.InstanceProperties defaultProperties = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
defaultProperties.setCoreThreadPoolSize(1);
defaultProperties.setQueueCapacity(1);
defaultProperties.setKeepAliveDuration(Duration.ofNanos(5));
defaultProperties.setMaxThreadPoolSize(10);
ThreadPoolBulkheadConfigurationProperties.InstanceProperties sharedProperties = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
sharedProperties.setCoreThreadPoolSize(2);
sharedProperties.setQueueCapacity(2);
sharedProperties.setContextPropagators(TestThreadLocalContextPropagator.class);
ThreadPoolBulkheadConfigurationProperties.InstanceProperties backendWithDefaultConfig = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
backendWithDefaultConfig.setBaseConfig("default");
backendWithDefaultConfig.setCoreThreadPoolSize(3);
ThreadPoolBulkheadConfigurationProperties.InstanceProperties backendWithSharedConfig = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
backendWithSharedConfig.setBaseConfig("sharedConfig");
backendWithSharedConfig.setCoreThreadPoolSize(4);
backendWithSharedConfig.setContextPropagators(TestThreadLocalContextPropagator.class);
ThreadPoolBulkheadConfigurationProperties bulkheadConfigurationProperties = new ThreadPoolBulkheadConfigurationProperties();
bulkheadConfigurationProperties.getConfigs().put("default", defaultProperties);
bulkheadConfigurationProperties.getConfigs().put("sharedConfig", sharedProperties);
bulkheadConfigurationProperties.getBackends().put("backendWithDefaultConfig", backendWithDefaultConfig);
bulkheadConfigurationProperties.getBackends().put("backendWithSharedConfig", backendWithSharedConfig);
ThreadPoolBulkheadConfiguration threadPoolBulkheadConfiguration = new ThreadPoolBulkheadConfiguration();
DefaultEventConsumerRegistry<BulkheadEvent> eventConsumerRegistry = new DefaultEventConsumerRegistry<>();
// When
try {
ThreadPoolBulkheadRegistry bulkheadRegistry = threadPoolBulkheadConfiguration.threadPoolBulkheadRegistry(bulkheadConfigurationProperties, eventConsumerRegistry, new CompositeRegistryEventConsumer<>(emptyList()), new CompositeCustomizer<>(Collections.emptyList()));
// Then
assertThat(bulkheadRegistry.getAllBulkheads().size()).isEqualTo(2);
// Should get default config and core number
ThreadPoolBulkhead bulkhead1 = bulkheadRegistry.bulkhead("backendWithDefaultConfig");
assertThat(bulkhead1).isNotNull();
assertThat(bulkhead1.getBulkheadConfig().getCoreThreadPoolSize()).isEqualTo(3);
assertThat(bulkhead1.getBulkheadConfig().getQueueCapacity()).isEqualTo(1);
assertThat(bulkhead1.getBulkheadConfig().getContextPropagator()).isNotNull();
assertThat(bulkhead1.getBulkheadConfig().getContextPropagator()).isEmpty();
// Should get shared config and overwrite core number
ThreadPoolBulkhead bulkhead2 = bulkheadRegistry.bulkhead("backendWithSharedConfig");
assertThat(bulkhead2).isNotNull();
assertThat(bulkhead2.getBulkheadConfig().getCoreThreadPoolSize()).isEqualTo(4);
assertThat(bulkhead2.getBulkheadConfig().getQueueCapacity()).isEqualTo(2);
assertThat(bulkhead2.getBulkheadConfig().getContextPropagator()).isNotNull();
assertThat(bulkhead2.getBulkheadConfig().getContextPropagator().size()).isEqualTo(2);
List<Class<? extends ContextPropagator>> ctxPropagators = bulkhead2.getBulkheadConfig().getContextPropagator().stream().map(ctx -> ctx.getClass()).collect(Collectors.toList());
assertThat(ctxPropagators).containsExactlyInAnyOrder(TestThreadLocalContextPropagator.class, TestThreadLocalContextPropagator.class);
// Unknown backend should get default config of Registry
ThreadPoolBulkhead bulkhead3 = bulkheadRegistry.bulkhead("unknownBackend");
assertThat(bulkhead3).isNotNull();
assertThat(bulkhead3.getBulkheadConfig().getCoreThreadPoolSize()).isEqualTo(1);
assertThat(bulkhead3.getBulkheadConfig().getContextPropagator()).isNotNull();
assertThat(bulkhead3.getBulkheadConfig().getContextPropagator()).isEmpty();
assertThat(eventConsumerRegistry.getAllEventConsumer()).hasSize(3);
} catch (Exception e) {
System.out.println("exception in testCreateThreadPoolBulkHeadRegistryWithSharedConfigs():" + e);
}
}
use of io.github.resilience4j.bulkhead.configure.threadpool.ThreadPoolBulkheadConfiguration in project resilience4j by resilience4j.
the class BulkHeadConfigurationTest method tesFixedThreadPoolBulkHeadRegistry.
@Test
public void tesFixedThreadPoolBulkHeadRegistry() {
// Given
ThreadPoolBulkheadConfigurationProperties.InstanceProperties backendProperties1 = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
backendProperties1.setCoreThreadPoolSize(1);
backendProperties1.setContextPropagators(TestThreadLocalContextPropagator.class);
ThreadPoolBulkheadConfigurationProperties.InstanceProperties backendProperties2 = new ThreadPoolBulkheadConfigurationProperties.InstanceProperties();
backendProperties2.setCoreThreadPoolSize(2);
ThreadPoolBulkheadConfigurationProperties bulkheadConfigurationProperties = new ThreadPoolBulkheadConfigurationProperties();
bulkheadConfigurationProperties.getBackends().put("backend1", backendProperties1);
bulkheadConfigurationProperties.getBackends().put("backend2", backendProperties2);
ThreadPoolBulkheadConfiguration threadPoolBulkheadConfiguration = new ThreadPoolBulkheadConfiguration();
DefaultEventConsumerRegistry<BulkheadEvent> eventConsumerRegistry = new DefaultEventConsumerRegistry<>();
// When
ThreadPoolBulkheadRegistry bulkheadRegistry = threadPoolBulkheadConfiguration.threadPoolBulkheadRegistry(bulkheadConfigurationProperties, eventConsumerRegistry, new CompositeRegistryEventConsumer<>(emptyList()), new CompositeCustomizer<>(Collections.emptyList()));
// Then
assertThat(bulkheadRegistry.getAllBulkheads().size()).isEqualTo(2);
ThreadPoolBulkhead bulkhead1 = bulkheadRegistry.bulkhead("backend1");
assertThat(bulkhead1).isNotNull();
assertThat(bulkhead1.getBulkheadConfig().getCoreThreadPoolSize()).isEqualTo(1);
assertThat(bulkhead1.getBulkheadConfig().getContextPropagator()).isNotNull();
assertThat(bulkhead1.getBulkheadConfig().getContextPropagator().size()).isEqualTo(1);
assertThat(bulkhead1.getBulkheadConfig().getContextPropagator().get(0).getClass()).isEqualTo(TestThreadLocalContextPropagator.class);
ThreadPoolBulkhead bulkhead2 = bulkheadRegistry.bulkhead("backend2");
assertThat(bulkhead2).isNotNull();
assertThat(bulkhead2.getBulkheadConfig().getCoreThreadPoolSize()).isEqualTo(2);
assertThat(bulkhead2.getBulkheadConfig().getContextPropagator()).isNotNull();
assertThat(bulkhead2.getBulkheadConfig().getContextPropagator()).isEmpty();
assertThat(eventConsumerRegistry.getAllEventConsumer()).hasSize(2);
}
Aggregations