文档章节

动态数据源

Raymond
 Raymond
发布于 2017/01/11 14:40
字数 481
阅读 21
收藏 0

    在非互联网的项目中,有时候需要跨库使用数据源。为了更加方便的使用,基本上都会开发一款动态数据源。以下是基于Spring的动态数据源源码:

public class DataSourceContext extends AbstractRoutingDataSource {

    private static final ThreadLocal<String> dataSourceKey = new InheritableThreadLocal<String>();

    public static void setDataSourceKey(String dataSource) {
        dataSourceKey.set(dataSource);
    }

    @Override
    protected Object determineCurrentLookupKey() {
        return dataSourceKey.get();
    }

}

    继承AbstractRoutingDataSource, 再使用数据源之前进行指定,可以手动指定如:

DataSourceContext.setDataSourceKey(dataSourceKey);

    也可以通过AOP的方式实现,如:

public class SpecifyDataSourceAop {


    public void before(JoinPoint point) {
        MethodSignature methodSignature = (MethodSignature) point.getSignature();
        Method method = methodSignature.getMethod();
        SpecifyDataSource dataSource = method.getAnnotation(SpecifyDataSource.class);
        storeKey(dataSource);
        assignDataSourceKey();
    }


    public void after(JoinPoint point){
        removeKey();
        assignDataSourceKey();
    }

    private void storeKey(SpecifyDataSource dataSource){
        if(dataSource == null) {
            //if dataSource == null then assign defaultDataSource
            putDSKey2Store(DataSourceContants.DEFAULT_DATASOURCE);
        }else if (null != dataSource) {
            // choose SpecifyDataSource`s name
            putDSKey2Store(dataSource.name());
        }
    }

    private void putDSKey2Store(String key){
        DATASOURCE_KEYS.get().addLast(key);
    }

    private String assignDataSourceKey(){
        LinkedList<String> keys = DATASOURCE_KEYS.get();
        DataSourceContext.setDataSourceKey(dataSource);
        return dataSource;
    }

    private String removeKey(){
        return DATASOURCE_KEYS.get().removeLast();
    }

    private static final ThreadLocal<LinkedList<String>> DATASOURCE_KEYS = new ThreadLocal<LinkedList<String>>(){
        @Override
        protected LinkedList<String> initialValue() {
            return new LinkedList<String>();
        }
    };
}

    默认为dataSource 数据源,使用时如果在接口层加上注解 @SpecifyDataSource 并指定数据源 (name="xxDataSource"), 在调用此接口实现的方法时,会切换到xxDataSource,调用完成后切回上一个数据源。

注解类为

@Target({ElementType.METHOD, ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface SpecifyDataSource {

    /**
     * 指定数据源
     * @return
     */
    String name() default "dataSource";

}

    针对此动态数据源的配置文件如下:

<!-- 配置多数据源映射关系 -->
<bean  class="org.nforgo.component.datasource.DataSourceContext">
    <property name="targetDataSources">
        <map key-type="java.lang.String">
            <entry key="dataSource" value-ref="dataSource"></entry>
            <entry key="ucDataSource" value-ref="xxDataSource"></entry>
        </map>
    </property>
    <!-- 默认目标数据源为你主库数据源 -->
    <property name="defaultTargetDataSource" ref="dataSource"/>
</bean>

<bean id="specifyDataSourceAop" class="org.nforgo.component.datasource.SpecifyDataSourceAop"></bean>
<!-- AOP拦截配置 -->
<aop:config>
    <aop:aspect id="dataSourceAspect" ref="specifyDataSourceAop" order="1">
        <aop:pointcut id="dsPoint" expression="execution(* org..*.business.manage..*.*(..))"/>
        <aop:before method="before" pointcut-ref="dsPoint"/>
        <aop:after-returning method="after" pointcut-ref="dsPoint"/>
        <!--<aop:after-throwing method="after" pointcut-ref="dsPoint"/>-->
    </aop:aspect>
</aop:config>

     此动态数据源还有很多需要更新的地方,在真正使用时再做二次开发,如事务的处理,在接口层之外的数据源个性化指定等等。

    

 

© 著作权归作者所有

下一篇: docker初探
Raymond
粉丝 0
博文 11
码字总数 3228
作品 0
浦东
程序员
私信 提问
Spring Boot MyBatis 动态数据源切换、多数据源,读写分离

项目地址 https://github.com/helloworlde/SpringBoot-DynamicDataSource 本项目使用 Spring Boot 和 MyBatis 实现多数据源,动态数据源的切换;有多种不同的实现方式,在学习的过程中发现没...

呜呜呜啦啦啦
2017/12/21
0
0
Spring主从数据库的配置和动态数据源切换原理

Spring主从数据库的配置和动态数据源切换原理 廖雪峰的官方网站2017-11-123 阅读 spring动态数据源主从数据库切换 在大型应用程序中,配置主从数据库并使用读写分离是常见的设计模式。在Spr...

廖雪峰的官方网站
2017/11/12
0
0
JAVA中使用代码创建多数据源,并实现动态切换(一)

2017-06-06 11:31:57补充:近日,在本文的基础之上,扩展了下,使用atomikos来管理事务,保证多数据源操作时,事务一致性。(https://my.oschina.net/simpleton/blog/916108) 另外,感谢朋友...

十月阳光
2017/03/27
0
0
谈谈Spring Boot 数据源加载及其多数据源简单实现

业务需求 提供所有微服务数据源的图形化维护功能 代码生成可以根据选择的数据源加载表等源信息 数据源管理要支持动态配置,实时生效 附录效果图 实现思路 本文提供方法仅供类似简单业务场景,...

冷冷gg
04/12
0
0
Spring3.3 整合 Hibernate3、MyBatis3.2 配置多数据源/动态切换数据源 方法

一、开篇 这里整合分别采用了Hibernate和MyBatis两大持久层框架,Hibernate主要完成增删改功能和一些单一的对象查询功能,MyBatis主要负责查询功能。所以在出来数据库方言的时候基本上没有什...

ibm_hoojo
2013/10/12
0
0

没有更多内容

加载失败,请刷新页面

加载更多

任务调度-第三方库Quartz实现分布式任务管理与调度

1. 为什么要用第三方库Quartz来实现分布式任务管理和调度? 首先管理的目的是通过集群多节点的管理提供容错,调度的目的是保证同一任务只会被完整执行一次;之前分享过的任务调度-单体应用定...

秋日芒草
7分钟前
0
0
Mysql Explain Type

前言 当我们执行sql,一般都会用Explain来查看sql的效率如何。今天在看sql执行效率的时候,忘记了其中Type的意思,现在在此记录一下。 效率 这里的type指的是访问类型,各个效率高低如下: ...

无敌小杰杰
16分钟前
0
0
外部浏览器网页复制公众号无法自动唤起微信并关注怎么办?

现在有很多用户在外部浏览器网页复制公众号时无法自动唤起微信并关注,这是因为第三方浏览器打开微信的接口,微信只给部分合作平台开放了接口权限,任何第三方想调用只能是通过一些技术手段来...

qjniop
20分钟前
0
0
建造者模式

建造者模式(Builder Pattern) 也叫生成器模式,其定义如下: Separate the construction of a complex object from its representation so that the same construction process can create d......

无知的小狼
25分钟前
0
0
距离计算方法

1、欧式距离(欧几里得距离) 欧式距离是最易理解的距离定义,即各坐标点的坐标之差的平方和相加,然后开根号。 二维平面上点 与点 之间的距离公式是: n维空间上点 和点 之间的距离公式是:...

城北徐公美
27分钟前
0
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部