我的Spring多数据源

Spring 多数据源已经不是什么稀奇的事了，在实际应用中主从数据库就需要用到多数据源的配置与动态切换。在搜索引擎中都可以找到很多资料，各也有各的做法，本文也就不做过多的阐述其原理，只是介绍下目前我在项目中对于多数据源的使用情况，欢迎大家前来拍砖。

先特别感谢好友 Tony

首先，继承spring-jdbc 的 AbstractRoutingDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {    
    protected Object determineCurrentLookupKey() {        
        return DataSourceHolder.getDataSources();    
    }
}

创建自己的DataSourceHolder

public class DataSourceHolder {    
    
    private static ThreadLocal<String> dataSources = new ThreadLocal<String>();    

    public static String getDataSources() {        
        String source = dataSources.get();        
        if (source == null) source = DataSourceType.MASTER;        
        return source;    
    }    
    public static void setDataSources(String source) {       
        dataSources.set(source);    
    }
}

当然还有DataSourceType

public class DataSourceType {    
    public static final String MASTER = "master";    
    public static final String SLAVE = "slaver";
}

到此为止，和所有搜索引擎中找到的资料一致，除了命名可能有些区别 _
那么接下来介绍一下不同的地方，网上也有很多资料是通过AOP来实现动态切换，比如：Spring 配置多数据源实现数据库读写分离 这篇文章中的介绍，就很完美的使用AOP实现了主从数据库的切换，而我的做法有一些大同小异，我使用了Java动态代理，先上代码。

先创建自己的Annotation：

从库注解

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

主库注解

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

为了以防万一，增加了OtherSource
而这个也只是为了防止在一个项目中有多个数据库的连接需求，比如：将用户信息独立成库，商品信息独立成库等

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface OtherSource {    
    String value();
}

以下则为真正重点——动态代理
继承MapperFactoryBean，实现InvocationHandler，在invoke之前，根据注解来实现动态数据源的切换

public class MapperFactoryBeanInvocation<T> extends MapperFactoryBean<T> implements InvocationHandler {    
    private T t;    
    public MapperFactoryBeanInvocation() {    }    
    public T getObject() throws Exception {        
        T t = super.getObject();        
        return this.getProxyObject(t);    
    }    
    private T getProxyObject(T t) {        
        this.t = t;        
        return (T) Proxy.newProxyInstance(t.getClass().getClassLoader(), t.getClass().getInterfaces(), this);    
    }    
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {        
        boolean isRead = true;        
        StringBuffer typeBuffer = new StringBuffer("");
        //获取是否有其他数据源
        OtherSource ortherSourceAnn = method.getAnnotation(OtherSource.class);        
        if (ortherSourceAnn != null) {            
            typeBuffer.append(ortherSourceAnn.value()).append("_");        
        }   
        //获取主从状态    
        Annotation ann_method = method.getAnnotation(Read.class);        
        if (ann_method == null) isRead = false;  
        if (isRead) typeBuffer.append(DataSourceType.SLAVE);        
        else typeBuffer.append(DataSourceType.MASTER);   
        //切换数据源    
        DataSourceHolder.setDataSources(typeBuffer.toString());        
        return method.invoke(this.t, args);    
    }
}

至此，Spring多数据源主从动态切换已经实现。
在实际使用过程中，对应的管理好多数据源配置，即可实现动态切换的目的

db.properties 配置

#jdbc driver
jdbc.driverClassName=com.mysql.jdbc.Driver
#master
design.master.jdbc.url=jdbc:mysql://192.168.1.101:3306/design_shop?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true
design.master.jdbc.username=root
design.master.jdbc.password=111111
#slaver
design.slaver.jdbc.url=jdbc:mysql://192.168.1.102:3306/design_shop?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true
design.slaver.jdbc.username=root
design.slaver.jdbc.password=111111
#master
shop.master.jdbc.url=jdbc:mysql://192.168.1.103:3306/shop_mall?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true
shop.master.jdbc.username=root
shop.master.jdbc.password=a12345678
#slaver
shop.slaver.jdbc.url=jdbc:mysql://192.168.1.104:3306/shop_mall?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true
shop.slaver.jdbc.username=root
shop.slaver.jdbc.password=a12345678
#c3p0 config
c3p0.initialPoolSize=8
c3p0.minPoolSize=5
c3p0.maxPoolSize=10
c3p0.acquireIncrement=10
c3p0.maxIdleTime=60
c3p0.idleConnectionTestPeriod=120
c3p0.maxStatements=100
c3p0.autoCommitOnClose=false
c3p0.testConnectionOnCheckout=false
c3p0.testConnectionOnCheckin=false
c3p0.preferredTestQuery=select now()

spring 配置

