AOP切面實現動態資料來源(含程式碼)
阿新 • • 發佈:2019-02-05
在service同級目錄增加一個folder ,將以下三個官方提供的類添至其中:
1 DatasourceAspect.java
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.JoinPoint;
/**
* 定義資料來源的AOP切面,通過該Service的方法名判斷是應該走讀庫還是寫庫
*
*/
public class DataSourceAspect {
/**
* 在進入Service方法之前執行
*
* @param point 切面物件
*/
public void before(JoinPoint point) {
// 獲取到當前執行的方法名
String methodName = point.getSignature().getName();
if (isSlave(methodName)) {
// 標記為讀庫
DynamicDataSourceHolder.markSlave();
} else {
// 標記為寫庫
DynamicDataSourceHolder.markMaster();
}
}
/**
* 判斷是否為讀庫
*
* @param methodName
* @return
*/
private Boolean isSlave(String methodName) {
// 方法名以query、find、get開頭的方法名走從庫
return StringUtils.startsWithAny(methodName, "query", "find", "get");
}
}
2 DynamicDataSource.java
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
* 定義動態資料來源,實現通過整合Spring提供的AbstractRoutingDataSource,只需要實現determineCurrentLookupKey方法即可
*
* 由於DynamicDataSource是單例的,執行緒不安全的,所以採用ThreadLocal保證執行緒安全,由DynamicDataSourceHolder完成。
*
*/
public class DynamicDataSource extends AbstractRoutingDataSource{
@Override
protected Object determineCurrentLookupKey() {
// 使用DynamicDataSourceHolder保證執行緒安全,並且得到當前執行緒中的資料來源key
return DynamicDataSourceHolder.getDataSourceKey();
}
}
3 DynamicDataSourceHolder.java
public class DynamicDataSourceHolder {
//寫庫對應的資料來源key
private static final String MASTER = "master";
//讀庫對應的資料來源key
private static final String SLAVE = "slave";
//使用ThreadLocal記錄當前執行緒的資料來源key
private static final ThreadLocal<String> holder = new ThreadLocal<String>();
/**
* 設定資料來源key
* @param key
*/
public static void putDataSourceKey(String key) {
holder.set(key);
}
/**
* 獲取資料來源key
* @return
*/
public static String getDataSourceKey() {
return holder.get();
}
/**
* 標記寫庫
*/
public static void markMaster(){
putDataSourceKey(MASTER);
}
/**
* 標記讀庫
*/
public static void markSlave(){
putDataSourceKey(SLAVE);
}
}
修改 jdbc.properties
####################### DB Connection Config #######################
###----------------- DB Type -----------------
#the database of the application:mysql|sqlserver|oracle
databaseType=mysql
#databaseType=sqlserver
#databaseType=oracle
###----------------- MySQL5+ -----------------
jdbc.master.driverClassName=com.mysql.jdbc.Driver
jdbc.master.url=jdbc:mysql://192.168.137.141:3306/demo?useUnicode=true&zeroDateTimeBehavior=convertToNull&characterEncoding=utf-8
jdbc.master.username=root
jdbc.master.password=root
jdbc.slave.driverClassName=com.mysql.jdbc.Driver
jdbc.slave.url=jdbc:mysql://192.168.137.128:3308/demo?useUnicode=true&zeroDateTimeBehavior=convertToNull&characterEncoding=utf-8
jdbc.slave.username=root
jdbc.slave.password=root
hibernate.dialect=org.hibernate.dialect.MySQLDialect
###----------------- SqlServer2005+ -----------------
#jdbc.driverClassName=com.microsoft.sqlserver.jdbc.SQLServerDriver
#jdbc.url=jdbc:sqlserver://localhost:1433;DatabaseName=sampledb
#jdbc.username=xxx
#jdbc.password=xxx
#hibernate.dialect=org.hibernate.dialect.SQLServerDialect
###----------------- Oracle10g+ -----------------
#jdbc.driverClassName=oracle.jdbc.driver.OracleDriver
#jdbc.url=jdbc:oracle:thin:@localhost:1521:orac10g
#jdbc.username=scott
#jdbc.password=scott123
#hibernate.dialect=org.hibernate.dialect.OracleDialect
###----------------- JNDI -----------------
#jndi.name=myjndi123
###----------------- Hibernate -----------------
hibernate.show_sql=true
hibernate.format_sql=true
hibernate.hbm2ddl.auto=update
hibernate.jdbc.fetch_size=100
hibernate.jdbc.batch_size=20
hibernate.cache.use_second_level_cache=true
hibernate.cache.use_query_cache=true
hibernate.memcached.cacheTimeSeconds=10800
#Hibernate4
hibernate.cache.region.factory_class=org.hibernate.cache.EhCacheRegionFactory
#Hibernate3
#hibernate.cache.region.factory_class=org.hibernate.cache.EhCacheProvider
###----------------- C3P0 -----------------
c3p0.maxPoolSize=300
c3p0.minPoolSize=1
c3p0.initialPoolSize=1
c3p0.maxIdleTime=60
c3p0.acquireIncrement=5
c3p0.idleConnectionTestPeriod=60
###----------------- DBCP -----------------
dbcp.maxActive=50
dbcp.maxIdle=50
dbcp.minIdle=1
dbcp.maxWait=10000
dbcp.initialSize=1