mybatis 学习之多数据源整合

继续上篇文章如何使用mybatis3+spring3并且配置多数据源呢

先上代码在讲解吧
替换上一篇中spring中datasource的配置

Java代码

1. <!--数据源配置-->
2. <beanid="ds1"class="org.apache.commons.dbcp.BasicDataSource"
3. destroy-method="close">
4. <propertyname="driverClassName"value="com.mysql.jdbc.Driver"/>
5. <propertyname="url"
6. value="jdbc:mysql://localhost:3306/payoffdatabase?useUnicode=true&amp;characterEncoding=UTF-8"/>
7. <propertyname="username"value="root"/>
8. <propertyname="password"value="root"/>
9. <propertyname="defaultAutoCommit"value="true"></property>
10. </bean>
12. <beanid="ds2"class="org.apache.commons.dbcp.BasicDataSource"
13. destroy-method="close">
14. <propertyname="driverClassName"value="com.mysql.jdbc.Driver"/>
15. <propertyname="url"
16. value="jdbc:mysql://localhost:3306/test?useUnicode=true&amp;characterEncoding=UTF-8"/>
17. <propertyname="username"value="root"/>
18. <propertyname="password"value="root"/>
19. <propertyname="defaultAutoCommit"value="true"></property>
20. </bean>
23. <beanid="dataSource"class="a.b.router.DynamicDataSource">
24. <propertyname="targetDataSources">
25. <mapkey-type="java.lang.String">
26. <entryvalue-ref="ds1"key="ds1"></entry>
27. <entryvalue-ref="ds2"key="ds2"></entry>
28. </map>
29. </property>
30. <propertyname="defaultTargetDataSource"ref="ds1"></property>
31. </bean>

新的类DynamicDataSource

Java代码

1. packagea.b.router;
3. importorg.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
5. publicclassDynamicDataSourceextendsAbstractRoutingDataSource{
6. @Override
7. protectedObjectdetermineCurrentLookupKey(){
8. returnDataSourceContextHolder.getDbType();
9. }
10. }

新的类DataSourceContextHolder

Java代码

1. packagea.b.router;
3. publicclassDataSourceContextHolder{
5. privatestaticfinalThreadLocal<String>contextHolder=newThreadLocal<String>();
7. publicstaticvoidsetDbType(StringdbType){
8. contextHolder.set(dbType);
9. }
11. publicstaticStringgetDbType(){
12. return((String)contextHolder.get());
13. }
15. publicstaticvoidclearDbType(){
16. contextHolder.remove();
17. }
18. }

原来的单元测试新添加一行信息，修改为

Java代码

1. @Test
2. publicvoidaddTest()throwsException{
3. UserLoginuserLogin=newUserLogin();
4. userLogin.setPassword("102");
5. userLogin.setUsername("102");
6. DataSourceContextHolder.setDbType("ds2");
8. service.add(userLogin);
9. }

最主要的变化是DynamicDataSource 类，这个类继承了AbstractRoutingDataSource，我们再继续考察，发现这个类实现了datasource这个接口。

再仔细研究，发现我配置了多个数据源给DynamicDataSource，默认的数据源是ds1。
我们研究下AbstractRoutingDataSource类的代码，主要是两个实现datasource的方法

Java代码

1. publicConnectiongetConnection()throwsSQLException{
2. returndetermineTargetDataSource().getConnection();
3. }
5. publicConnectiongetConnection(Stringusername,Stringpassword)throwsSQLException{
6. returndetermineTargetDataSource().getConnection(username,password);
7. }

根据这段代码发现主要玄机都在determineTargetDataSource()方法上。

Java代码

1. protectedDataSourcedetermineTargetDataSource(){
2. Assert.notNull(this.resolvedDataSources,"DataSourcerouternotinitialized");
3. ObjectlookupKey=determineCurrentLookupKey();
4. DataSourcedataSource=this.resolvedDataSources.get(lookupKey);
5. if(dataSource==null&&(this.lenientFallback||lookupKey==null)){
6. dataSource=this.resolvedDefaultDataSource;
7. }
8. if(dataSource==null){
9. thrownewIllegalStateException("CannotdeterminetargetDataSourceforlookupkey["+lookupKey+"]");
10. }
11. returndataSource;
12. }

根据这段代码发现，首先在使用数据源之前，首先判断使用数据源的key，也就是我们配置给

Java代码

1. privateMap<Object,Object>targetDataSources;

这个map中的key值，找到key值之后再找到对应的datasource然后并使用这个数据源。


从上面我们发现，实际上DynamicDataSource只是在内部封装了数据源，然后调用它，只不过在内部他加了一些控制而已。（此处不知道是否可以理解为代理模式）

再深一步想想，此处使用的orm层是mybatis，如果换成hibernate呢，或者jdbctemplate呢。
实际上这个方法都适用。