SpringBoot多数据源切换，AOP实现动态数据源切换

SpringBoot多数据源切换，AOP实现动态数据源切换

操作数据一般都是在DAO层进行处理，可以选择直接使用JDBC进行编程
或者是使用多个DataSource 然后创建多个SessionFactory，在使用Dao层的时候通过不同的SessionFactory进行处理，不过这样的入侵性比较明显，一般的情况下我们都是使用继承HibernateSupportDao进行封装了的处理，如果多个SessionFactory这样处理就是比较的麻烦了，修改的地方估计也是蛮多的
最后一个，也就是使用AbstractRoutingDataSource的实现类通过AOP或者手动处理实现动态的使用我们的数据源，这样的入侵性较低，非常好的满足使用的需求。比如我们希望对于读写分离或者其他的数据同步的业务场景

●单数据源的场景（一般的Web项目工程这样配置进行处理，就已经比较能够满足我们的业务需求）
多数据源多SessionFactory这样的场景，估计作为刚刚开始想象想处理在使用框架的情况下处理业务，配置多个SessionFactory，然后在Dao层中对于特定的请求，通过特定的SessionFactory即可处理实现这样的业务需求，不过这样的处理带来了很多的不便之处，所有很多情况下我们宁愿直接使用封装的JDBC编程，或者使用Mybatis处理这样的业务场景

●使用AbstractRoutingDataSource 的实现类，进行灵活的切换，可以通过AOP或者手动编程设置当前的DataSource，不用修改我们编写的对于继承HibernateSupportDao的实现类的修改，这样的编写方式比较好，至于其中的实现原理，让我细细到来。我们想看看如何去应用，实现原理慢慢的说！

●编写AbstractRoutingDataSource的实现类,HandlerDataSource就是提供给我们动态选择数据源的数据的信息，我们这里编写一个根据当前线程来选择数据源，然后通过AOP拦截特定的注解，设置当前的数据源信息，也可以手动的设置当前的数据源，在编程的类中。

/**
 * 动态数据源
 *
 * @author
 * @date 2017年3月5日 上午9:11:49
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
	private Map<Object, Object> allResolvedDataSources = new HashMap<>();

	@Override
	protected Object determineCurrentLookupKey() {
		return DataSourceContextUtil.getDataSourceType();
	}

	/**
	 * 添加数据源
	 * @param name
	 * @param dataSource
	 */
	public void addDataSource(Object name,DataSource dataSource){
		if(allResolvedDataSources.containsKey(name))
			allResolvedDataSources.remove(name);
		allResolvedDataSources.put(name,dataSource);
		setTargetDataSources(allResolvedDataSources);
	}
        /**
         * 删除数据源
         * @param name
         * @return
         */
	    public Object deleteDataSource(Object name){
		Object rst = null;
		if(allResolvedDataSources.containsKey(name))
		{
			rst = allResolvedDataSources.get(name);
			allResolvedDataSources.remove(name);
			setTargetDataSources(allResolvedDataSources);
		}
		return rst;
	    }

        /**
         * 检查数据源是否存在
         * @param name
         * @return
         */
	    public Boolean isExistDataSource(Object name){
	        return allResolvedDataSources.containsKey(name);
            }
}

●设置动态选择的Datasource，这里的Set方法可以留给AOP调用，或者留给我们的具体的Dao层或者Service层中手动调用，在执行SQL语句之前。

/**
 * 动态数据源
 *
 * @author
 * @date 2017年3月5日 上午9:11:49
 */
public class DynamicDataSource extends AbstractRoutingDataSource {
	private Map<Object, Object> allResolvedDataSources = new HashMap<>();

	@Override
	protected Object determineCurrentLookupKey() {
		return DataSourceContextUtil.getDataSourceType();
	}

	/**
	 * 添加数据源
	 * @param name
	 * @param dataSource
	 */
	public void addDataSource(Object name,DataSource dataSource){
		if(allResolvedDataSources.containsKey(name))
			allResolvedDataSources.remove(name);
		allResolvedDataSources.put(name,dataSource);
		setTargetDataSources(allResolvedDataSources);
	}
        /**
         * 删除数据源
         * @param name
         * @return
         */
	public Object deleteDataSource(Object name){
		Object rst = null;
		if(allResolvedDataSources.containsKey(name))
		{
			rst = allResolvedDataSources.get(name);
			allResolvedDataSources.remove(name);
			setTargetDataSources(allResolvedDataSources);
		}
		return rst;
	}

        /**
         * 检查数据源是否存在
         * @param name
         * @return
         */
	public Boolean isExistDataSource(Object name){
	    return allResolvedDataSources.containsKey(name);
        }
}

●在项目启动的时候初始化所有的数据源信息，通过DynamicDataSource类以键值对的形式存储

        DruidDataSource dataSource = dataSource();
        DruidDataSource masterDataSource = masterDataSource();
        DruidDataSource clusterDataSource = clusterDataSource();
        //DruidDataSource bizDataSource = bizDataSource();

        try {
            dataSource.init();
            clusterDataSource.init();
            masterDataSource.init();

         //   bizDataSource.init();
        }catch (SQLException sql){
            sql.printStackTrace();
        }

        DynamicDataSource dynamicDataSource = new DynamicDataSource();
        dynamicDataSource.addDataSource(DSEnum.DATA_SOURCE_NAME,dataSource);                     dynamicDataSource.addDataSource(DSEnum.MASTER_DATA_SOURCE_NAME,masterDataSource);
     dynamicDataSource.addDataSource(DSEnum.CLUSTER_DATA_SOURCE_NAME,clusterDataSource);
 DataSourceContextUtil.setDataSourceType(DSEnum.DATA_SOURCE_NAME);//设置默认的数据源的Key
 dynamicDataSource.setDefaultTargetDataSource(dataSource);//设置默认的数据源

●代码使用方法，注意的是这里使用过后需要调用清空的方法，让其返回默认的数据源，否则将会带着这个数据源满世界的报错找不到表，找不到列，映射不到类。。。。。。

DataSourceContextUtil.setDataSourceType(DSEnum.CLUSTER_DATA_SOURCE_NAME);
        String sql2="select * from sys_config";
     List<Map<String,Object>>sqlListPage2=commonDao.getSqlListNotPage(sql2.toString(),null);
        DataSourceContextUtil.clearDataSourceType();

下面是我使用的枚举类型，用来当做Key来使用的

/**
 * 
 * 多数据源的枚举
 *
 * @author
 * @date 2017年3月5日 上午10:15:02
 */
public interface DSEnum {

	String DATA_SOURCE_NAME = "dataSource";		//base数据源

	String MASTER_DATA_SOURCE_NAME = "masterDataSource";		//base数据源

	String CLUSTER_DATA_SOURCE_NAME = "clusterDataSource";		//base数据源
}

SpringBoot多数据源切换，AOP实现动态数据源切换

猜你喜欢