...
<bean id="parentDataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource" destroy-method="close">    
    <property name="driverClass" value="${jdbc.driverClassName}"/>    
    <property name="initialPoolSize" value="${c3p0.initialPoolSize}"/>    
    <property name="minPoolSize" value="${c3p0.minPoolSize}"/>    
    <property name="maxPoolSize" value="${c3p0.maxPoolSize}"/>    
    <property name="acquireIncrement" value="${c3p0.acquireIncrement}"/>    
    <property name="maxIdleTime" value="${c3p0.maxIdleTime}"/>    
    <property name="idleConnectionTestPeriod" value="${c3p0.idleConnectionTestPeriod}"/>    
    <property name="maxStatements" value="${c3p0.maxStatements}"/>    
    <property name="autoCommitOnClose" value="${c3p0.autoCommitOnClose}"/>   
    <property name="testConnectionOnCheckout" value="${c3p0.testConnectionOnCheckout}"/>    
    <property name="testConnectionOnCheckin" value="${c3p0.testConnectionOnCheckin}"/>    
    <property name="preferredTestQuery" value="${c3p0.preferredTestQuery}"/>
</bean>
<import resource="dataSource.xml"/>
<import resource="data_mapping.xml"/>
...

dataSource.xml 配置

...
<!-- 配置主库数据源 -->    
<bean id="designMasterDataSource" parent="parentDataSource">        
    <property name="jdbcUrl" value="${design.master.jdbc.url}"/>        
    <property name="user" value="${design.master.jdbc.username}"/>        
    <property name="password" value="${design.master.jdbc.password}"/>    
</bean>    
<!-- 配置从库数据源 -->    
<bean id="designSlaverDataSource" parent="parentDataSource">        
    <property name="jdbcUrl" value="${design.slaver.jdbc.url}"/>        
    <property name="user" value="${design.slaver.jdbc.username}"/>        
    <property name="password" value="${design.slaver.jdbc.password}"/>    
</bean>    
<!-- 配置主库数据源 -->    
<bean id="shopMasterDataSource" parent="parentDataSource">        
    <property name="jdbcUrl" value="${shop.master.jdbc.url}"/>        
    <property name="user" value="${shop.master.jdbc.username}"/>        
    <property name="password" value="${shop.master.jdbc.password}"/>    
</bean>    
<!-- 配置从库数据源 -->    
<bean id="shopSlaverDataSource" parent="parentDataSource">        
    <property name="jdbcUrl" value="${shop.slaver.jdbc.url}"/>        
    <property name="user" value="${shop.slaver.jdbc.username}"/>        
    <property name="password" value="${shop.slaver.jdbc.password}"/>    
</bean>    
<!-- 配置动态数据源 -->    
<bean id="designDataSource" class="com.design.datasource.DynamicDataSource">        
    <property name="targetDataSources">            
    <map key-type="java.lang.String">                
        <entry key="slaver" value-ref="designSlaverDataSource"/>                
        <entry key="master" value-ref="designMasterDataSource"/>                
        <entry key="shop_slaver" value-ref="shopSlaverDataSource"/>                
        <entry key="shop_master" value-ref="shopMasterDataSource"/>            
    </map>        
    </property>        
    <!-- 默认丛库 -->        
    <property name="defaultTargetDataSource" ref="designSlaverDataSource"/>    
</bean>    
<!-- session factory -->    
<bean id="designSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">        
    <property name="dataSource" ref="designDataSource"/>        
    <property name="configLocation" value="classpath:mybatis-config.xml"/>    
</bean>
...

data_mapping.xml 配置
需要注意的是，此处class指向的是自己创建的MapperFactoryBeanInvocation动态代理

<bean id="accountLogMapper" class="com.design.datasource.factory.MapperFactoryBeanInvocation">
    <property name="mapperInterface" value="com.design.shop.dao.AccountLogMapper"/>    
    <property name="sqlSessionFactory" ref="designSessionFactory"/>
</bean>
<bean id="imageMapper" class="com.design.datasource.factory.MapperFactoryBeanInvocation">    
    <property name="mapperInterface" value="com.design.shop.dao.MallWechatImageTextTempMapper"/>       
    <property name="sqlSessionFactory" ref="designSessionFactory"/>
</bean>

Mybatis Mapper

