动态数据源切换类AbstractRoutingDataSource

最新推荐文章于 2025-10-22 12:24:47 发布

原创

最新推荐文章于 2025-10-22 12:24:47 发布 · 352 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #数据库 #开发语言

本文详细介绍了如何在Spring框架中使用AbstractRoutingDataSource实现动态数据源切换，包括其工作原理、源码分析和实际应用示例，重点讲解了如何通过键值对管理多个数据源并根据业务需求进行切换。

写在前面

在工作中为了能够提高数据库的读写能力，经常会用到分库分表等技术，此时不可避免的就要涉及到动态数据源切换的内容，针对这个问题，spring提供了AbstractRoutingDataSource类来满足我们的需求，本文就一起来看下。

1：动态切换的原理

任何技术，原理都是最重要的，知道了原理，解决问题就只是时间问题了，所以，我们有必要先来看下数据源能够实现动态切换的原理，首先看下Datasource类源码，如下：

package javax.sql;

public interface DataSource  extends CommonDataSource, Wrapper {

  Connection getConnection() throws SQLException;
  Connection getConnection(String username, String password)
    throws SQLException;
}

可以看到，DataSource本质上就是一个jdbc connnection的工厂类，因此不同的datasource，其实就是不同的datasoruce实例了，对应到spring就是不同的datasource的bean，诸如c3p0，druid,hikaricp等连接池技术的连接池对象都是实现了该接口的，如下是druid的：
在这里插入图片描述

所以，在真正的操作数据之前，我们只需要更换datasource的实现类，那么最终获取到jdbc connection就是对应数据库的connection了，就可以操作对应的数据库，从而实现动态切换了，那么具体怎么做呢？可以这样，用一个map来维护一组数据源，之后根据用户指定的key来获取对应的数据源，就行了，伪代码如下:

void changeDataSource(String userSpecifiedKey) {
    Map datsourceMap = new HashMap();
    {
        datsourceMap.put("key1", new YourDatasource1());
        datsourceMap.put("key2", new YourDatasource2());
        datsourceMap.put("key3", new YourDatasource3());
        ...
    }    
    changeDataSourceTo(datsourceMap.get(userSpecifiedKey));
}

本文我们要学习的AbstractRoutingDatasource使用的也正是这种思想，接着来看下。

2：AbstractRoutingDatasource

源码如下:

注意只保留重要源码！！！

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
   
   
    // 默认的数据源
	private Object defaultTargetDataSource;
    // 所有可用的数据源字典
	private Map<Object, DataSource> resolvedDataSources;

    // 重要！！！
    // spring bean生命周期中InitializingBean对应的方法，在该方法中设置外部指定的数据源到resolvedDataSources
	@Override
	public void afterPropertiesSet() {
   
   
		if (this.targetDataSources == null) {
   
   
			throw new IllegalArgumentException("Property 'targetDataSources' is required");
		}
		this.resolvedDataSources = new HashMap<Object, DataSource>(this.targetDataSources.size());
		for (Map.Entry<Object, Object> entry : this