（七）spring+druid多數據源配置

druid多數據源配置

一、druid簡介

　　Druid首先是一個數據庫連接池，但它不僅僅是一個數據庫連接池，它還包含一個ProxyDriver，一系列內置的JDBC組件庫，一個SQL Parser。

　　Druid是目前最好的數據庫連接池，在功能、性能、擴展性方面，都超過其他數據庫連接池，包括DBCP、C3P0、BoneCP、Proxool、JBoss DataSource。Druid已經在阿里巴巴部署了超過600個應用，經過一年多生產環境大規模部署的嚴苛考驗。

　　Druid 是目前比較流行的高性能的，分布式列存儲的OLAP框架(具體來說是MOLAP)。它有如下幾個特點：

　　一. 亞秒級查詢

    　　 druid提供了快速的聚合能力以及亞秒級的OLAP查詢能力，多租戶的設計，是面向用戶分析應用的理想方式。

　　二.實時數據注入

     　　druid支持流數據的注入，並提供了數據的事件驅動，保證在實時和離線環境下事件的實效性和統一性

　　三.可擴展的PB級存儲

    　　 druid集群可以很方便的擴容到PB的數據量，每秒百萬級別的數據注入。即便在加大數據規模的情況下，也能保證時其效性

　　四.多環境部署

     　　druid既可以運行在商業的硬件上，也可以運行在雲上。它可以從多種數據系統中注入數據，包括hadoop，spark，kafka，storm和samza等

　　五.豐富的社區

二、druid多數據源使用

　　1.直接配置

　　配置兩個 dataSource，兩個sqlSessionFactory，兩個transactionManager，以及關鍵的地方在於MapperScannerConfigurer 的配置——使用sqlSessionFactoryBeanName屬性，注入不同的sqlSessionFactory的名稱，這樣就為不同的數據庫對應的 mapper 接口注入了對應的 sql於master-slave類型的多數據源配置而言不太適應，不支持分布式事務

　　2.基於AbstractRoutingDataSource和AOP的多數據源配置

　　我們自己定義一個DataSource類ThreadLocalRountingDataSource，來繼承AbstractRoutingDataSource，然后在配置文件中向ThreadLocalRountingDataSource注入 master 和 slave 的數據源，然后通過 AOP 來靈活配置。

　　applicationContext.xml

<?xml version="1.0" encoding="UTF-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xmlns:aop="http://www.springframework.org/schema/aop"
    xmlns:context="http://www.springframework.org/schema/context"
    xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
        http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop-4.3.xsd
        http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context-4.3.xsd>
    
    <!-- 啟動spring注解 -->
    <context:annotation-config/>
    <!-- 掃描注解所在的包 -->
    <context:component-scan base-package="com.example"/>
    <!-- 啟動aop注解 -->
    <aop:aspectj-autoproxy proxy-target-class="true"/> 
       
       <!-- 引入屬性文件 -->
    <context:property-placeholder location="classpath:jdbc.properties"/>
  <!--   <bean id="dataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
        <property name="driverClassName" value="${driver}"/>
        <property name="url" value="${url}"/>
    </bean> 
     -->
     <!-- 配置數據源master -->
     <bean id="dataSourceMaster" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">   
        <property name="driverClassName" value="${driver}"/>
        <property name="url" value="${url}" />  
        <!-- 初始化連接大小 -->
        <property name="initialSize" value="0" />  
       <!-- 連接池最大使用連接數量 -->
        <property name="maxActive" value="20" />  
        <!-- 連接池最小空閑 -->
        <property name="minIdle" value="1" />  
        <!-- 連接池最大空閑 -->
        <property name="maxIdle" value="20" />  
        <!-- 獲取連接最大等待時間 -->
        <property name="maxWait" value="60000" />  
    </bean>  
    <!-- 配置數據源master -->
     <bean id="dataSourceSlave" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">   
        <property name="driverClassName" value="${driver}"/>
        <property name="url" value="${url_slave}" />  
        <!-- 初始化連接大小 -->
        <property name="initialSize" value="0" />  
       <!-- 連接池最大使用連接數量 -->
        <property name="maxActive" value="20" />  
        <!-- 連接池最小空閑 -->
        <property name="minIdle" value="0" />  
        <!-- 連接池最大空閑 -->
        <property name="maxIdle" value="20" />  
        <!-- 獲取連接最大等待時間 -->
        <property name="maxWait" value="60000" />  
    </bean>  
     <bean id="dataSource" class="com.example.util.ThreadLocalRountingDataSource">
         <property name="targetDataSources">
             <map key-type="com.example.enums.DataSources">
                 <entry key="MASTER" value-ref="dataSourceMaster" />
                 <entry key="SLAVE" value-ref="dataSourceSlave"/>
             </map>
         </property>
         <property name="defaultTargetDataSource" ref="dataSourceMaster"></property>
        
    </bean> 
    
    <!-- 配置SQLSessionFactory -->
    <bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
        <property name="dataSource" ref="dataSource"/>  
        <property name="configLocation" value="classpath:mybatis-config.xml"></property> 
        <!-- 加載映射文件 -->
        <property name="mapperLocations" value="classpath*:/com/example/dao/*Mapper.xml"></property>  
    </bean>
    
    <!-- 接口方式 -->
    <bean class="org.mybatis.spring.mapper.MapperScannerConfigurer">
        <property name="basePackage" value="com.example.dao"></property>
        <property name="sqlSessionFactoryBeanName" value="sqlSessionFactory"></property>
    </bean> 
  
</beans>

　　1.定義一個DataSource枚舉類

//定義一個enum來表示不同的數據源
public enum DataSources {
    MASTER,SLAVE
}

　　2.ThreadLocalRountingDataSource，繼承AbstractRoutingDataSource

public class ThreadLocalRountingDataSource extends AbstractRoutingDataSource{

    @Override
    protected Object determineCurrentLookupKey() {
        // TODO Auto-generated method stub
        return DataSourceTypeManager.get();
    }

}

　　3.DataSourceTypeManager

//通過 TheadLocal 來保存每個線程選擇哪個數據源的標志(key)

public class DataSourceTypeManager {
    
    private static final ThreadLocal<DataSources> dataSourceTypes=new ThreadLocal<>();
    
    public static DataSources get(){
        return dataSourceTypes.get();
    }
    
    public static void set(DataSources dataSourceType){
        dataSourceTypes.set(dataSourceType);
    }
    
    public static void reset(){
        dataSourceTypes.set(DataSources.MASTER);
    }
}

　　4.aop管理

@Aspect
@Component
public class DataSourceInterceptor {

    @Pointcut("execution(public * com.example.service..*.selectByPrimaryKey(..))")
    public void dataSourceMaster(){
        
    };
    
    @Before("dataSourceMaster()")
    public void before(JoinPoint jp){
        DataSourceTypeManager.set(DataSources.MASTER);
    }
    //...
    /*這里我們定義了一個 Aspect 類，我們使用 @Before 來在符合
     *  @Pointcut("execution(public * net.aazj.service..*.selectByPrimaryKey(..))") 中的方法被調用之前，
     *  調用 DataSourceTypeManager.set(DataSources.SLAVE) 設置了 key 的類型為 DataSources.MASTER，
     *  所以 dataSource 會根據key=DataSources.MASTER 選擇 dataSourceSlave 這個dataSource。
     *  所以該方法對於的sql語句會在slave數據庫上執行.
     *  我們可以不斷的擴充 DataSourceInterceptor這個 Aspect，在中進行各種各樣的定義，
     *  來為某個service的某個方法指定合適的數據源對應的dataSource。
     *  這樣我們就可以使用 Spring AOP 的強大功能來，十分靈活進行配置了。
     */
}

　　當調用selectByPrimaryKey方法的時候會進入切面類中切換數據源，方法調用完畢會把數據源切換回來

三、AbstractRoutingDataSource原理

源碼：

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean
void afterPropertiesSet() throws Exception; 
 
    @Override
    public void afterPropertiesSet() {
        if (this.targetDataSources == null) {
            throw new IllegalArgumentException("Property 'targetDataSources' is required");
        }
        this.resolvedDataSources = new HashMap<Object, DataSource>(this.targetDataSources.size());
        for (Map.Entry<Object, Object> entry : this.targetDataSources.entrySet()) {
            Object lookupKey = resolveSpecifiedLookupKey(entry.getKey());
            DataSource dataSource = resolveSpecifiedDataSource(entry.getValue());
            this.resolvedDataSources.put(lookupKey, dataSource);
        }
        if (this.defaultTargetDataSource != null) {
            this.resolvedDefaultDataSource = resolveSpecifiedDataSource(this.defaultTargetDataSource);
        }
    }

　　targetDataSources 是我們在xml配置文件中注入的 dataSourceMaster 和 dataSourceSlave. afterPropertiesSet方法就是使用注入的dataSourceMaster 和 dataSourceSlave來構造一個HashMap——resolvedDataSources。方便后面根據 key 從該map 中取得對應的dataSource。

protected DataSource determineTargetDataSource() {
    Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
    Object lookupKey = determineCurrentLookupKey();
    DataSource dataSource = this.resolvedDataSources.get(lookupKey);
    if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
        dataSource = this.resolvedDefaultDataSource;
    }
    if (dataSource == null) {
        throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
    }
    return dataSource;
}

　　Object lookupKey = determineCurrentLookupKey(); 該方法是我們實現的，在其中獲取ThreadLocal中保存的 key 值。（見上述步驟2）獲得了key之后，在從afterPropertiesSet()（InitializingBean中實現的方法）中初始化好了的resolvedDataSources這個map中獲得key對應的dataSource。而ThreadLocal中保存 key 值。