说到多数据源,使用的场景一般有:1,主从数据库切换;2,多租户间数据逻辑隔离;
本文后继的例子均以SSM架构为基础讲解,如你的框架不适合,请参考思路即可。
最关键的是org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource类。
---------------------------------------------------------------------------------------------------------------------
针对1的方案,结合github上相关的项目,简要指出关键点。
在Spring的配置文件中,配置主从数据源对象,同时配置动态数据管理对象。
<!-- 主库数据源 -->
<bean id="masterDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init"
destroy-method="close">
<!-- 基本属性 url、user、password -->
<property name="driverClassName" value="${master.jdbc.driver}"/>
<property name="url" value="${master.jdbc.url}"/>
<property name="username" value="${master.jdbc.username}"/>
<property name="password" value="${master.jdbc.password}"/>
<!-- 配置初始化大小、最小、最大 -->
<property name="initialSize" value="1"/>
<property name="minIdle" value="1"/>
<property name="maxActive" value="20"/>
<!-- 配置获取连接等待超时的时间 -->
<property name="maxWait" value="60000"/>
<!-- 配置间隔多久才进行一次检测,检测需要关闭的空闲连接,单位是毫秒 -->
<property name="timeBetweenEvictionRunsMillis" value="60000"/>
<!-- 配置一个连接在池中最小生存的时间,单位是毫秒 -->
<property name="minEvictableIdleTimeMillis" value="300000"/>
<!-- 校验语句 -->
<property name="validationQuery" value="SELECT 1"/>
<property name="testWhileIdle" value="true"/>
<property name="testOnBorrow" value="false"/>
<property name="testOnReturn" value="false"/>
<!-- 配置监控统计拦截的filters -->
<property name="filters" value="stat"/>
</bean>
<!-- 从库数据源 -->
<bean id="slaveDataSource" class="com.alibaba.druid.pool.DruidDataSource" init-method="init" destroy-method="close">
...
</bean>
<!-- 动态数据源 -->
<bean id="dataSource" class="person.wzg.modules.common.db.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<!-- 可配置多个数据源 -->
<entry value-ref="masterDataSource" key="masterDataSource"></entry>
<entry value-ref="slaveDataSource" key="slaveDataSource"></entry>
</map>
</property>
<property name="defaultTargetDataSource" ref="masterDataSource"></property>
</bean>
<!-- 为Mybatis创建SqlSessionFactory,同时指定数据源 -->
<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean">
<property name="dataSource" ref="dataSource"/>
<property name="configLocation" value="classpath:mybatis-config.xml"/>
<property name="mapperLocations" value="classpath*:person/wzg/modules/dao/mapper/*Mapper.xml"/>
</bean>
动态数据源管理类person.wzg.modules.common.db.DynamicDataSource的内容如下:
package person.wzg.modules.common.db;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
/**
* 动态数据源(数据源切换)
*/
public class DynamicDataSource extends AbstractRoutingDataSource {
private final static Logger _log = LoggerFactory.getLogger(DynamicDataSource.class);
private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();
@Override
protected Object determineCurrentLookupKey() {
String dataSource = getDataSource();
_log.info("当前操作使用的数据源:{}", dataSource);
return dataSource;
}
/**
* 设置数据源
*
* @param dataSource
*/
public static void setDataSource(String dataSource) {
contextHolder.set(dataSource);
}
/**
* 获取数据源
*
* @return
*/
public static String getDataSource() {
String dataSource = contextHolder.get();
// 如果没有指定数据源,使用默认数据源
if (null == dataSource) {
DynamicDataSource.setDataSource(DataSourceEnum.MASTER.getDefault());
}
return contextHolder.get();
}
/**
* 清除数据源
*/
public static void clearDataSource() {
contextHolder.remove();
}
}
package person.wzg.modules.common.db;
/**
* 多数据源枚举
*/
public enum DataSourceEnum {
// 主库
MASTER("masterDataSource", true),
// 从库
SLAVE("slaveDataSource", false),;
// 数据源名称
private String name;
// 是否是默认数据源
private boolean master;
DataSourceEnum(String name, boolean master) {
this.name = name;
this.master = master;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public boolean isMaster() {
return master;
}
public void setMaster(boolean master) {
this.master = master;
}
public String getDefault() {
String defaultDataSource = "";
for (DataSourceEnum dataSourceEnum : DataSourceEnum.values()) {
if (!"".equals(defaultDataSource)) {
break;
}
if (dataSourceEnum.master) {
defaultDataSource = dataSourceEnum.getName();
}
}
return defaultDataSource;
}
}
因为应用场景是主从数据库读写分离,使用的时候,如果想使用从数据库,一般在dao层指定数据源。手动指定如,DynamicDataSource.setDataSource("slaveDataSource");或使用注解或DAO层方法名特征使用不同的数据源。