Spring Boot多数据源配置

@Configuration
public class DataSourceConfig {
    @Bean("datasource")
    @Primary  // 这个注解表示主数据源
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource datasource() {
        // 这里也可以使用其他连接池, 比如 DruidDataSource
        return new DruidDataSource();
    }
    @Bean("ds1")
    @ConfigurationProperties(prefix = "custom.datasource.ds1")
    public DataSource ds1() {
        return DataSourceBuilder.create().build();
    }
    @Bean("ds2")
    @ConfigurationProperties(prefix = "custom.datasource.ds2")
    public DataSource ds2() {
        return DataSourceBuilder.create().build();
    }
}

// DS0Config.java
@Configuration
@MapperScan(basePackages = "com.example.demo.dao.ds0", sqlSessionFactoryRef = "ds0SqlSessionFactory")
public class DS0Config {
    @Bean
    @Primary  // 这个数据源设置为主要数据源
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource ds0() {
        return new DruidDataSource();
    }
    @Bean  // 如果这里不用 Qualifier 指定, 则注入的是Primary数据源
    public SqlSessionFactory ds0SqlSessionFactory(@Qualifier("ds0") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(dataSource);
        // 领域模型包位置
        sessionFactory.setTypeAliasesPackage("com.example.demo.domain");
        // 设置映射文件的位置
        sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:sqlmaps/ds0/*.xml"));
        return sessionFactory.getObject();
    }
    @Bean
    public SqlSessionTemplate ds0SqlSessionTemplate(@Qualifier("ds0SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

// DS1Config.java
@Configuration
@MapperScan(basePackages = "com.example.demo.dao.ds1", sqlSessionFactoryRef = "ds1SqlSessionFactory")  // 这个包里的mapp使用的是ds1数据源
public class DS1Config {
    @Bean
    @ConfigurationProperties(prefix = "custom.datasource.ds1")  // 自动会把属性注入到对应的字段
    public DataSource ds0() {
       return DataSourceBuilder.create().build();
    }
    @Bean
    public SqlSessionFactory ds1SqlSessionFactory(@Qualifier("ds1") DataSource dataSource) throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(dataSource);
        // 领域模型包位置
        sessionFactory.setTypeAliasesPackage("com.example.demo.domain");
        // 设置映射文件的位置
        sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:sqlmaps/ds1/*.xml"));
        return sessionFactory.getObject();
    }
    @Bean
    public SqlSessionTemplate ds1SqlSessionTemplate(@Qualifier("ds1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
}

// 这个是注入默认的数据源
@Resource
private DataSource dataSource
// 注入ds1数据源
@Resource("ds1")
private DataSource ds1
// 在方法里注入ds2数据源
@Bean
public SqlSessionFactory sqlSessionFactory(@Qualifier("ds2") DataSource dataSource) {
    ....
}

@Component
public class MultiDataSourceRegister implements BeanDefinitionRegistryPostProcessor, EnvironmentAware {
    private static final Logger logger = LoggerFactory.getLogger(MultiDataSourceRegister.class);
    // 存放DataSource配置的集合, <dsName, dbProperties>
    private Map<String, PropertyValues> dataSourceMap = Maps.newHashMap();

    /**
     * 这个方法主要用于加载多数据源配置, 添加到dataSourceMap中, 之后在postProcessBeanDefinitionRegistry进行注册。
     */
    @Override
    public void setEnvironment(Environment environment) {
        // 获取到前缀是 "custom.datasource." 的属性列表值
        RelaxedPropertyResolver propertyResolver = new RelaxedPropertyResolver(environment, "custom.datasource.");
        // 获取到所有数据源的名称列表
        String dsNames = propertyResolver.getProperty("names"); // 拿到 custom.datasource.names 定义的数据源列表
        for (String dsName : Splitter.on(",").omitEmptyStrings().splitToList(dsNames)) {
            // 把数据源的配置(url, username, password, driver-class-name, type等)咱存到 dsMap 中
            Map<String, Object> dsMap = propertyResolver.getSubProperties(dsName + ".");
            if (CollectionUtils.isEmpty(dsMap)) {
                logger.warn("未找到数据源{}的属性配置", dsName);
            } else {
                PropertyValues propertyValues = new MutablePropertyValues(dsMap);
                dataSourceMap.put(dsName, propertyValues);  // 数据源 dsName 的所有配置
            }
        }
    }

    /**
     * 自动注册 dataSourceMap 中的所有数据源
     */
    @Override
    @SuppressWarnings("unchecked")
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {
        for (String dsName : dataSourceMap.keySet()) {
            PropertyValues pv = dataSourceMap.get(dsName);
            DataSourceBuilder dataSourceBuilder = DataSourceBuilder.create();  // 这里并没有创建数据源, 只是依靠Spring拿到数据源类型
            if (pv.contains("type")) {
                String type = pv.getPropertyValue("type").getValue().toString();  // 如果指定了type, 则使用指定的数据源类型
                try {
                    dataSourceBuilder.type((Class<? extends DataSource>) Class.forName(type));
                } catch (ClassNotFoundException e) {
                    logger.error("加载数据源{}失败, 使用默认数据源");
                }
            }
            // 注册
            AnnotatedGenericBeanDefinition definition = new AnnotatedGenericBeanDefinition(dataSourceBuilder.findType());
            registry.registerBeanDefinition(dsName, definition);
        }
    }

    /**
     * 把自定义数据源的属性绑定到对应的bean上
     */
    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        // 把默认数据源设置为主数据源;
        beanFactory.getBeanDefinition("dataSource").setPrimary(true);
        // 设置自定义数据源的属性绑定
        for (String dsName : dataSourceMap.keySet()) {
            DataSource customDS = beanFactory.getBean(dsName, DataSource.class);
            RelaxedDataBinder dataBinder = new RelaxedDataBinder(customDS);
            dataBinder.bind(dataSourceMap.get(dsName));
        }
    }
}