cn.opencodes.framework.autoconfigure.DynamicDataSourceConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of alpha-spring-boot-starter Show documentation

简化配置、快速开发app、pc端项目

There is a newer version: 1.0.4

Show newest version

package cn.opencodes.framework.autoconfigure;

import java.util.HashMap;
import java.util.Map;


import javax.sql.DataSource;


import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.transaction.annotation.EnableTransactionManagement;


import cn.opencodes.framework.autoconfigure.properties.DynamicDataSourceProperties;
import cn.opencodes.framework.core.datasources.DynamicDataSource;
import cn.opencodes.framework.tools.utils.BeanUtils;

import cn.opencodes.framework.tools.utils.StringUtils;

import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceBuilder;

/**
 * 配置多数据源
 * @author HJ
 */
@EnableTransactionManagement(order = 2)
public class DynamicDataSourceConfiguration {

	@Bean
	public DynamicDataSource dataSource(DynamicDataSourceProperties properties, DataSourceProperties parentProperties) {
		Map dataSourcePropertiesMap = properties.getDynamic();
        Map targetDataSources = new HashMap<>(dataSourcePropertiesMap.size());
        final String[] tmp = new String[1];
        dataSourcePropertiesMap.forEach((k, v) -> {
        	DruidDataSource druidDataSource = DruidDataSourceBuilder.create().build();
        	BeanUtils.copyBeanNotNull2Bean(properties, druidDataSource);
        	BeanUtils.copyBeanNotNull2Bean(parentProperties, druidDataSource);
        	BeanUtils.copyBeanNotNull2Bean(v, druidDataSource);
            targetDataSources.put(k, druidDataSource);
            // 获取第一个数据源名称为默认数据源
            tmp[0] = StringUtils.isEmpty(tmp[0]) ? k : tmp[0];
        });
        
        DruidDataSource master = (DruidDataSource)targetDataSources.get(tmp[0]);
        return new DynamicDataSource(master, targetDataSources);
    }
    
}