在SSM项目中扩展配置多数据源
持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第2天，点击查看活动详情
一个ssm项目中需要扩展多个数据源，原来只有一个mysql，现在需要再扩展一个mysql，现将需要改动的地方做一个记录。
db.properties修改
首先，需要在数据源配置文件中增加新数据源配置。因为都是mysql，除了需要改动的，其他配置都是用了一样的参数，如果需要更改，可以自行增加配置参数。
# 数据源1
jdbc.user=root
jdbc.password=root
jdbc.jdbcUrl=jdbc:mysql://localhost:3306/db1?characterEncoding=utf-8&serverTimezone=UTC&useSSL=false
# 数据源2
jdbc2.jdbcUrl=jdbc:mysql://localhost:3306/db1?characterEncoding=utf-8&serverTimezone=UTC&useSSL=false
jdbc2.user=root
jdbc2.password=root
# 公用配置
jdbc.driverClass=com.mysql.jdbc.Driver
jdbc.initPoolSize=5
jdbc.maxPoolSize=1024spring.xml修改
spring.xml中需要新增数据源配置，以及多源数据库如何选择的配置。
<!-- 导入资源文件 -->
<context:property-placeholder location="classpath:db.properties"/>
<!-- db1数据源 -->
<bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource">
<property name="user" value="${jdbc.user}"></property>
<property name="password" value="${jdbc.password}"></property>
<property name="driverClass" value="${jdbc.driverClass}"></property>
<property name="jdbcUrl" value="${jdbc.jdbcUrl}"></property>
<property name="initialPoolSize" value="${jdbc.initPoolSize}"></property>
<property name="maxPoolSize" value="${jdbc.maxPoolSize}"></property>
</bean>
<!-- db2数据库-->
<bean id="dataSource2" class="com.mchange.v2.c3p0.ComboPooledDataSource">
<property name="user" value="${jdbc2.user}"></property>
<property name="password" value="${jdbc2.password}"></property>
<property name="driverClass" value="${jdbc.driverClass}"></property>
<property name="jdbcUrl" value="${jdbc2.jdbcUrl}"></property>
<property name="initialPoolSize" value="${jdbc.initPoolSize}"></property>
<property name="maxPoolSize" value="${jdbc.maxPoolSize}"></property>
</bean>
<!-- 动态选择数据源，指向com.demo.test.dataSource.DynamicDataSource，后文会贴出代码 -->
<bean id="dynamicDataSource" class="com.demo.test.dataSource.DynamicDataSource">
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry key="defaultDB" value-ref="dataSource"/>
<entry key="DataDB" value-ref="dataSource2"/>
</map>
</property>
<!--默认数据源-->
<property name="defaultTargetDataSource" ref="dataSource2"/>
</bean>
<!-- 配置 SessionFactory -->
<bean id="sessionFactory"
class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
<!-- 更改指向 dynamicDataSource -->
<property name="dataSource" ref="dynamicDataSource"></property>
<property name="configLocation" value="classpath:hibernate.cfg.xml"></property>
</bean>增加数据源切换类
数据源枚举类
public enum DataSourceEnum {
DS1("defaultDB"), DS2("DataDB");
private String key;
DataSourceEnum(String key) {
this.key = key;
}
public String getKey() {
return key;
}
public void setKey(String key) {
this.key = key;
}
}线程持有数据源标识类
用于标记线程使用哪个数据源标识
public class DataSourceHolder {
private static final ThreadLocal<String> dataSources = new ThreadLocal<String>();
public static void setDataSources(String dataSource) {
dataSources.set(dataSource);
}
public static String getDataSources() {
return dataSources.get();
}
}配置动态数据源类
AbstractRoutingDataSource 类：可以根据用户定义的规则选择当前的数据源，多源数据库需要使用该类实现多源数据库选择。
在每次数据库查询操作前执行，determineCurrentLookupKey() 决定使用哪个数据源。
实现的具体逻辑：继承 AbstractRoutingDataSource 类并重写 determineCurrentLookupKey 方法。把配置的多个数据源会放在AbstractRoutingDataSource的 targetDataSources和defaultTargetDataSource中，然后通过afterPropertiesSet()方法将数据源分别进行复制到resolvedDataSources和resolvedDefaultDataSource中。调用AbstractRoutingDataSource的getConnection()的方法的时候，先调用determineTargetDataSource()方法返回DataSource在进行getConnection()。
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceHolder.getDataSources();
}
}切换数据源代码
配置完上诉代码，到此可以在查询数据源时增加一个选择数据源语句实现选择数据源。例如：
public User test01(String username) {
DataSourceHolder.setDataSources(DataSourceEnum.DS1.getKey());
return userDao.findUserByName(username);
}
public User test02(String username) {
DataSourceHolder.setDataSources(DataSourceEnum.DS2.getKey());
return userDao.findUserByName(username);
}自动切换数据源
手动切换数据源需要每个方法都需要进行修改，太过于麻烦。
这里可以利用aop，实现根据包名自动切换数据源。
切面类
db1数据源都在db1包名下，db2数据源都在db2数据源下。如果你的项目文件结构不适用，可以根据能区分开数据源的策略进行自动切换，这个策略需要你自己去想了👻。
public class DataSourceExchange {
public void before(JoinPoint point) {
//获取目标对象的类类型
Class<?> aClass = point.getTarget().getClass();
String c = aClass.getName();
String[] ss = c.split("\.");
//获取包名，进行数据源选择。这里根据需求进行修改。
String packageName = ss[3];
if ("db2".equals(packageName)) {
DataSourceHolder.setDataSources(DataSourceEnum.DS2.getKey());
// System.out.println("数据源：" + DataSourceEnum.DS2.getKey());
} else {
DataSourceHolder.setDataSources(DataSourceEnum.DS1.getKey());
// System.out.println("数据源：" + DataSourceEnum.DS1.getKey());
}
}
/**
* 执行后将数据源置为空
*/
public void after() {
DataSourceHolder.setDataSources(null);
}
}
在配置中设置切面
bean id为bean的名称，class指向类的位置；aop的execution表示筛选的类，method表示对类使用的方法。
<!-- DataSourceExchange 注入bean -->
<bean id="dataSourceExchange" class="com.demo.test.dataSource.DataSourceExchange"/>
<!-- 配置aop切面 -->
<aop:config>
<aop:aspect ref="dataSourceExchange">
<!-- 限定在哪些包下执行 -->
<aop:pointcut id="dataSourcePointcut" expression="execution(* com.demo.test..service.impl.*ServiceImpl.*(..))"/>
<aop:before pointcut-ref="dataSourcePointcut" method="before"/>
<aop:after pointcut-ref="dataSourcePointcut" method="after"/>
</aop:aspect>
</aop:config>execution语法说明：
比如：* com.demo.test..service.impl.*ServiceImpl.*(..))
第一个 * 表示任意返回值
.. 表示 test 包和它的所有子包
*ServiceImpl 表示任何前缀的ServiceImpl类
最后的*表示，命中类的所有方法
总结
修改配置到这里就已经完成了ssm配置多数据源修改了，如果还出现了报错，请查看错误日志，很可能是配置的参数中指向错误，细心点检查下。
最后的运行结果：
测试完，记得把输出注释掉哦