public interface AccountLogMapper {
...
   @SelectProvider(type=AccountLogSqlProvider.class, method="selectByExample")
   @Results({    
       @Result(column="log_id", property="logId", jdbcType=JdbcType.INTEGER, id=true), 
       @Result(column="user_id", property="userId", jdbcType=JdbcType.INTEGER),
       @Result(column="user_money", property="userMoney", jdbcType=JdbcType.DECIMAL),
       @Result(column="frozen_money", property="frozenMoney", jdbcType=JdbcType.DECIMAL), 
       @Result(column="rank_points", property="rankPoints", jdbcType=JdbcType.INTEGER), 
       @Result(column="pay_points", property="payPoints", jdbcType=JdbcType.INTEGER), 
       @Result(column="change_time", property="changeTime", jdbcType=JdbcType.INTEGER), 
       @Result(column="change_desc", property="changeDesc", jdbcType=JdbcType.VARCHAR), 
       @Result(column="change_type", property="changeType", jdbcType=JdbcType.TINYINT)
  })
   //因为默认为一库主从，因此这个Mapper没有使用@OtherSource注解，而@Read则指向从库，对应dataSource.xml 中的 salver
   @Read
   List<AccountLog> selectByExample(AccountLogExample example);
...
}

public interface MallWechatImageTextTempMapper {
...
    @SelectProvider(type=MallWechatImageTextTempSqlProvider.class, method="countByExample")
    //此处@OtherSource("shop")指向了shop_mall数据库  @Read 指向从库  所以对应 dataSource.xml 中的 shop_slaver
    @Read
    @OtherSource("shop")
    long countByExample(MallWechatImageTextTempExample example);
...
}

Spring Test
测试代码使用kotlin编写，基本和Java区别不是很大，函数式写起来更方便 _

@RunWith(SpringJUnit4ClassRunner::class)
@ContextConfiguration("classpath:appServer.xml")
class Test {    

    @Autowired    
    var mapper: AccountLogMapper? = null    
    @Autowired    
    var imageMapper: MallWechatImageTextTempMapper? = null  
  
    @Test    
    fun test() {        
        println("*********************************** test shop_mall mall_wechat_image_temp countByExample ***********************************************")        
        val count = imageMapper?.countByExample(MallWechatImageTextTempExample().apply {
            createCriteria().andTempSetCodeEqualTo("wechat_temp_1")            
            resultColumn = "tempid"            
            offset = 0            
            limit = 100}) ?: 0L        
        println(count)        
        assert(true)    
    }
   
    @Test    
    fun testMyselfDB() {        
        println("*********************************** test myself  design_shop account_log countByExample ***********************************************")        
        val list = mapper?.selectByExample(AccountLogExample()) ?: arrayListOf<AccountLog>()        
        if (list.size > 0) {            
            assert(true)        
        } else {            
            assert(false)        
        }    
    }
}

好了，所有和动态数据源相关的代码与示例已经完成了，我们现在来看下实际运行效果：

