博客

在spring中有一个抽象类AbstractRoutingDataSource类，通过这个类可以实现动态数据源切换。如下是这个类的成员变量

// 是存储将要切换的多数据源bean信息，一般是<beanName,DataSource-bean>
private Map<Object, Object> targetDataSources;
private Object defaultTargetDataSource;
private Map<Object, DataSource> resolvedDataSources;

在其afterPropertiesSet方法中。

@Override
public void afterPropertiesSet() {
	if (this.targetDataSources == null) {
		throw new IllegalArgumentException("Property 'targetDataSources' is required");
	}
	this.resolvedDataSources = new HashMap<Object, DataSource> (this.targetDataSources.size()); //初始化resolvedDataSources 
    // 1. 循环原有存储多数据源bean信息的targetDataSources
	for (Map.Entry<Object, Object> entry : this.targetDataSources.entrySet()) {
        // 2. 判断lookUpKey要不要重写，一般都是beanName，不用重写
		Object lookupKey = resolveSpecifiedLookupKey(entry.getKey());
        // 3. 判断value是否Datasource对象，不是则生成DataSource对象，一般都是
		DataSource dataSource = resolveSpecifiedDataSource(entry.getValue());
        // 4. 将魔改的<BeanName,DataSource> 存入resolvedDataSources中备用
		this.resolvedDataSources.put(lookupKey, dataSource);
	}
        //如果默认数据源不为空则指定对应数据源
	if (this.defaultTargetDataSource != null) {
		this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource);
	}
}

连接数据库的getConnection()方法，调用的是determineTargetDataSource()方法，来创建连接。

@Override
public Connection getConnection() throws SQLException {
	return determineTargetDataSource().getConnection();
}
 
@Override
public Connection getConnection(String username, String password) throws SQLException {
	return determineTargetDataSource().getConnection(username, password);
}

而determineTargetDataSource()方法是决定spring容器连接那个数据源。

而选择哪个数据源又是由determineCurrentLookupKey()方法来决定的，此方法是抽象方法，需要我们继承AbstractRoutingDataSource抽象类来重写此方法。该方法返回一个key，该key是bean中的beanName，并赋值给lookupKey，由此key可以通过resolvedDataSources属性的键来获取对应的DataSource值，从而达到数据源切换的功能。

protected DataSource determineTargetDataSource() {
	Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
    //具体选择哪个数据源
	Object lookupKey = determineCurrentLookupKey();
	DataSource dataSource = this.resolvedDataSources.get(lookupKey);
	if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
		dataSource = this.resolvedDefaultDataSource;
	}
	if (dataSource == null) {
		throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
	}
	return dataSource;
}