從零開始學 Java - Spring AOP 實現主從讀寫分離

深刻討論為什么要讀寫分離？

為了服務器承載更多的用戶？提升了網站的響應速度？分攤數據庫服務器的壓力？就是為了雙機熱備又不想浪費備份服務器？上面這些回答，我認為都不是錯誤的，但也都不是完全正確的。「讀寫分離」並不是多么神奇的東西，也帶不來多么大的性能提升，也許更多的作用的就是數據安全的備份吧。

從一個庫到讀寫分離，從理論上對服務器壓力來說是會帶來一倍的性能提升，但你仔細思考一下，你的應用服務器真的很需要這一倍的提升么？那倒不如你去試着在服務器使用一下緩存系統，如 Memcached、Redis 這些分布式緩存，那性能可能是幾十倍的提升。而且，在服務器硬件異常強悍及性能廉價的今天，完全更沒必要了，所以，在今天，我認為它更多的職責就是為了數據安全而設計的，同時又提升了一些性能，這樣也挺好。

可能我們更應該稱之為主從分離。

利用 AOP 實現讀寫分離

讀寫分離方式很簡單，就是在你讀數據是去連接從庫，在你寫數據的時候去連接主庫，具體代碼實現當然就是連接時候去操作了，這沒什么難度，在代碼里寫就是了。可是，有追求的程序猿都是不是這么解決問題的呢！
其實通過上篇的 從零開始學 Java - Spring AOP 攔截器的基本實現 我們知道 AOP 可以實現在方法開始執行前后插入執行我們想要的代碼，那這樣，我們是不是可以在執行數據庫操作前根據業務來動態切換數據源呢？
思考一下這個方式理論上好像是可行的，這種方式首先不需要在業務代碼中去做切換，二是可能以后我們不需要讀寫分離了，把 AOP 切換的代碼去掉就行了，三是可能就是拓展性好了。

等不了了，開始擼代碼

你可能想深入的了解的話，我這里給你幾個程序里用到的關鍵字enum（枚舉）、annotation（自定義注解）、JoinPoint（注入點）、AbstractRoutingDataSource（數據源接口子類），你理解了這些就知道了，其實你並不需要深入某些深層的東西，了解下即可。

**一、建立JdbcContextHolder.java類

public class JdbcContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    public static void setJdbcType(String jdbcType) {
        contextHolder.set(jdbcType);
    }

    public static void setSlave() {
        setJdbcType("slave");
    }

    public static void setMaster() {
        clearJdbcType();
    }

    public static String getJdbcType() {
        return (String) contextHolder.get();
    }

    public static void clearJdbcType() {
        contextHolder.remove();
    }
}

這個類的作用就是用來設置、獲取數據源連接

二、新建DynamicDataSource.java類，繼承於AbstractRoutingDataSource

import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;

import cn.mayongfa.common.JdbcContextHolder;

public class DynamicDataSource extends AbstractRoutingDataSource {

    @Override
    protected Object determineCurrentLookupKey() {
        // 獲取當前數據源連接
        return JdbcContextHolder.getJdbcType();
    }
}

通過研究，我們知道determineCurrentLookupKey方法是獲取相關數據源連接的，所以重寫determineCurrentLookupKey方法就可以啦，然后我們去通過剛剛我們建立的JdbcContextHolder類去獲取。那怎么設置呢？

三、建立數據源DataSourceType.java枚舉類

public enum DataSourceType {

    //主庫
    Master("master"),

    //從庫
    Slave("slave");

    private DataSourceType(String name) {
        this.name = name;
    }

    private String name;

    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
}

這個枚舉類的作用其實就是為了設置數據源而生的，它的目的就是讓設置數據源時更方便，如絲般順滑。

**四、新建DataSource.javaAnnotation（自定義注解）類

import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)  
@Target(ElementType.METHOD)
@Documented
public @interface DataSource {  

    DataSourceType value() default DataSourceType.Master;

}  

自定義注解的意義不再過多討論，一句話來說就是可以讓你在類或方法名上以打標簽的形式讓該方法變得不一樣。具體怎么「不一樣」，這個在於你。

五、新建DataSourceChoose.java數據庫切換類

import java.lang.reflect.Method;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.reflect.MethodSignature;

import cn.mayongfa.common.JdbcContextHolder;

public class DataSourceChoose {

//方法執行前
public void before(JoinPoint point){
    Object target = point.getTarget();  
        String method = point.getSignature().getName();  
        Class<?>[] classz = target.getClass().getInterfaces();  
        MethodSignature methodSignature = (MethodSignature)point.getSignature();
        Class<?>[] parameterTypes = methodSignature.getMethod().getParameterTypes();
        try {
            Method m = classz[0].getMethod(method, parameterTypes);  
            if (m!=null && m.isAnnotationPresent(DataSource.class)) {  
                DataSource data = m.getAnnotation(DataSource.class);  
                JdbcContextHolder.clearJdbcType();
                JdbcContextHolder.setJdbcType(data.value().getName());
            }  
        } catch (Exception e) {  
            // TODO: handle exception  
        } 
}
}

這個其實是一個攔截器類，主要作用就是攔截那些方法名上有@DataSource這個自定義注解的，完了根據獲取注解的value()值，來做相應的數據源切換。

到這里，整個讀寫分離的分析及業務邏輯和具體代碼都完了，代碼都可以訪問我的 Github 看到 https://github.com/mafly/SpringDemo 。接下來就是配置文件配置多個數據源、攔截器，這部分就相對固定以及沒難度，放在下一篇文章中：

從零開始學 Java - Spring 一主多從、多主多從 數據庫配置