*********************************** test shop_mall  mall_wechat_image_temp countByExample ***********************************************
02:31:56 [main] INFO  com.mchange.v2.c3p0.impl.AbstractPoolBackedDataSource - Initializing c3p0 pool... com.mchange.v2.c3p0.ComboPooledDataSource [ acquireIncrement -> 10, acquireRetryAttempts -> 30, acquireRetryDelay -> 1000, autoCommitOnClose -> false, automaticTestTable -> null, breakAfterAcquireFailure -> false, checkoutTimeout -> 0, connectionCustomizerClassName -> null, connectionTesterClassName -> com.mchange.v2.c3p0.impl.DefaultConnectionTester, contextClassLoaderSource -> caller, dataSourceName -> 1hge1619lmh41qu1kjm50|366ac49b, debugUnreturnedConnectionStackTraces -> false, description -> null, driverClass -> com.mysql.jdbc.Driver, extensions -> {}, factoryClassLocation -> null, forceIgnoreUnresolvedTransactions -> false, forceSynchronousCheckins -> false, forceUseNamedDriverClass -> false, identityToken -> 1hge1619lmh41qu1kjm50|366ac49b, idleConnectionTestPeriod -> 120, initialPoolSize -> 8, jdbcUrl -> jdbc:mysql://139.224.40.142:3306/mall?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true, maxAdministrativeTaskTime -> 0, maxConnectionAge -> 0, maxIdleTime -> 60, maxIdleTimeExcessConnections -> 0, maxPoolSize -> 10, maxStatements -> 100, maxStatementsPerConnection -> 0, minPoolSize -> 5, numHelperThreads -> 3, preferredTestQuery -> select now(), privilegeSpawnedThreads -> false, properties -> {user=******, password=******}, propertyCycle -> 0, statementCacheNumDeferredCloseThreads -> 0, testConnectionOnCheckin -> false, testConnectionOnCheckout -> false, unreturnedConnectionTimeout -> 0, userOverrides -> {}, usesTraditionalReflectiveProxies -> false ]
02:31:56 [main] DEBUG com.design.shop.dao.MallWechatImageTextTempMapper.countByExample - ==>  Preparing: SELECT count(tempid) FROM comp_mall_wechat_imagetext_temp WHERE ((temp_set_code = ?)) LIMIT 0,100 
02:31:56 [main] DEBUG com.design.shop.dao.MallWechatImageTextTempMapper.countByExample - ==> Parameters: wechat_temp_1(String)
02:31:56 [main] DEBUG com.design.shop.dao.MallWechatImageTextTempMapper.countByExample - <==      Total: 1
4
*********************************** test myself  design_shop account_log countByExample ***********************************************
02:31:56 [main] INFO  com.mchange.v2.c3p0.impl.AbstractPoolBackedDataSource - Initializing c3p0 pool... com.mchange.v2.c3p0.ComboPooledDataSource [ acquireIncrement -> 10, acquireRetryAttempts -> 30, acquireRetryDelay -> 1000, autoCommitOnClose -> false, automaticTestTable -> null, breakAfterAcquireFailure -> false, checkoutTimeout -> 0, connectionCustomizerClassName -> null, connectionTesterClassName -> com.mchange.v2.c3p0.impl.DefaultConnectionTester, contextClassLoaderSource -> caller, dataSourceName -> 1hge1619lmh41qu1kjm50|5f0e9815, debugUnreturnedConnectionStackTraces -> false, description -> null, driverClass -> com.mysql.jdbc.Driver, extensions -> {}, factoryClassLocation -> null, forceIgnoreUnresolvedTransactions -> false, forceSynchronousCheckins -> false, forceUseNamedDriverClass -> false, identityToken -> 1hge1619lmh41qu1kjm50|5f0e9815, idleConnectionTestPeriod -> 120, initialPoolSize -> 8, jdbcUrl -> jdbc:mysql://127.0.0.1:3306/design_shop?useUnicode=true&characterEncoding=UTF-8&autoReconnect=true&zeroDateTimeBehavior=convertToNull&useSSL=true, maxAdministrativeTaskTime -> 0, maxConnectionAge -> 0, maxIdleTime -> 60, maxIdleTimeExcessConnections -> 0, maxPoolSize -> 10, maxStatements -> 100, maxStatementsPerConnection -> 0, minPoolSize -> 5, numHelperThreads -> 3, preferredTestQuery -> select now(), privilegeSpawnedThreads -> false, properties -> {user=******, password=******}, propertyCycle -> 0, statementCacheNumDeferredCloseThreads -> 0, testConnectionOnCheckin -> false, testConnectionOnCheckout -> false, unreturnedConnectionTimeout -> 0, userOverrides -> {}, usesTraditionalReflectiveProxies -> false ]
02:31:56 [main] DEBUG com.design.shop.dao.AccountLogMapper.selectByExample - ==>  Preparing: SELECT log_id, user_id, user_money, frozen_money, rank_points, pay_points, change_time, change_desc, change_type FROM des_account_log 
02:31:56 [main] DEBUG com.design.shop.dao.AccountLogMapper.selectByExample - ==> Parameters: 
02:31:56 [main] DEBUG com.design.shop.dao.AccountLogMapper.selectByExample - <==      Total: 35

2个不同的从库实现了动态切换，是不是有点暗爽？当然 Mybatis Mapper 是由 MybatisGenerator生成的，原版生成可不会带@Read @Write @OtherSource 等注解 也没有 offset、limit 分页属性，以上内容可以通过对MybatisGenerator做一些小改造，提供插件即可全部实现。在后续文章中会提及MybatisGenerator的插件制作，尽请期待 _

第一次发文，文笔还有没有介绍清楚的地方，还请海涵。

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 201,681评论 5赞 474
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 84,710评论 2赞 377
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 148,623评论 0赞 334
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 54,202评论 1赞 272
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 63,232评论 5赞 363
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 48,368评论 1赞 281
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 37,795评论 3赞 393
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 36,461评论 0赞 256
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 40,647评论 1赞 295
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,476评论 2赞 317
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,525评论 1赞 329
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 33,226评论 3赞 318
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 38,785评论 3赞 303
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,857评论 0赞 19
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,090评论 1赞 258
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 42,647评论 2赞 348
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 42,215评论 2赞 341

推荐阅读更多精彩内容

  • Spring Cloud
    Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具（例如配置管理，服务发现，断路器，智...
    卡卡罗2017阅读 134,570评论 18赞 139
  • Spring boot参考指南
    Spring Boot 参考指南 介绍 转载自:https://www.gitbook.com/book/qbgb...
    毛宇鹏阅读 46,714评论 6赞 342
  • Android - 收藏集
    Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,325评论 25赞 707
  • 轮播
    1. 轮播的实现原理是怎样的？如果让你来实现，你会抽象出哪些函数(or接口)供使用？（比如 play()） CSS...
    谨言_慎行阅读 208评论 1赞 0
  • 有些荒凉，总要自己去承受
    六月的最后一天，我终于承受不住长久以来的指责和非议，在房间里大声哭了出来…压抑了很久痛到不行的心情以及夜里辗转难眠...
    鸡蛋说阅读 308评论 0赞 0