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数据源
}

 

©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页