项目需要从其他网站获取数据,因为是临时加的需求,在开始项目时没想到需要多数据源
于是百度了一下,发现只需要改动一下Spring 的applicationContext.xml文件和编写三个工具类就可以完美实现
applicationContext.xml
<!-- 多数据源配置 -->
<bean id="ds1" class="org.apache.commons.dbcp.BasicDataSource">
<property name="driverClassName" value="${jdbc.driverClassName}" />
<property name="url" value="${jdbc.url}" />
<property name="username" value="${jdbc.username}" />
<property name="password" value="" />
</bean>
<bean id="ds2" class="org.apache.commons.dbcp.BasicDataSource">
<property name="driverClassName" value="" />
<property name="url" value="" />
<property name="username" value="" />
<property name="password" value="" />
</bean>
<!-- 动态配置数据源 -->
<bean id="dataSource" class="com.test.utils.DynamicDataSource">//这里是你项目里DynamicDataSource.java的路径
<property name="targetDataSources">
<map key-type="java.lang.String">
<entry value-ref="ds_admin" key="ds1"></entry>
<entry value-ref="ds_partner" key="ds2"></entry>
</map>
</property>
<!-- 默认使用ds1的数据源 -->
<property name="defaultTargetDataSource" ref="ds_admin"></property>
</bean>
DataSourceContextHolder.java
public class DataSourceContextHolder {
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
public static void setDbType(String dbType) {
contextHolder.set(dbType);
}
public static String getDbType() {
return ((String) contextHolder.get());
}
public static void clearDbType() {
contextHolder.remove();
}
}
DataSourceType.java(设置静态变量)
public class DataSourceType {
// 默认数据库
public static final String SOURCE_ADMIN = "ds1";
// 第二个数据库,在applicationContext.xml里的id
public static final String SOURCE_PARTNER = "ds2";
}
接下来这个是关键DynamicDataSource.java 它继承了AbstractRoutingDataSource中的抽象方法determineCurrentLookupKey是实现数据源的route的核心.这里对该方法进行Override。
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
public class DynamicDataSource extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return DataSourceContextHolder.getDbType();
}
}
本人也是新手,哪里写的不对希望大神指点