48 lines
1.7 KiB
Java
48 lines
1.7 KiB
Java
package com.ruoyi.framework.config;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import javax.sql.DataSource;
|
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
|
|
import org.springframework.boot.context.properties.ConfigurationProperties;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
import org.springframework.context.annotation.Primary;
|
|
import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;
|
|
import com.ruoyi.framework.aspectj.lang.enums.DataSourceType;
|
|
import com.ruoyi.framework.datasource.DynamicDataSource;
|
|
|
|
/**
|
|
* druid 配置多数据源
|
|
*
|
|
* @author ruoyi
|
|
*/
|
|
@Configuration
|
|
public class DruidConfig
|
|
{
|
|
@Bean
|
|
@ConfigurationProperties("spring.datasource.druid.master")
|
|
public DataSource masterDataSource()
|
|
{
|
|
return DruidDataSourceBuilder.create().build();
|
|
}
|
|
|
|
@Bean
|
|
@ConfigurationProperties("spring.datasource.druid.slave")
|
|
@ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")
|
|
public DataSource slaveDataSource()
|
|
{
|
|
return DruidDataSourceBuilder.create().build();
|
|
}
|
|
|
|
@Bean(name = "dynamicDataSource")
|
|
@Primary
|
|
public DynamicDataSource dataSource(DataSource masterDataSource, DataSource slaveDataSource)
|
|
{
|
|
Map<Object, Object> targetDataSources = new HashMap<>();
|
|
targetDataSources.put(DataSourceType.MASTER.name(), masterDataSource);
|
|
targetDataSources.put(DataSourceType.SLAVE.name(), slaveDataSource);
|
|
return new DynamicDataSource(masterDataSource, targetDataSources);
|
|
}
|
|
}
|