SSM框架下Mysql主从模式实现

灰太狼 2022-06-05 07:55 142阅读 0赞

继昨天的Mysql主从的概述及基本的配置。今天趁着不是很忙的时候整理一下主从JAVA端的代码实现。下面开始贴代码：

Spring MVC 的datasorce配置，这里我们使用的阿里的druid（德鲁伊）数据库连接池，先配置两个数据库链接池。分别连接主、从两个库。然后使用spring 的一个主要特性AOP切面编程来根据方法的前缀命名来选择是使用主数据库还是使用从数据库。（我选择切的是controller层，当用户访问方法的 时候，我就已经决定了是选择哪个数据源进行处理）

\[html\] [view plain][] [copy][view plain]

1.  **<?****xml** version="1.0" encoding="UTF-8"**?>**  
2.  **<****beans** xmlns="http://www.springframework.org/schema/beans"  
3.      xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns:tx="http://www.springframework.org/schema/tx"  
4.      xmlns:aop="http://www.springframework.org/schema/aop"  
5.      xsi:schemaLocation="  
6.  http://www.springframework.org/schema/beans   
7.  http://www.springframework.org/schema/beans/spring-beans-3.0.xsd   
8.  http://www.springframework.org/schema/tx   
9.  http://www.springframework.org/schema/tx/spring-tx-3.0.xsd  
10. http://www.springframework.org/schema/aop   
11. http://www.springframework.org/schema/aop/spring-aop-3.0.xsd  
12. "**>**  
13.       
14.       
15.       
16.     **<****bean** name="dataSourceMaster" class="com.alibaba.druid.pool.DruidDataSource"  
17.         init-method="init" destroy-method="close"**>**  
18.         **<****property** name="url" value="$\{master\_jdbc\_url\}" **/>**  
19.         **<****property** name="username" value="$\{master\_jdbc\_username\}" **/>**  
20.         **<****property** name="password" value="$\{master\_jdbc\_password\}" **/>**  
21.   
22.         **<****property** name="initialSize" value="$\{initialSize\}" **/>**  
23.         **<****property** name="maxActive" value="$\{maxActive\}" **/>**  
24.         **<****property** name="maxIdle" value="$\{maxIdle\}" **/>**  
25.         **<****property** name="minIdle" value="$\{minIdle\}" **/>**  
26.         **<****property** name="maxWait" value="$\{maxWait\}" **/>**  
27.   
28.         **<****property** name="validationQuery" value="$\{validationQuery\}" **/>**  
29.         **<****property** name="testOnBorrow" value="$\{testOnBorrow\}" **/>**  
30.         **<****property** name="testOnReturn" value="$\{testOnReturn\}" **/>**  
31.         **<****property** name="testWhileIdle" value="$\{testWhileIdle\}" **/>**  
32.           
33.         **<****property** name="timeBetweenEvictionRunsMillis" value="$\{timeBetweenEvictionRunsMillis\}" **/>**  
34.         **<****property** name="minEvictableIdleTimeMillis" value="$\{minEvictableIdleTimeMillis\}" **/>**  
35.           
36.         **<****property** name="removeAbandoned" value="$\{removeAbandoned\}" **/>**  
37.         **<****property** name="removeAbandonedTimeout" value="$\{removeAbandonedTimeout\}" **/>**  
38.         **<****property** name="logAbandoned" value="$\{logAbandoned\}" **/>**  
39.           
40.         **<****property** name="filters" value="stat" **/>**    
41.     **</****bean****>**  
42.   
43.       
44.     **<****bean** name="dataSourceSlave" class="com.alibaba.druid.pool.DruidDataSource"  
45.         init-method="init" destroy-method="close"**>**  
46.         **<****property** name="url" value="$\{slave\_jdbc\_url\}" **/>**  
47.         **<****property** name="username" value="$\{slave\_jdbc\_username\}" **/>**  
48.         **<****property** name="password" value="$\{slave\_jdbc\_password\}" **/>**  
49.   
50.         **<****property** name="initialSize" value="$\{initialSize\}" **/>**  
51.         **<****property** name="maxActive" value="$\{maxActive\}" **/>**  
52.         **<****property** name="maxIdle" value="$\{maxIdle\}" **/>**  
53.         **<****property** name="minIdle" value="$\{minIdle\}" **/>**  
54.         **<****property** name="maxWait" value="$\{maxWait\}" **/>**  
55.   
56.         **<****property** name="validationQuery" value="$\{validationQuery\}" **/>**  
57.         **<****property** name="testOnBorrow" value="$\{testOnBorrow\}" **/>**  
58.         **<****property** name="testOnReturn" value="$\{testOnReturn\}" **/>**  
59.         **<****property** name="testWhileIdle" value="$\{testWhileIdle\}" **/>**  
60.           
61.         **<****property** name="timeBetweenEvictionRunsMillis" value="$\{timeBetweenEvictionRunsMillis\}" **/>**  
62.         **<****property** name="minEvictableIdleTimeMillis" value="$\{minEvictableIdleTimeMillis\}" **/>**  
63.           
64.         **<****property** name="removeAbandoned" value="$\{removeAbandoned\}" **/>**  
65.         **<****property** name="removeAbandonedTimeout" value="$\{removeAbandonedTimeout\}" **/>**  
66.         **<****property** name="logAbandoned" value="$\{logAbandoned\}" **/>**  
67.           
68.         **<****property** name="filters" value="stat" **/>**    
69.   
70.     **</****bean****>**  
71.       
72.       
73.     **<****bean** id="dynamicDataSourceHolder" class="com.changba.idc.datasource.DynamicDataSourceHolder"**>**  
74.         **<****property** name="slavetDataSources"**>**  
75.             **<****map** key-type="java.lang.String"**>**  
76.                 **<****entry** key="slaveDataSource1" value-ref="dataSourceSlave"**>****</****entry****>**  
77.             **</****map****>**  
78.         **</****property****>**  
79.         **<****property** name="masterDataSources"**>**  
80.             **<****map** key-type="java.lang.String"**>**  
81.                 **<****entry** key="masterDataSource1" value-ref="dataSourceMaster"**>****</****entry****>**  
82.             **</****map****>**  
83.         **</****property****>**  
84.         **<****property** name="defaultTargetDataSource" ref="dataSourceMaster" **/>**  
85.     **</****bean****>**  
86.   
87.       
88.     **<****bean** id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"**>**  
89.         **<****property** name="dataSource" ref="dynamicDataSourceHolder" **/>**  
90.         **<****property** name="mapperLocations" value="classpath:com/changba/idc/mapper/imp/\*.xml" **/>**  
91.         **<****property** name="configLocation" value="classpath:mybatis-config.xml" **/>**   
92.     **</****bean****>**  
93.       
94.     **<****bean** class="org.mybatis.spring.mapper.MapperScannerConfigurer"**>**  
95.         **<****property** name="basePackage" value="com.changba.idc.mapper" **/>**  
96.         **<****property** name="sqlSessionFactoryBeanName" value="sqlSessionFactory" **/>**  
97.     **</****bean****>**  
98.       
99.       
100.     **<****bean** id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager"**>**  
101.         **<****property** name="dataSource" ref="dynamicDataSourceHolder" **/>**  
102.     **</****bean****>**  
103.   
104.       
105.     **<****tx:advice** id="transactionAdvice" transaction-manager="transactionManager"**>**  
106.         **<****tx:attributes****>**  
107.             **<****tx:method** name="add\*" propagation="REQUIRED" **/>**  
108.             **<****tx:method** name="append\*" propagation="REQUIRED" **/>**  
109.             **<****tx:method** name="insert\*" propagation="REQUIRED" **/>**  
110.             **<****tx:method** name="save\*" propagation="REQUIRED" **/>**  
111.             **<****tx:method** name="update\*" propagation="REQUIRED" **/>**  
112.             **<****tx:method** name="modify\*" propagation="REQUIRED" **/>**  
113.             **<****tx:method** name="edit\*" propagation="REQUIRED" **/>**  
114.             **<****tx:method** name="delete\*" propagation="REQUIRED" **/>**  
115.             **<****tx:method** name="remove\*" propagation="REQUIRED" **/>**  
116.             **<****tx:method** name="repair" propagation="REQUIRED" **/>**  
117.             **<****tx:method** name="delAndRepair" propagation="REQUIRED" **/>**  
118.   
119.             **<****tx:method** name="get\*" propagation="SUPPORTS" read-only="true" **/>**  
120.             **<****tx:method** name="find\*" propagation="SUPPORTS" read-only="true" **/>**  
121.             **<****tx:method** name="load\*" propagation="SUPPORTS" read-only="true" **/>**  
122.             **<****tx:method** name="search\*" propagation="SUPPORTS" read-only="true" **/>**  
123.             **<****tx:method** name="datagrid\*" propagation="SUPPORTS" read-only="true" **/>**  
124.   
125.             **<****tx:method** name="\*" propagation="SUPPORTS" **/>**  
126.         **</****tx:attributes****>**  
127.     **</****tx:advice****>**  
128.       
129.       
130.     **<****aop:config****>**  
131.         **<****aop:pointcut** id="transactionPointcut" expression="execution(\* com.changba.idc.service..\*.\*(..))" **/>**  
132.           
133.         **<****aop:advisor** pointcut-ref="transactionPointcut" advice-ref="transactionAdvice" **/>**  
134.     **</****aop:config****>**  
135.   
136.       
137.     **<****bean** id="dynamicDataSourceAop" class="com.changba.idc.datasource.DynamicDataSourceAop"**>****</****bean****>**  
138.       
139.       
140.     **<****aop:config** expose-proxy="true"**>**  
141.           
142.           
143.           
144.         **<****aop:aspect** id="dataSourceAspect" ref="dynamicDataSourceAop"**>**  
145.               
146.             **<****aop:pointcut** id="dataSourcePointcut" expression="execution(\* com.changba.idc.controller.\*.\*(..))" **/>**  
147.             **<****aop:around** method="doAroundMethod" pointcut-ref="dataSourcePointcut"**/>**  
148.         **</****aop:aspect****>**  
149.     **</****aop:config****>**  
150.       
151. **</****beans****>**

上边的配置是实现读写分离基本的配置，具体的JAVA代码实现如下：

\[java\] [view plain][] [copy][view plain]

1.  **package** com.changba.idc.datasource;  
2.    
3.  **import** org.aspectj.lang.ProceedingJoinPoint;  
4.  **import** org.slf4j.Logger;  
5.  **import** org.slf4j.LoggerFactory;  
6.  **import** org.springframework.beans.factory.annotation.Autowired;  
7.    
8.  /\*\* 
9.   \* 数据源的选取,在操作controller方法前先选取数据源 
10.  \*  
11.  \* @author An Dong 
12.  \* 
13.  \*/  
14. **public** **class** DynamicDataSourceAop \{  
15.     Logger logger = LoggerFactory.getLogger(DynamicDataSourceHolder.**class**);  
16.   
17.     @Autowired  
18.     **private** DynamicDataSourceHolder dataSourceHolder;  
19.   
20.     //AOP环绕切点  
21.     **public** Object doAroundMethod(ProceedingJoinPoint pjp) **throws** Throwable \{  
22.         Object response = **null**;  
23.         // method为方法名。并不是URL访问的名称  
24.         String method = pjp.getSignature().getName();  
25.         **boolean** hasBinded = **false**;  
26.         **try** \{  
27.             // hasBinded 是否已经绑定datasource  
28.             hasBinded = dataSourceHolder.hasBindedDataSourse();  
29.             logger.info("判断是否已经绑定数据源 \*\*\*\*\*\*\* "\+hasBinded+"   当前要执行的方法名称--------"\+method);  
30.             **if** (!hasBinded) \{  
31.                 **if** (method.startsWith("query") || method.startsWith("select") || method.startsWith("find")  
32.                         || method.startsWith("get") || method.startsWith("load") || method.startsWith("search")) \{  
33.                     dataSourceHolder.markSlave();  
34.                 \} **else** \{  
35.                     dataSourceHolder.markMaster();  
36.                 \}  
37.             \}  
38.             //调用proceed()方法作为环绕通知执行方法前后的分水岭  
39.             response = pjp.proceed();  
40.         \} **finally** \{  
41.             **if** (!hasBinded) \{  
42.                 logger.info("请求的方法执行完毕，清除已标记选取的数据源");  
43.                 dataSourceHolder.markRemove();  
44.             \}  
45.         \}  
46.         **return** response;  
47.     \}  
48.   
49. \}

