1. 程式人生 > >AOP切面實現動態資料來源(含程式碼)

AOP切面實現動態資料來源(含程式碼)

在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