Spring多数据源（数据源路由）AbstractRoutingDatasource 配置

蔚落 2021-12-16 04:53 461阅读 0赞

**多数据源的配置，数据源路由（AbstractRoutingDatasource）**

1.定义一个数据源路由MyRoutingDataSource，继承AbstractRoutingDatasource类。

① 重写父类的setTargetDataSources（）方法，该方法用于设置多个数据源实例，并用一个Map类型变量（该变量在父类中）存储数据源实例。

②数据源切换的key，必须保证线程独有，使用ThreadLocal变量存放该key，并提供get，set方法。

③实现父类抽象方法determineCurrentLookupKey（），在该方法中自定义数据源的切换逻辑。

public class MyRoutingDataSource extends AbstractRoutingDataSource{
        //数据源切换的key，必须保证线程独有，使用ThreadLocal变量存放该key
    	private  ThreadLocal<String> key = new ThreadLocal<>();
    	
    	@Override
        //实现父类抽象方法，定义数据源切换逻辑
    	protected Object determineCurrentLookupKey() {
    		if("dataSource1".equals(key.get())) {
    			return "dataSource1";
    		}else{
    			return "dataSource2";
    		}
    	}
    	
    	@Override
    	//该方法用于设置多个数据源实例
    	public void setTargetDataSources(Map<Object, Object> targetDataSources) {
    		super.setTargetDataSources(targetDataSources);
    	}
    
    	public ThreadLocal<String> getKey() {
    		return this.key;
    	}
    
    	public void setKey(ThreadLocal<String> key) {
    		this.key = key;
    	}
    }

2.spring配置文件  
先配置多个数据源，再在数据路由中引用这些数据源。注意：需要配置默认数据源，避免未选择数据源情况下导致无数据源使用。

<bean id="dataSource1" class="com.alibaba.druid.pool.DruidDataSource">
       		<property name="driverClassName" value="${jdbc.driver}"/>
       		<property name="url" value="${jdbc.url}"/>
       		<property name="username" value="${jdbc.username}"/>
       		<property name="password" value="${jdbc.password}"/>
       		<property name="maxActive" value="10"/>
       		<property name="initialSize" value="5"/>
       	</bean>
       	<bean id="dataSource2" class="com.alibaba.druid.pool.DruidDataSource">
       		<property name="driverClassName" value="${user_driver}"/>
       		<property name="url" value="${user_url}"/>
       		<property name="username" value="${user_username}"/>
       		<property name="password" value="${user_password}"/>
       		<property name="maxActive" value="10"/>
       		<property name="initialSize" value="5"/>
       	</bean>
       	
       	
       	<bean id="myRoutingDataSource" class="com.fly.yosomall.ws.MyRoutingDataSource">
       		<property name="targetDataSources">
       			<map key-type="java.lang.String">
       				<entry key="dataSource1" value-ref="dataSource1"></entry>
       				<entry key="dataSource2" value-ref="dataSource2"></entry>
       			</map>
       		</property>
       		
       		<property name="defaultTargetDataSource" ref="dataSource2"></property>
       	</bean>

3.使用：在调用业务方法前设置key的值。为什在调用业务方法前呢？  
**因为一旦调用业务方法，事务就开启了。我们知道事务管理器中也需要一个数据源实例，所以数据源路由必须在事务开启前确定好使用哪个具体数据源。**