其中dataSourceHolder 类中具体的方法，这里也是真正实现读写分离的主要逻辑处理的地方。

\[java\] [view plain][] [copy][view plain]

1.  **package** com.changba.idc.datasource;  
2.    
3.  **import** java.util.ArrayList;  
4.  **import** java.util.HashMap;  
5.  **import** java.util.List;  
6.  **import** java.util.Map;  
7.  **import** java.util.Map.Entry;  
8.  **import** java.util.Random;  
9.    
10. **import** javax.sql.DataSource;  
11. **import** org.slf4j.Logger;  
12. **import** org.slf4j.LoggerFactory;  
13. **import** org.springframework.beans.factory.InitializingBean;  
14. **import** org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;  
15. /\*\* 
16.  \* 向spring datasource 提供数据源选取key，对外提供从slave选取数据源，或从master选取数据源方法。 默认从master选取数据源 
17.  \* InitializingBean接口为bean提供了初始化方法的方式，它只包括afterPropertiesSet方法，凡是继承该接口的类，在初始化bean的时候会执行该方法。 
18.  \*  
19.  \* @author An Dong 
20.  \*  
21.  \*/  
22. **public** **class** DynamicDataSourceHolder **extends** AbstractRoutingDataSource **implements** InitializingBean\{  
23.     // 线程本地环境  
24.     **private** **static** **final** ThreadLocal<String> dataSourceHolder = **new** ThreadLocal<String>();  
25.     // 可选取slave的keys  
26.     **private** List<String> slaveDataSourcesKeys;  
27.     // 可选取master的keys  
28.     **private** List<String> masterDataSourcesKeys;  
29.     //从库数据源  
30.     **private** Map<String, DataSource> slavetDataSources;  
31.     //主库数据源  
32.     **private** Map<String, DataSource> masterDataSources;  
33.     **private** Logger logger = LoggerFactory.getLogger(DynamicDataSourceHolder.**class**);  
34.   
35.     @Override  
36.     //spring 执行数据源切换的核心方法。  
37.     **protected** Object determineCurrentLookupKey() \{  
38.         **return** dataSourceHolder.get();  
39.     \}  
40.       
41.     @Override  
42.     **public** **void** afterPropertiesSet() \{  
43.         // 数据检验和合并  
44.         logger.info("开始向Spring DataSource 提供数据源选取");  
45.         Map<Object, Object> allDataSources = **new** HashMap<Object, Object>();  
46.         allDataSources.putAll(masterDataSources);  
47.         **if** (slavetDataSources != **null**) \{  
48.             allDataSources.putAll(slavetDataSources);  
49.         \}  
50.         **super**.setTargetDataSources(allDataSources);  
51.         **super**.afterPropertiesSet();  
52.         logger.info("主库数据源：："\+masterDataSources.keySet());  
53.         logger.info("从库数据源：："\+slavetDataSources.keySet());  
54.         logger.info("已经完提供数据源选取");  
55.     \}  
56.     /\*\* 
57.      \* 注册slave datasource 
58.      \*  
59.      \* @param slavetDataSources 
60.      \*/  
61.     **public** **void** setSlavetDataSources(Map<String, DataSource> slavetDataSources) \{  
62.         **if** (slavetDataSources == **null** || slavetDataSources.size() == 0) \{  
63.             **return**;  
64.         \}  
65.         logger.info("提供可选取从库数据源：\{\}", slavetDataSources.keySet());  
66.         **this**.slavetDataSources = slavetDataSources;  
67.         slaveDataSourcesKeys = **new** ArrayList<String>();  
68.         **for** (Entry<String, DataSource> entry : slavetDataSources.entrySet()) \{  
69.             slaveDataSourcesKeys.add(entry.getKey());  
70.         \}  
71.     \}  
72.     /\*\* 
73.      \* 注册master datasource 
74.      \*  
75.      \* @param masterDataSources 
76.      \*/  
77.     **public** **void** setMasterDataSources(Map<String, DataSource> masterDataSources) \{  
78.         **if** (masterDataSources == **null**) \{  
79.             **throw** **new** IllegalArgumentException("Property 'masterDataSources' is required");  
80.         \}  
81.         logger.info("提供可选取主库数据源：\{\}", masterDataSources.keySet());  
82.         **this**.masterDataSources = masterDataSources;  
83.         **this**.masterDataSourcesKeys = **new** ArrayList<String>();  
84.         **for** (Entry<String, DataSource> entry : masterDataSources.entrySet()) \{  
85.             masterDataSourcesKeys.add(entry.getKey());  
86.         \}  
87.     \}  
88.   
89.     /\*\* 
90.      \* 标记选取从库数据源 
91.      \*/  
92.     **public** **void** markSlave() \{  
93.         **if** (dataSourceHolder.get() != **null**) \{  
94.             // 从现在的策略来看,不允许标记两次,直接抛异常,优于早发现问题  
95.             **throw** **new** IllegalArgumentException("当前已有选取数据源,不允许覆盖,已选数据源key:" + dataSourceHolder.get());  
96.         \}  
97.         logger.info("查询的是从库");  
98.         String dataSourceKey = selectFromSlave();  
99.         setDataSource(dataSourceKey);  
100.     \}  
101.   
102.     /\*\* 
103.      \* 标记选取主库数据源 
104.      \*/  
105.     **public** **void** markMaster() \{  
106.         **if** (dataSourceHolder.get() != **null**) \{  
107.             // 从现在的策略来看,不允许标记两次,直接抛异常,优于早发现问题  
108.             **throw** **new** IllegalArgumentException("当前已有选取数据源,不允许覆盖,已选数据源key:" + dataSourceHolder.get());  
109.         \}  
110.         logger.info("查询的是主库");  
111.         String dataSourceKey = selectFromMaster();  
112.         setDataSource(dataSourceKey);  
113.     \}  
114.   
115.     /\*\* 
116.      \* 删除己标记选取的数据源 
117.      \*/  
118.     **public** **void** markRemove() \{  
119.         dataSourceHolder.remove();  
120.     \}  
121.     /\*\* 
122.      \* 是否已经绑定datasource 
123.      \* 绑定：true 
124.      \* 没绑定：false 
125.      \* @return 
126.      \*/  
127.     **public** **boolean** hasBindedDataSourse()\{  
128.         **boolean** hasBinded= dataSourceHolder.get()!=**null**;  
129.         **return** hasBinded;  
130.     \}  
131.   
132.     **private** String selectFromSlave() \{  
133.         **if** (slavetDataSources == **null**) \{  
134.             logger.info("提供可选取slave数据源：\{\},将自动切换从主master选取数据源", slavetDataSources);  
135.             **return** selectFromMaster();  
136.         \} **else** \{  
137.             Random random = **new** Random();  
138.             **return** slaveDataSourcesKeys.get(random.nextInt(slaveDataSourcesKeys.size()));  
139.         \}  
140.     \}  
141.   
142.     **private** String selectFromMaster() \{  
143.         Random random = **new** Random();  
144.         String dataSourceKey = masterDataSourcesKeys.get(random.nextInt(masterDataSourcesKeys.size()));  
145.         **return** dataSourceKey;  
146.     \}  
147.   
148.     **private** **void** setDataSource(String dataSourceKey) \{  
149.         logger.info("数据源  key \*-\*-\*-\*-\*-\* "\+dataSourceKey);  
150.         dataSourceHolder.set(dataSourceKey);  
151.     \}  
152. \}

上边的两个JAVA类就不过多解释了。代码注释也比较多，能够看明白的。好了，主从的JAVA代码方面也已经介绍完了。有一个说的不是很准的地方，是这个主从的代码不仅仅应用于MySQL数据库的。其他数据库应该也是通用的，只是我没有尝试。有小伙伴们尝试后也可以给博主进行反馈呢。

转载地址：[http://blog.csdn.net/andong154564667/article/details/52116818][http_blog.csdn.net_andong154564667_article_details_52116818]

[view plain]: http://blog.csdn.net/andong154564667/article/details/52116818#
[http_blog.csdn.net_andong154564667_article_details_52116818]: http://blog.csdn.net/andong154564667/article/details/52116818