spring动态创建数据源

在最近的项目业务中，需要在程序的运行过程中，添加新的数据库添链接进来，然后从新数据库链接中读取数据。

网上查阅了资料，发现spring为多数据源提供了一个抽象类AbstractRoutingDataSource，该类中只有一个抽象方法determineCurrentLookupKey()需要由我们实现。

以下是使用方法

假设我们创建一个类DynimaticDataSource，继承AbstractRoutingDataSource，并重写determineCurrentLookupKey()方法。

spring启动初始化DynimaticDataSource：

1、可以通过spring的自动注入，对AbstractRoutingDataSource类注入相应的属性，注入属性不是必须的，可以通过继承的子类重写这些方法来重新设置对这些属性的调用。

2、注入后spring会执行protected方法afterPropertiesSet()，该方法先判断属性targetDataSources不能为null，即必须初始化时注入至少一个数据源，否则抛出异常"Property 'targetDataSources' is required"。然后将该属性(类型为map<object,object>)的值全部转换到属性resolvedDataSources(类型为map<Object, DataSource>)中去。如果属性defaultTargetDataSource不为null，即已经设置默认数据源，将其转换为DataSource类型并赋值给属性defaultTargetDataSource。

经过以上处理后，属性resolvedDataSources中会被存放我们添加的数据源，该属性是一个map集合，key为Object类型，value为数据源。同时可能会有一个默认数据源(可以注入也可以不注入)。

使用DynimaticDataSource类获取连接：

1、调用该类的public方法getConnection()来获取连接。

2、getConnection在抽象类中被重写，会先调用protected方法determineTargetDataSource()。该方法先判断属性resolvedDataSources不为null，即初始化时候注入了至少一个数据源，否则抛出异常"DataSource router not initialized"。然后调用由子类重写的抽象方法determineCurrentLookupKey()获取dataSource在resolvedDataSources中对应的key，判断使用哪个数据源。

3、根据key从resolvedDataSources中获取数据源，如果resolvedDataSources中不存在，再判断lenientFallback为true(默认为true，可以设置)或key为null，返回默认数据源resolvedDefaultDataSource。否则抛出异常"Cannot determine target DataSource for lookup key [" + key+ "]"。

4、调用获取数据源的getConnection()方法获取连接。

在初始化时指定多数据源案例代码：

1、创建一个类DynimaticDataSource，继承AbstractRoutingDataSource，并重写determineCurrentLookupKey()方法。该方法负责判断当前线程使用哪一种数据源。这是最简单的一种实现方法，不重写任何非抽象方法，但必须在初始化时配置至少一个的数据源。

public class DynamicDatasource extends AbstractRoutingDataSource{

    private static Map<Object, Object> targetDataSources;
    
    protected DruidDataSource dataSource;
    
    @Override
    protected String determineCurrentLookupKey() {
　　　　　　//指定使用哪个数据源
        return DataSourceUtils.getDbtype();
    }
}

2、srping配置文件(部分)

    <bean id="dataSource" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close">
        <property name="url" value="${jdbc.url}" />
        <property name="username" value="${jdbc.username}" />
        <property name="password" value="${jdbc.password}" />
        <property name="driverClassName" value="${jdbc.driver}" />
        <property name="maxActive" value="50" /><!-- 最大连接池数 -->
        <property name="minIdle" value="5" /><!-- 最小连接池数 -->
    </bean>
    
    <bean id="dataSource2" class="com.alibaba.druid.pool.DruidDataSource" destroy-method="close">
        <property name="url" value="${jdbc.url2}" />
        <property name="username" value="${jdbc.username2}" />
        <property name="password" value="${jdbc.password2}" />
        <property name="driverClassName" value="${jdbc.driver}" />
        <property name="maxActive" value="50" /><!-- 最大连接池数 -->
        <property name="minIdle" value="5" /><!-- 最小连接池数 -->
    </bean>
    
    <bean id="multDataSource" class="com.ftpSystem.dao.DynamicDatasource">
        <property name="targetDataSources">
            <map >
                <entry value-ref="dataSource" key="masterDataSource"></entry>
                <entry value-ref="dataSource2" key="yangDataSource"></entry>
            </map>
        </property>
　　　　<property name="defaultTargetDataSource" ref="dataSource"></property>

    </bean>

3、使用数据源，我这里使用jdbcTemple，可以使用其他持久层框架，方式同单数据源配置一致。

    <bean id="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">
        <property name="dataSource" ref="multDataSource"></property>
    </bean>

4、切换数据源类

public class DataSourceUtils {

    private static final ThreadLocal<String> local = new ThreadLocal<String>();
    
    public static String getDbtype() {
        return local.get();
    }
    
    public static void setDbtype(String dbtype) {
        local.set(dbtype);
    }
    
    public static void clear() {
        local.remove();
    }
}

5、客户端代码

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("classpath:applicationContext-dao.xml")
public class TestDao {

    @Autowired
    private JdbcTemplate jdbcTemplate;
    
    @Test
    public void Test(){
        System.out.println("ok");
        int i = (Integer) jdbcTemplate.queryForObject("select count(*) from t_gg_zsdw", Integer.class);
        System.out.println("i="+i);
        DataSourceUtils.setDbtype("dataSource2");//切换数据源至dataSource2
        i = (Integer) jdbcTemplate.queryForObject("select count(*) from t_gg_zsdw", Integer.class);
        System.out.println("i="+i);
    }
}

