Commit 996472b4 authored by izeye's avatar izeye Committed by Stephane Nicoll

Polish

See gh-26145
parent fea53294
...@@ -71,13 +71,13 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin ...@@ -71,13 +71,13 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin
private DataSourcePoolMetadataProvider poolMetadataProvider; private DataSourcePoolMetadataProvider poolMetadataProvider;
public DataSourceHealthContributorAutoConfiguration(Map<String, DataSource> dataSources, public DataSourceHealthContributorAutoConfiguration(
ObjectProvider<DataSourcePoolMetadataProvider> metadataProviders) { ObjectProvider<DataSourcePoolMetadataProvider> metadataProviders) {
this.metadataProviders = metadataProviders.orderedStream().collect(Collectors.toList()); this.metadataProviders = metadataProviders.orderedStream().collect(Collectors.toList());
} }
@Override @Override
public void afterPropertiesSet() throws Exception { public void afterPropertiesSet() {
this.poolMetadataProvider = new CompositeDataSourcePoolMetadataProvider(this.metadataProviders); this.poolMetadataProvider = new CompositeDataSourcePoolMetadataProvider(this.metadataProviders);
} }
...@@ -97,15 +97,15 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin ...@@ -97,15 +97,15 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin
private HealthContributor createContributor(Map<String, DataSource> beans) { private HealthContributor createContributor(Map<String, DataSource> beans) {
Assert.notEmpty(beans, "Beans must not be empty"); Assert.notEmpty(beans, "Beans must not be empty");
if (beans.size() == 1) { if (beans.size() == 1) {
return createIndicator(beans.values().iterator().next()); return createContributor(beans.values().iterator().next());
} }
return CompositeHealthContributor.fromMap(beans, this::createIndicator); return CompositeHealthContributor.fromMap(beans, this::createContributor);
} }
private HealthContributor createIndicator(DataSource source) { private HealthContributor createContributor(DataSource source) {
if (source instanceof AbstractRoutingDataSource) { if (source instanceof AbstractRoutingDataSource) {
AbstractRoutingDataSource routingDataSource = (AbstractRoutingDataSource) source; AbstractRoutingDataSource routingDataSource = (AbstractRoutingDataSource) source;
return new RoutingDataSourceHealthIndicator(routingDataSource, this::createIndicator); return new RoutingDataSourceHealthContributor(routingDataSource, this::createContributor);
} }
return new DataSourceHealthIndicator(source, getValidationQuery(source)); return new DataSourceHealthIndicator(source, getValidationQuery(source));
} }
...@@ -120,15 +120,15 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin ...@@ -120,15 +120,15 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin
* where the overall health is composed of a {@link DataSourceHealthIndicator} for * where the overall health is composed of a {@link DataSourceHealthIndicator} for
* each routed datasource. * each routed datasource.
*/ */
static class RoutingDataSourceHealthIndicator implements CompositeHealthContributor { static class RoutingDataSourceHealthContributor implements CompositeHealthContributor {
private CompositeHealthContributor delegate; private final CompositeHealthContributor delegate;
RoutingDataSourceHealthIndicator(AbstractRoutingDataSource routingDataSource, RoutingDataSourceHealthContributor(AbstractRoutingDataSource routingDataSource,
Function<DataSource, HealthContributor> indicatorFunction) { Function<DataSource, HealthContributor> contributorFunction) {
Map<String, DataSource> routedDataSources = routingDataSource.getResolvedDataSources().entrySet().stream() Map<String, DataSource> routedDataSources = routingDataSource.getResolvedDataSources().entrySet().stream()
.collect(Collectors.toMap((e) -> e.getKey().toString(), Map.Entry::getValue)); .collect(Collectors.toMap((e) -> e.getKey().toString(), Map.Entry::getValue));
this.delegate = CompositeHealthContributor.fromMap(routedDataSources, indicatorFunction); this.delegate = CompositeHealthContributor.fromMap(routedDataSources, contributorFunction);
} }
@Override @Override
......
...@@ -24,7 +24,7 @@ import javax.sql.DataSource; ...@@ -24,7 +24,7 @@ import javax.sql.DataSource;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.springframework.boot.actuate.autoconfigure.health.HealthContributorAutoConfiguration; import org.springframework.boot.actuate.autoconfigure.health.HealthContributorAutoConfiguration;
import org.springframework.boot.actuate.autoconfigure.jdbc.DataSourceHealthContributorAutoConfiguration.RoutingDataSourceHealthIndicator; import org.springframework.boot.actuate.autoconfigure.jdbc.DataSourceHealthContributorAutoConfiguration.RoutingDataSourceHealthContributor;
import org.springframework.boot.actuate.health.CompositeHealthContributor; import org.springframework.boot.actuate.health.CompositeHealthContributor;
import org.springframework.boot.actuate.health.NamedContributor; import org.springframework.boot.actuate.health.NamedContributor;
import org.springframework.boot.actuate.jdbc.DataSourceHealthIndicator; import org.springframework.boot.actuate.jdbc.DataSourceHealthIndicator;
...@@ -82,7 +82,7 @@ class DataSourceHealthContributorAutoConfigurationTests { ...@@ -82,7 +82,7 @@ class DataSourceHealthContributorAutoConfigurationTests {
CompositeHealthContributor composite = context.getBean(CompositeHealthContributor.class); CompositeHealthContributor composite = context.getBean(CompositeHealthContributor.class);
assertThat(composite.getContributor("dataSource")).isInstanceOf(DataSourceHealthIndicator.class); assertThat(composite.getContributor("dataSource")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(composite.getContributor("routingDataSource")) assertThat(composite.getContributor("routingDataSource"))
.isInstanceOf(RoutingDataSourceHealthIndicator.class); .isInstanceOf(RoutingDataSourceHealthContributor.class);
}); });
} }
...@@ -92,16 +92,16 @@ class DataSourceHealthContributorAutoConfigurationTests { ...@@ -92,16 +92,16 @@ class DataSourceHealthContributorAutoConfigurationTests {
.withPropertyValues("management.health.db.ignore-routing-datasources:true").run((context) -> { .withPropertyValues("management.health.db.ignore-routing-datasources:true").run((context) -> {
assertThat(context).doesNotHaveBean(CompositeHealthContributor.class); assertThat(context).doesNotHaveBean(CompositeHealthContributor.class);
assertThat(context).hasSingleBean(DataSourceHealthIndicator.class); assertThat(context).hasSingleBean(DataSourceHealthIndicator.class);
assertThat(context).doesNotHaveBean(RoutingDataSourceHealthIndicator.class); assertThat(context).doesNotHaveBean(RoutingDataSourceHealthContributor.class);
}); });
} }
@Test @Test
void runWithOnlyRoutingDataSourceShouldIncludeRoutingDataSourceWithComposedIndicators() { void runWithOnlyRoutingDataSourceShouldIncludeRoutingDataSourceWithComposedIndicators() {
this.contextRunner.withUserConfiguration(RoutingDataSourceConfig.class).run((context) -> { this.contextRunner.withUserConfiguration(RoutingDataSourceConfig.class).run((context) -> {
assertThat(context).hasSingleBean(RoutingDataSourceHealthIndicator.class); assertThat(context).hasSingleBean(RoutingDataSourceHealthContributor.class);
RoutingDataSourceHealthIndicator routingHealthContributor = context RoutingDataSourceHealthContributor routingHealthContributor = context
.getBean(RoutingDataSourceHealthIndicator.class); .getBean(RoutingDataSourceHealthContributor.class);
assertThat(routingHealthContributor.getContributor("one")).isInstanceOf(DataSourceHealthIndicator.class); assertThat(routingHealthContributor.getContributor("one")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(routingHealthContributor.getContributor("two")).isInstanceOf(DataSourceHealthIndicator.class); assertThat(routingHealthContributor.getContributor("two")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(routingHealthContributor.iterator()).toIterable().extracting("name") assertThat(routingHealthContributor.iterator()).toIterable().extracting("name")
......
...@@ -128,7 +128,7 @@ final class ApplicationPluginAction implements PluginApplicationAction { ...@@ -128,7 +128,7 @@ final class ApplicationPluginAction implements PluginApplicationAction {
private String loadResource(String name) { private String loadResource(String name) {
try (InputStreamReader reader = new InputStreamReader(getClass().getResourceAsStream(name))) { try (InputStreamReader reader = new InputStreamReader(getClass().getResourceAsStream(name))) {
char[] buffer = new char[4096]; char[] buffer = new char[4096];
int read = 0; int read;
StringWriter writer = new StringWriter(); StringWriter writer = new StringWriter();
while ((read = reader.read(buffer)) > 0) { while ((read = reader.read(buffer)) > 0) {
writer.write(buffer, 0, read); writer.write(buffer, 0, read);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment