Browse Source

Avoid NPE when routing DataSource has target with null key

See gh-27698
pull/28375/head
Safeer A 5 years ago committed by Andy Wilkinson
parent
commit
efdc19647c
  1. 7
      spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/jdbc/DataSourceHealthContributorAutoConfiguration.java
  2. 5
      spring-boot-project/spring-boot-actuator-autoconfigure/src/test/java/org/springframework/boot/actuate/autoconfigure/jdbc/DataSourceHealthContributorAutoConfigurationTests.java

7
spring-boot-project/spring-boot-actuator-autoconfigure/src/main/java/org/springframework/boot/actuate/autoconfigure/jdbc/DataSourceHealthContributorAutoConfiguration.java

@ -19,6 +19,7 @@ package org.springframework.boot.actuate.autoconfigure.jdbc; @@ -19,6 +19,7 @@ package org.springframework.boot.actuate.autoconfigure.jdbc;
import java.util.Collection;
import java.util.Iterator;
import java.util.Map;
import java.util.Optional;
import java.util.function.Function;
import java.util.stream.Collectors;
@ -57,6 +58,7 @@ import org.springframework.util.Assert; @@ -57,6 +58,7 @@ import org.springframework.util.Assert;
* @author Stephane Nicoll
* @author Arthur Kalimullin
* @author Julio Gomez
* @author Safeer Ansari
* @since 2.0.0
*/
@Configuration(proxyBeanMethods = false)
@ -124,10 +126,13 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin @@ -124,10 +126,13 @@ public class DataSourceHealthContributorAutoConfiguration implements Initializin
private final CompositeHealthContributor delegate;
private static final String UNNAMED_DATASOURCE_KEY = "unnamed";
RoutingDataSourceHealthContributor(AbstractRoutingDataSource routingDataSource,
Function<DataSource, HealthContributor> contributorFunction) {
Map<String, DataSource> routedDataSources = routingDataSource.getResolvedDataSources().entrySet().stream()
.collect(Collectors.toMap((e) -> e.getKey().toString(), Map.Entry::getValue));
.collect(Collectors.toMap((e) -> Optional.ofNullable(e.getKey()).map(Object::toString)
.orElse(UNNAMED_DATASOURCE_KEY), Map.Entry::getValue));
this.delegate = CompositeHealthContributor.fromMap(routedDataSources, contributorFunction);
}

5
spring-boot-project/spring-boot-actuator-autoconfigure/src/test/java/org/springframework/boot/actuate/autoconfigure/jdbc/DataSourceHealthContributorAutoConfigurationTests.java

@ -49,6 +49,7 @@ import static org.mockito.Mockito.mock; @@ -49,6 +49,7 @@ import static org.mockito.Mockito.mock;
*
* @author Phillip Webb
* @author Julio Gomez
* @author Safeer Ansari
*/
class DataSourceHealthContributorAutoConfigurationTests {
@ -102,10 +103,11 @@ class DataSourceHealthContributorAutoConfigurationTests { @@ -102,10 +103,11 @@ class DataSourceHealthContributorAutoConfigurationTests {
assertThat(context).hasSingleBean(RoutingDataSourceHealthContributor.class);
RoutingDataSourceHealthContributor routingHealthContributor = context
.getBean(RoutingDataSourceHealthContributor.class);
assertThat(routingHealthContributor.getContributor("unnamed")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(routingHealthContributor.getContributor("one")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(routingHealthContributor.getContributor("two")).isInstanceOf(DataSourceHealthIndicator.class);
assertThat(routingHealthContributor.iterator()).toIterable().extracting("name")
.containsExactlyInAnyOrder("one", "two");
.containsExactlyInAnyOrder("unnamed", "one", "two");
});
}
@ -155,6 +157,7 @@ class DataSourceHealthContributorAutoConfigurationTests { @@ -155,6 +157,7 @@ class DataSourceHealthContributorAutoConfigurationTests {
@Bean
AbstractRoutingDataSource routingDataSource() {
Map<Object, DataSource> dataSources = new HashMap<>();
dataSources.put(null, mock(DataSource.class));
dataSources.put("one", mock(DataSource.class));
dataSources.put("two", mock(DataSource.class));
AbstractRoutingDataSource routingDataSource = mock(AbstractRoutingDataSource.class);

Loading…
Cancel
Save