application.yml配置多数据源
时间: 2023-09-05 18:08:48 浏览: 112
在application.yml中配置多数据源,你可以按照以下方式进行设置:
```yaml
spring:
datasource:
primary:
url: jdbc:mysql://localhost:3306/primary_db
username: primary_user
password: primary_password
secondary:
url: jdbc:mysql://localhost:3306/secondary_db
username: secondary_user
password: secondary_password
# 配置多数据源
mybatis:
configuration:
# 设置驼峰命名规则
map-underscore-to-camel-case: true
mapper-locations: classpath:mapper/*.xml
# 配置数据源切换
datasource:
primary: primary
secondary: secondary
```
在上面的配置中,我们定义了两个数据源:primary和secondary。你可以根据需求添加更多的数据源。
接下来,你需要在Spring Boot的配置类中创建两个数据源的Bean,并将它们注入到数据源路由器中:
```java
@Configuration
public class DataSourceConfig {
@Bean(name = "primary")
@ConfigurationProperties(prefix = "spring.datasource.primary")
public DataSource primaryDataSource() {
return DataSourceBuilder.create().build();
}
@Bean(name = "secondary")
@ConfigurationProperties(prefix = "spring.datasource.secondary")
public DataSource secondaryDataSource() {
return DataSourceBuilder.create().build();
}
@Bean
@Primary
public DataSourceRouting dataSource(@Qualifier("primary") DataSource primaryDataSource,
@Qualifier("secondary") DataSource secondaryDataSource) {
Map<Object, Object> targetDataSources = new HashMap<>();
targetDataSources.put("primary", primaryDataSource);
targetDataSources.put("secondary", secondaryDataSource);
DataSourceRouting dataSourceRouting = new DataSourceRouting();
dataSourceRouting.setTargetDataSources(targetDataSources);
dataSourceRouting.setDefaultTargetDataSource(primaryDataSource);
return dataSourceRouting;
}
}
```
在上面的代码中,我们创建了名为primary和secondary的数据源Bean,并使用@ConfigurationProperties注解设置它们的属性。然后,我们将这些数据源注入到DataSourceRouting中,并根据需要设置默认数据源。
最后,你需要创建一个数据源路由器(DataSourceRouting)的类,用于实现动态切换数据源的逻辑:
```java
public class DataSourceRouting extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDataSource();
}
}
```
在上面的代码中,我们重写了determineCurrentLookupKey()方法,通过DataSourceContextHolder获取当前线程的数据源标识,并返回给Spring框架。
通过以上步骤,你就成功配置了多数据源。在需要切换数据源的地方,你可以使用DataSourceContextHolder.setDataSource("primary")或DataSourceContextHolder.setDataSource("secondary")来切换数据源。记得在每次请求结束后,使用DataSourceContextHolder.clearDataSource()来清除数据源标识,以免影响其他请求的数据源切换。
希望以上信息能够帮助到你!如有疑问,请随时提问。
阅读全文