由于我们已经在spring的配置文件中指定了属性defaultTargetDataSource，因此程序会默认使用该数据源。然后我们执行一次后切换为dataSource2，之后的执行会改为使用dataSource2的数据源。

以上方式可以对程序配置多数据源，但是缺点在程序初始化之时就指定所有的数据源，无法在运行时动态添加。

程序动态配置多数据源：

通过最上面我们对获取DynimaticDataSource对象的数据源连接的过程分析可知，AbstractRoutingDataSource类是通过determineTargetDataSource()方法来获取数据源。在该方法里面，又使用了我们重写的抽象方法determineCurrentLookupKey()来判断使用哪一个数据源的key，通过这个key，在数据源初始化后存放的集合resolvedDataSources中获取想要的数据源。

因此，如果我们想要动态的添加数据源进去，有两种思路可以考虑，一是重写determineTargetDataSource()方法，并且我们自己配置一个数据源集合，通过该方法，调用我们自己的数据源集合中对应的数据源。二是在resolvedDataSources属性中动态添加key-value进去，可以在determineTargetDataSource()方法中获取该数据源即可。

通过对源码查看可知，第二种思路所需的属性resolvedDataSources是私有的，且类中没有提供相应的get方法获取。因此只能采用第一种方式。

样例如下：

先是继承AbstractRoutingDataSource的实体类

/**
 * 数据源集合管理类
 * 使用数据源之前必须先添加，然后指定使用哪个数据源
 * @author yangxiangfeng
 *
 */
public class DynamicDatasource extends AbstractRoutingDataSource {

    private static Map<String, DataSource> dataSourceMap = new HashMap<String, DataSource>();
    
    private final static Map<DataSourceEntity, String> dseMap = new HashMap<>();
    
　　//spring注入需要set方法，不是必要的，可以注入也可以不注入
    public static void setDataSourceMap(Map<String, DataSource> dataSourceMap) {
        DynamicDatasource.dataSourceMap = dataSourceMap;
    }
    
　　//获取数据源信息集合
    public static Map<DataSourceEntity, String> getDsemap() {
        return dseMap;
    }

　　//检查是否包含指定id的数据源
    public static boolean checkDbKey(String dbKey){
        if(dataSourceMap.get(dbKey) != null)return true;
        return false;
    }
    //抽象方法，必须重写，用来判断使用哪个数据源
    @Override
    protected String determineCurrentLookupKey() {
        return DataSourceUtils.getDbKey();
    }
    
    /**
     * 对数据源的初始化方法，由于这里已经将数据源集合放在本类中，如果不重写将会由于父类参数为null而抛出异常。
     */
    @Override
    public void afterPropertiesSet() {}
    
    /**
     * 确定使用哪一个数据源
     * 这里不做null判断，因为是经过null判断后再进入的。
     */
    @Override
    protected DataSource determineTargetDataSource() {
        System.out.println("tttt");
        String dsKey = determineCurrentLookupKey();
        DataSource dds = dataSourceMap.get(dsKey);
        return dds;
    }

    /**
     * 添加数据源
     * 为了防止多线程添加同一个数据源，这里采用同步,同时会判断是否已存在
     * @param dbkey
     * @param ip
     * @param port
     * @param service 实例名
     * @param username
     * @param password
     * @return String 新建数据源对应的key，如果已经存在，则返回之前的key。
     */
    public synchronized String addDataSource(String dbkey, String ip, 
            int port, String service, String username, String password){
        DataSourceEntity d1 = new DataSourceEntity(ip, port, service, username);
        String value = dseMap.get(d1);
        if(dseMap.get(d1) != null){
            return value;//已存在则返回该数据源的id
        }
        
        DataSource ds = createDataSource(ip, port, service, username, password);
        dataSourceMap.put(dbkey, ds);//存储数据源集合
        dseMap.put(d1, dbkey);//保存已经存储了哪些数据源
        
        return dbkey;
    }
    
    /**
     * 创建一个数据源
     * @param ip
     * @param port
     * @param service
     * @param username
     * @param password
     * @return
     */
    private DataSource createDataSource(String ip, int port, String service, String username, String password){
        DruidDataSource dds = new DruidDataSource();
        dds.setDriverClassName("oracle.jdbc.driver.OracleDriver");
        dds.setUrl("jdbc:oracle:thin:@"+ip+":"+port+":"+service);
        dds.setUsername(username);
        dds.setPassword(password);
        return dds;
    }

2、切换数据源的工具类

public class DataSourceUtils {

    private static final ThreadLocal<String> local = new ThreadLocal<String>();
    
    public static String getDbKey() {
        return local.get();
    }
    
    public static void setDbKey(String dbKey) {
        if(DynamicDatasource.checkDbKey(dbKey)){
            local.set(dbKey);
        } else {
            throw new NullPointerException("不存在id为""+dbKey+""的数据源！");
　　　　 }
    }
    
    public static void clear() {
        local.remove();
    }
}

客户端测试类

    @Test
    public void Test(){
        System.out.println("ok");
        //增加数据源
        dynamicDatasource.addDataSource("dataSource", "***.***.**.***", ****, "***", "***", "*****");
        //指定使用的数据源
        DataSourceUtils.setDbKey("dataSource");
        //执行sql
        int i = (Integer) jdbcTemplate.queryForObject("select count(*) from tablename", Integer.class);
        System.out.println("i="+i);
    }    

以上代码是经过测试后可以直接运行的，由于本人能力有限，综合考虑后对spring中动态创建数据源的实现如上。欢迎各位指点可以优化改进的地方。