Move JDBC metrics auto-configuration into spring-boot-jdbc
This commit is contained in:
Andy Wilkinson
committed by
Phillip Webb
Phillip Webb
parent
bb3bd2a81e
commit
fedc08a757
@@ -21,6 +21,7 @@ dependencies {
|
||||
optional(project(":spring-boot-project:spring-boot-actuator-autoconfigure"))
|
||||
optional(project(":spring-boot-project:spring-boot-autoconfigure"))
|
||||
optional(project(":spring-boot-project:spring-boot-docker-compose"))
|
||||
optional(project(":spring-boot-project:spring-boot-metrics"))
|
||||
optional(project(":spring-boot-project:spring-boot-testcontainers"))
|
||||
optional("com.h2database:h2")
|
||||
optional("com.mchange:c3p0")
|
||||
|
||||
@@ -0,0 +1,165 @@
|
||||
/*
|
||||
* Copyright 2012-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.jdbc.actuate.metrics.autoconfigure;
|
||||
|
||||
import java.util.Collection;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import javax.sql.DataSource;
|
||||
|
||||
import com.zaxxer.hikari.HikariConfigMXBean;
|
||||
import com.zaxxer.hikari.HikariDataSource;
|
||||
import com.zaxxer.hikari.metrics.micrometer.MicrometerMetricsTrackerFactory;
|
||||
import io.micrometer.core.instrument.MeterRegistry;
|
||||
import io.micrometer.core.instrument.binder.MeterBinder;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
|
||||
import org.springframework.beans.factory.ObjectProvider;
|
||||
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
|
||||
import org.springframework.beans.factory.support.SimpleAutowireCandidateResolver;
|
||||
import org.springframework.boot.autoconfigure.AutoConfiguration;
|
||||
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
|
||||
import org.springframework.boot.jdbc.DataSourceUnwrapper;
|
||||
import org.springframework.boot.jdbc.actuate.metrics.DataSourcePoolMetrics;
|
||||
import org.springframework.boot.jdbc.autoconfigure.DataSourceAutoConfiguration;
|
||||
import org.springframework.boot.jdbc.metadata.DataSourcePoolMetadataProvider;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.core.log.LogMessage;
|
||||
import org.springframework.util.StringUtils;
|
||||
|
||||
/**
|
||||
* {@link EnableAutoConfiguration Auto-configuration} for metrics on all available
|
||||
* {@link DataSource datasources}.
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Yanming Zhou
|
||||
* @since 4.0.0
|
||||
*/
|
||||
@AutoConfiguration(after = DataSourceAutoConfiguration.class,
|
||||
afterName = "org.springframework.boot.metrics.autoconfigure.CompositeMeterRegistryAutoConfiguration")
|
||||
@ConditionalOnClass({ DataSource.class, MeterRegistry.class })
|
||||
@ConditionalOnBean({ DataSource.class, MeterRegistry.class })
|
||||
public class DataSourcePoolMetricsAutoConfiguration {
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
@ConditionalOnBean(DataSourcePoolMetadataProvider.class)
|
||||
static class DataSourcePoolMetadataMetricsConfiguration {
|
||||
|
||||
private static final String DATASOURCE_SUFFIX = "dataSource";
|
||||
|
||||
@Bean
|
||||
DataSourcePoolMetadataMeterBinder dataSourcePoolMetadataMeterBinder(ConfigurableListableBeanFactory beanFactory,
|
||||
ObjectProvider<DataSourcePoolMetadataProvider> metadataProviders) {
|
||||
return new DataSourcePoolMetadataMeterBinder(SimpleAutowireCandidateResolver
|
||||
.resolveAutowireCandidates(beanFactory, DataSource.class, false, true), metadataProviders);
|
||||
}
|
||||
|
||||
static class DataSourcePoolMetadataMeterBinder implements MeterBinder {
|
||||
|
||||
private final Map<String, DataSource> dataSources;
|
||||
|
||||
private final ObjectProvider<DataSourcePoolMetadataProvider> metadataProviders;
|
||||
|
||||
DataSourcePoolMetadataMeterBinder(Map<String, DataSource> dataSources,
|
||||
ObjectProvider<DataSourcePoolMetadataProvider> metadataProviders) {
|
||||
this.dataSources = dataSources;
|
||||
this.metadataProviders = metadataProviders;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void bindTo(MeterRegistry registry) {
|
||||
List<DataSourcePoolMetadataProvider> metadataProvidersList = this.metadataProviders.stream().toList();
|
||||
this.dataSources.forEach((name, dataSource) -> bindDataSourceToRegistry(name, dataSource,
|
||||
metadataProvidersList, registry));
|
||||
}
|
||||
|
||||
private void bindDataSourceToRegistry(String beanName, DataSource dataSource,
|
||||
Collection<DataSourcePoolMetadataProvider> metadataProviders, MeterRegistry registry) {
|
||||
String dataSourceName = getDataSourceName(beanName);
|
||||
new DataSourcePoolMetrics(dataSource, metadataProviders, dataSourceName, Collections.emptyList())
|
||||
.bindTo(registry);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the name of a DataSource based on its {@code beanName}.
|
||||
* @param beanName the name of the data source bean
|
||||
* @return a name for the given data source
|
||||
*/
|
||||
private String getDataSourceName(String beanName) {
|
||||
if (beanName.length() > DATASOURCE_SUFFIX.length()
|
||||
&& StringUtils.endsWithIgnoreCase(beanName, DATASOURCE_SUFFIX)) {
|
||||
return beanName.substring(0, beanName.length() - DATASOURCE_SUFFIX.length());
|
||||
}
|
||||
return beanName;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
@ConditionalOnClass(HikariDataSource.class)
|
||||
static class HikariDataSourceMetricsConfiguration {
|
||||
|
||||
@Bean
|
||||
HikariDataSourceMeterBinder hikariDataSourceMeterBinder(ObjectProvider<DataSource> dataSources) {
|
||||
return new HikariDataSourceMeterBinder(dataSources);
|
||||
}
|
||||
|
||||
static class HikariDataSourceMeterBinder implements MeterBinder {
|
||||
|
||||
private static final Log logger = LogFactory.getLog(HikariDataSourceMeterBinder.class);
|
||||
|
||||
private final ObjectProvider<DataSource> dataSources;
|
||||
|
||||
HikariDataSourceMeterBinder(ObjectProvider<DataSource> dataSources) {
|
||||
this.dataSources = dataSources;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void bindTo(MeterRegistry registry) {
|
||||
this.dataSources.stream(ObjectProvider.UNFILTERED, false).forEach((dataSource) -> {
|
||||
HikariDataSource hikariDataSource = DataSourceUnwrapper.unwrap(dataSource, HikariConfigMXBean.class,
|
||||
HikariDataSource.class);
|
||||
if (hikariDataSource != null) {
|
||||
bindMetricsRegistryToHikariDataSource(hikariDataSource, registry);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
private void bindMetricsRegistryToHikariDataSource(HikariDataSource hikari, MeterRegistry registry) {
|
||||
if (hikari.getMetricRegistry() == null && hikari.getMetricsTrackerFactory() == null) {
|
||||
try {
|
||||
hikari.setMetricsTrackerFactory(new MicrometerMetricsTrackerFactory(registry));
|
||||
}
|
||||
catch (Exception ex) {
|
||||
logger.warn(LogMessage.format("Failed to bind Hikari metrics: %s", ex.getMessage()));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,20 @@
|
||||
/*
|
||||
* Copyright 2012-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
/**
|
||||
* Auto-configuration for JPA metrics.
|
||||
*/
|
||||
package org.springframework.boot.jdbc.actuate.metrics.autoconfigure;
|
||||
@@ -1,4 +1,5 @@
|
||||
org.springframework.boot.jdbc.actuate.health.autoconfigure.DataSourceHealthContributorAutoConfiguration
|
||||
org.springframework.boot.jdbc.actuate.metrics.autoconfigure.DataSourcePoolMetricsAutoConfiguration
|
||||
org.springframework.boot.jdbc.autoconfigure.DataSourceAutoConfiguration
|
||||
org.springframework.boot.jdbc.autoconfigure.DataSourceInitializationAutoConfiguration
|
||||
org.springframework.boot.jdbc.autoconfigure.DataSourceTransactionManagerAutoConfiguration
|
||||
|
||||
@@ -0,0 +1,397 @@
|
||||
/*
|
||||
* Copyright 2012-2025 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.jdbc.actuate.metrics.autoconfigure;
|
||||
|
||||
import java.sql.SQLException;
|
||||
import java.util.UUID;
|
||||
|
||||
import javax.sql.DataSource;
|
||||
|
||||
import com.zaxxer.hikari.HikariDataSource;
|
||||
import io.micrometer.core.instrument.MeterRegistry;
|
||||
import io.micrometer.core.instrument.Tag;
|
||||
import io.micrometer.core.instrument.simple.SimpleMeterRegistry;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import org.springframework.aop.framework.ProxyFactory;
|
||||
import org.springframework.beans.factory.config.BeanDefinition;
|
||||
import org.springframework.beans.factory.config.BeanPostProcessor;
|
||||
import org.springframework.boot.LazyInitializationBeanFactoryPostProcessor;
|
||||
import org.springframework.boot.autoconfigure.AutoConfigurations;
|
||||
import org.springframework.boot.jdbc.DataSourceBuilder;
|
||||
import org.springframework.boot.jdbc.autoconfigure.DataSourceAutoConfiguration;
|
||||
import org.springframework.boot.jdbc.autoconfigure.DataSourceInitializationAutoConfiguration;
|
||||
import org.springframework.boot.jdbc.metadata.DataSourcePoolMetadataProvider;
|
||||
import org.springframework.boot.metrics.autoconfigure.MetricsAutoConfiguration;
|
||||
import org.springframework.boot.test.context.runner.ApplicationContextRunner;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.context.annotation.Scope;
|
||||
import org.springframework.core.Ordered;
|
||||
import org.springframework.core.PriorityOrdered;
|
||||
import org.springframework.jdbc.datasource.DelegatingDataSource;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
|
||||
/**
|
||||
* Tests for {@link DataSourcePoolMetricsAutoConfiguration}.
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Andy Wilkinson
|
||||
* @author Tommy Ludwig
|
||||
* @author Yanming Zhou
|
||||
*/
|
||||
class DataSourcePoolMetricsAutoConfigurationTests {
|
||||
|
||||
private final ApplicationContextRunner contextRunner = new ApplicationContextRunner()
|
||||
.withPropertyValues("spring.datasource.generate-unique-name=true",
|
||||
"management.metrics.use-global-registry=false")
|
||||
.withBean(SimpleMeterRegistry.class)
|
||||
.withConfiguration(
|
||||
AutoConfigurations.of(MetricsAutoConfiguration.class, DataSourcePoolMetricsAutoConfiguration.class));
|
||||
|
||||
@Test
|
||||
void autoConfiguredDataSourceIsInstrumented() {
|
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean(DataSource.class).getConnection().getMetaData();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
registry.get("jdbc.connections.max").tags("name", "dataSource").meter();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void dataSourceInstrumentationCanBeDisabled() {
|
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.withPropertyValues("management.metrics.enable.jdbc=false")
|
||||
.run((context) -> {
|
||||
context.getBean(DataSource.class).getConnection().getMetaData();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("jdbc.connections.max").tags("name", "dataSource").meter()).isNull();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void allDataSourcesCanBeInstrumented() {
|
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.withUserConfiguration(TwoDataSourcesConfiguration.class)
|
||||
.run((context) -> {
|
||||
context.getBean("nonDefaultDataSource", DataSource.class).getConnection().getMetaData();
|
||||
context.getBean("nonAutowireDataSource", DataSource.class).getConnection().getMetaData();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("jdbc.connections.max").meters()).map((meter) -> meter.getId().getTag("name"))
|
||||
.containsOnly("dataSource", "nonDefault");
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void allDataSourcesCanBeInstrumentedWithLazyInitialization() {
|
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.withInitializer(
|
||||
(context) -> context.addBeanFactoryPostProcessor(new LazyInitializationBeanFactoryPostProcessor()))
|
||||
.withUserConfiguration(TwoDataSourcesConfiguration.class)
|
||||
.run((context) -> {
|
||||
context.getBean("nonDefaultDataSource", DataSource.class).getConnection().getMetaData();
|
||||
context.getBean("nonAutowireDataSource", DataSource.class).getConnection().getMetaData();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("jdbc.connections.max").meters()).map((meter) -> meter.getId().getTag("name"))
|
||||
.containsOnly("dataSource", "nonDefault");
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void autoConfiguredHikariDataSourceIsInstrumented() {
|
||||
this.contextRunner.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean(DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
registry.get("hikaricp.connections").meter();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void autoConfiguredHikariDataSourceIsInstrumentedWhenUsingDataSourceInitialization() {
|
||||
this.contextRunner.withPropertyValues("spring.sql.init.schema:db/create-custom-schema.sql")
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class,
|
||||
DataSourceInitializationAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean(DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
registry.get("hikaricp.connections").meter();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void hikariCanBeInstrumentedAfterThePoolHasBeenSealed() {
|
||||
this.contextRunner.withUserConfiguration(HikariSealingConfiguration.class)
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
assertThat(context).hasNotFailed();
|
||||
context.getBean(DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("hikaricp.connections").meter()).isNotNull();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void hikariDataSourceInstrumentationCanBeDisabled() {
|
||||
this.contextRunner.withPropertyValues("management.metrics.enable.hikaricp=false")
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean(DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("hikaricp.connections").meter()).isNull();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void allHikariDataSourcesCanBeInstrumented() {
|
||||
this.contextRunner.withUserConfiguration(MultipleHikariDataSourcesConfiguration.class)
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean("standardDataSource", DataSource.class).getConnection();
|
||||
context.getBean("nonDefault", DataSource.class).getConnection();
|
||||
context.getBean("nonAutowire", DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("hikaricp.connections").meters()).map((meter) -> meter.getId().getTag("pool"))
|
||||
.containsOnly("standardDataSource", "nonDefault");
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void someHikariDataSourcesCanBeInstrumented() {
|
||||
this.contextRunner.withUserConfiguration(MixedDataSourcesConfiguration.class)
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean("firstDataSource", DataSource.class).getConnection();
|
||||
context.getBean("secondOne", DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.get("hikaricp.connections").meter().getId().getTags())
|
||||
.containsExactly(Tag.of("pool", "firstDataSource"));
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void allHikariDataSourcesCanBeInstrumentedWhenUsingLazyInitialization() {
|
||||
this.contextRunner.withUserConfiguration(MultipleHikariDataSourcesConfiguration.class)
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.withInitializer(
|
||||
(context) -> context.addBeanFactoryPostProcessor(new LazyInitializationBeanFactoryPostProcessor()))
|
||||
.run((context) -> {
|
||||
context.getBean("standardDataSource", DataSource.class).getConnection();
|
||||
context.getBean("nonDefault", DataSource.class).getConnection();
|
||||
context.getBean("nonAutowire", DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.find("hikaricp.connections").meters()).map((meter) -> meter.getId().getTag("pool"))
|
||||
.containsOnly("standardDataSource", "nonDefault");
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void hikariProxiedDataSourceCanBeInstrumented() {
|
||||
this.contextRunner.withUserConfiguration(ProxiedHikariDataSourcesConfiguration.class)
|
||||
.withConfiguration(AutoConfigurations.of(DataSourceAutoConfiguration.class))
|
||||
.run((context) -> {
|
||||
context.getBean("proxiedDataSource", DataSource.class).getConnection();
|
||||
context.getBean("delegateDataSource", DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
registry.get("hikaricp.connections").tags("pool", "firstDataSource").meter();
|
||||
registry.get("hikaricp.connections").tags("pool", "secondOne").meter();
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void hikariDataSourceIsInstrumentedWithoutMetadataProvider() {
|
||||
this.contextRunner.withUserConfiguration(OneHikariDataSourceConfiguration.class).run((context) -> {
|
||||
assertThat(context).doesNotHaveBean(DataSourcePoolMetadataProvider.class);
|
||||
context.getBean("hikariDataSource", DataSource.class).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.get("hikaricp.connections").meter().getId().getTags())
|
||||
.containsExactly(Tag.of("pool", "hikariDataSource"));
|
||||
});
|
||||
}
|
||||
|
||||
@Test
|
||||
void prototypeDataSourceIsIgnored() {
|
||||
this.contextRunner
|
||||
.withUserConfiguration(OneHikariDataSourceConfiguration.class, PrototypeDataSourceConfiguration.class)
|
||||
.run((context) -> {
|
||||
context.getBean("hikariDataSource", DataSource.class).getConnection();
|
||||
((DataSource) context.getBean("prototypeDataSource", "", "")).getConnection();
|
||||
MeterRegistry registry = context.getBean(MeterRegistry.class);
|
||||
assertThat(registry.get("hikaricp.connections").meter().getId().getTags())
|
||||
.containsExactly(Tag.of("pool", "hikariDataSource"));
|
||||
});
|
||||
}
|
||||
|
||||
private static HikariDataSource createHikariDataSource(String poolName) {
|
||||
String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
||||
HikariDataSource hikariDataSource = DataSourceBuilder.create().url(url).type(HikariDataSource.class).build();
|
||||
hikariDataSource.setPoolName(poolName);
|
||||
return hikariDataSource;
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class TwoDataSourcesConfiguration {
|
||||
|
||||
@Bean(defaultCandidate = false)
|
||||
DataSource nonDefaultDataSource() {
|
||||
return createDataSource();
|
||||
}
|
||||
|
||||
@Bean(autowireCandidate = false)
|
||||
DataSource nonAutowireDataSource() {
|
||||
return createDataSource();
|
||||
}
|
||||
|
||||
private DataSource createDataSource() {
|
||||
String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
||||
return DataSourceBuilder.create().url(url).build();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class MultipleHikariDataSourcesConfiguration {
|
||||
|
||||
@Bean
|
||||
DataSource standardDataSource() {
|
||||
return createHikariDataSource("standardDataSource");
|
||||
}
|
||||
|
||||
@Bean(defaultCandidate = false)
|
||||
DataSource nonDefault() {
|
||||
return createHikariDataSource("nonDefault");
|
||||
}
|
||||
|
||||
@Bean(autowireCandidate = false)
|
||||
DataSource nonAutowire() {
|
||||
return createHikariDataSource("nonAutowire");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class ProxiedHikariDataSourcesConfiguration {
|
||||
|
||||
@Bean
|
||||
DataSource proxiedDataSource() {
|
||||
return (DataSource) new ProxyFactory(createHikariDataSource("firstDataSource")).getProxy();
|
||||
}
|
||||
|
||||
@Bean
|
||||
DataSource delegateDataSource() {
|
||||
return new DelegatingDataSource(createHikariDataSource("secondOne"));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class OneHikariDataSourceConfiguration {
|
||||
|
||||
@Bean
|
||||
DataSource hikariDataSource() {
|
||||
return createHikariDataSource("hikariDataSource");
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class MixedDataSourcesConfiguration {
|
||||
|
||||
@Bean
|
||||
DataSource firstDataSource() {
|
||||
return createHikariDataSource("firstDataSource");
|
||||
}
|
||||
|
||||
@Bean
|
||||
DataSource secondOne() {
|
||||
return createTomcatDataSource();
|
||||
}
|
||||
|
||||
private HikariDataSource createHikariDataSource(String poolName) {
|
||||
String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
||||
HikariDataSource hikariDataSource = DataSourceBuilder.create()
|
||||
.url(url)
|
||||
.type(HikariDataSource.class)
|
||||
.build();
|
||||
hikariDataSource.setPoolName(poolName);
|
||||
return hikariDataSource;
|
||||
}
|
||||
|
||||
private org.apache.tomcat.jdbc.pool.DataSource createTomcatDataSource() {
|
||||
String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
||||
return DataSourceBuilder.create().url(url).type(org.apache.tomcat.jdbc.pool.DataSource.class).build();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class PrototypeDataSourceConfiguration {
|
||||
|
||||
@Bean
|
||||
@Scope(BeanDefinition.SCOPE_PROTOTYPE)
|
||||
DataSource prototypeDataSource(String username, String password) {
|
||||
return createHikariDataSource(username, password);
|
||||
}
|
||||
|
||||
private HikariDataSource createHikariDataSource(String username, String password) {
|
||||
String url = "jdbc:hsqldb:mem:test-" + UUID.randomUUID();
|
||||
HikariDataSource hikariDataSource = DataSourceBuilder.create()
|
||||
.url(url)
|
||||
.type(HikariDataSource.class)
|
||||
.username(username)
|
||||
.password(password)
|
||||
.build();
|
||||
return hikariDataSource;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
static class HikariSealingConfiguration {
|
||||
|
||||
@Bean
|
||||
static HikariSealer hikariSealer() {
|
||||
return new HikariSealer();
|
||||
}
|
||||
|
||||
static class HikariSealer implements BeanPostProcessor, PriorityOrdered {
|
||||
|
||||
@Override
|
||||
public int getOrder() {
|
||||
return Ordered.HIGHEST_PRECEDENCE;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object postProcessAfterInitialization(Object bean, String beanName) {
|
||||
if (bean instanceof HikariDataSource dataSource) {
|
||||
try {
|
||||
dataSource.getConnection().close();
|
||||
}
|
||||
catch (SQLException ex) {
|
||||
throw new IllegalStateException(ex);
|
||||
}
|
||||
}
|
||||
return bean;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user