要配置多个数据源,我们就需要写多个bean,每个bean对应一个数据源。难点在于,如何控制多个数据源之间的切换。这里我们需要借助ThreadLocal类,这个类位于java.lang包下,首先说明ThreadLocal存放的值是线程内共享的,线程间互斥的,主要用于线程内共享一些数据,避免通过参数来传递,这样处理后,能够优雅的解决一些实际问题
自定义DynamicDataSource类,用于切换数据源
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
public static final String DEFAULT_DATA_SOURCE = "default";
public static final String SILENCER_DATA_SOURCE = "silencer";
public static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
public static String getCustomerType() {
return contextHolder.get();
}
public static void setCustomerType(String customerType) {
contextHolder.set(customerType);
}
public static void clearCustomerType(String customerType) {
contextHolder.remove();
}
@Override
protected Object determineCurrentLookupKey() {
return getCustomerType();
}
}
配置多个bean
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource" />
<property name="configLocation" value="classpath:config/mybatis-config.xml" />
<property name="mapperLocations" value="classpath:mappers/**/*.xml" />
</bean>
<bean id="dataSource" class="com.silencer.web.utils.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="default" value-ref="defaultDataSource" />
<entry key="silencer" value-ref="silencerDataSource" />
</map>
</property>
<property name="defaultTargetDataSource" ref="defaultDataSource" />
</bean>
<bean id="defaultDataSource"
class="com.mchange.v2.c3p0.ComboPooledDataSource"
destroy-method="close">
<property name="driverClass" value="${default.jdbc.driver}" />
<property name="jdbcUrl" value="${default.jdbc.url}" />
<property name="user" value="${default.jdbc.username}" />
<property name="password" value="${default.jdbc.password}" />
<property name="initialPoolSize" value="${default.initialPoolSize}" />
<property name="maxPoolSize" value="${default.maxPoolSize}" />
<property name="minPoolSize" value="${default.minPoolSize}" />
<property name="maxIdleTime" value="${default.maxIdleTime}" />
<property name="checkoutTimeout" value="${default.checkoutTimeout}" />
</bean>
<bean id="silencerDataSource"
class="com.mchange.v2.c3p0.ComboPooledDataSource"
destroy-method="close">
<property name="driverClass" value="${silencer.jdbc.driver}" />
<property name="jdbcUrl" value="${silencer.jdbc.url}" />
<property name="user" value="${silencer.jdbc.username}" />
<property name="password" value="${silencer.jdbc.password}" />
<property name="initialPoolSize" value="${silencer.initialPoolSize}" />
<property name="maxPoolSize" value="${silencer.maxPoolSize}" />
<property name="minPoolSize" value="${silencer.minPoolSize}" />
<property name="maxIdleTime" value="${silencer.maxIdleTime}" />
<property name="checkoutTimeout" value="${silencer.checkoutTimeout}" />
</bean>
map中的key值和DynamicDataSourec类中定义的DEFAULT_DATA_SOURCE名称相同
每次请求都会连接默认数据源,其他数据源需要手动切换
DynamicDataSource.setCustomerType(DynamicDataSource.SILENCER_DATA_SOURCE);
也可以同过aop定义注解的方式来切换数据源