spring 多資料來源配置實現

在網上找到的配置多資料來源的方法。

1.擴充 org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource類

實現代碼

 

代碼

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

public class DynamicDataSource extends AbstractRoutingDataSource {

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

}

 

 

2.實現一個安全執行緒的DbContextHolder類。

 

代碼

public class DbContextHolder {
        private static final ThreadLocal contextHolder = new ThreadLocal();    
        
        public static void setDbType(String dbType) 
        {    
            contextHolder.set(dbType);   
        }    
        
        
        /**
         * 取得dbtype類型
         * @return
         */
        public static String getDbType() 
        {    
            String str=(String) contextHolder.get();
            if(null==str || "".equals(str))
                str="1";
            return str;   
        }    
        
        public static void clearDbType() 
        {    
            contextHolder.remove();   
        }  
}

 

 

3.修改spring的設定檔。

 

代碼

<!--動態選擇資料來源-->  
    <bean id="dataSource" class="tecamo.core.utils.DynamicDataSource">  
        <property name="targetDataSources">  
            <map key-type="java.lang.String">  
                <entry key="1" value-ref="ds1"/>  
                <entry key="2" value-ref="ds2"/>
            </map>  
        </property>  
        <property name="defaultTargetDataSource" ref="ds1"/>  
    </bean>  
    
    
    
    
    
    <!-- 資料來源定義 -->
    <bean id="ds1" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
        <property name="driverClassName" value="${jdbc.driverClassName}"/>
        <property name="url" value="${jdbc.url}"/>
        <property name="username" value="${jdbc.username}"/>
        <property name="password" value="${jdbc.password}"/>
    </bean>
    
    <bean id="ds2" class="org.apache.commons.dbcp.BasicDataSource" destroy-method="close">
        <property name="driverClassName" value="${jdbcOrcale.driverClassName}"/>
        <property name="url" value="${jdbcOrcale.url}"/>
        <property name="username" value="${jdbcOrcale.username}"/>
        <property name="password" value="${jdbcOrcale.password}"/>
    </bean>

 

 

 

4.切換資料來源方法。

在serverice方法之前調用

DbContextHolder.setDbType("2");表示使用第二個資料來源。