|
51 | 51 | * @author Stephane Nicoll
|
52 | 52 | * @author Andy Wilkinson
|
53 | 53 | * @author Tommy Ludwig
|
| 54 | + * @author Yanming Zhou |
54 | 55 | */
|
55 | 56 | class DataSourcePoolMetricsAutoConfigurationTests {
|
56 | 57 |
|
@@ -219,6 +220,20 @@ void hikariDataSourceIsInstrumentedWithoutMetadataProvider() {
|
219 | 220 | });
|
220 | 221 | }
|
221 | 222 |
|
| 223 | + @Test |
| 224 | + void nonDefaultHikariDataSourcesCanBeInstrumented() { |
| 225 | + this.contextRunner.withUserConfiguration(NonDefaultHikariDataSourceConfiguration.class) |
| 226 | + .withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class)) |
| 227 | + .run((context) -> { |
| 228 | + assertThat(context).getBeanNames(DataSource.class) |
| 229 | + .containsExactlyInAnyOrder("dataSource", "nonDefaultHikariDataSource"); |
| 230 | + context.getBean("nonDefaultHikariDataSource", DataSource.class).getConnection(); |
| 231 | + MeterRegistry registry = context.getBean(MeterRegistry.class); |
| 232 | + assertThat(registry.get("hikaricp.connections").meter().getId().getTags()) |
| 233 | + .containsExactly(Tag.of("pool", "nonDefaultHikariDataSource")); |
| 234 | + }); |
| 235 | + } |
| 236 | + |
222 | 237 | private static HikariDataSource createHikariDataSource(String poolName) {
|
223 | 238 | String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
224 | 239 | HikariDataSource hikariDataSource = DataSourceBuilder.create().url(url).type(HikariDataSource.class).build();
|
@@ -271,6 +286,16 @@ DataSource secondOne() {
|
271 | 286 |
|
272 | 287 | }
|
273 | 288 |
|
| 289 | + @Configuration(proxyBeanMethods = false) |
| 290 | + static class NonDefaultHikariDataSourceConfiguration { |
| 291 | + |
| 292 | + @Bean(defaultCandidate = false) |
| 293 | + DataSource nonDefaultHikariDataSource() { |
| 294 | + return createHikariDataSource("nonDefaultHikariDataSource"); |
| 295 | + } |
| 296 | + |
| 297 | + } |
| 298 | + |
274 | 299 | @Configuration(proxyBeanMethods = false)
|
275 | 300 | static class ProxiedHikariDataSourcesConfiguration {
|
276 | 301 |
|
|
0 commit comments