Java实现动态增加和切换数据源以访问不同的数据库

        有时候我们需要把数据存放到多个数据库中，但是一个数据源只能访问一个数据库。想访问不同的数据库，那么就需要切换不同的数据源。有时候我们要切换的数据源是未知的，在程序运行的过程中才能知道要访问哪一个数据库，这时候就需要使用动态增加数据源的方法。我们可以先在配置文件中配置一个默认数据源，程序运行过程中需要访问其它数据库的时候，就动态的创建新的数据源织入到程序当中，让程序使用该新建的数据源。将这些新建的数据源缓存起来，后面需要用到就可以获取到，实现切换不同的数据源。主要的步骤为：“首先在配置文件中配置一个默认数据源”、“新建DynamicDataSource类来实现切换不同的数据源”、“DynamicDataSource实现动态增加数据源”、“使用Spring的AOP，将线程ThreadLocal的dbName变量清空，防止影响下一次访问数据库”、“在我们的产品中，不同的公司对应着不同的数据库dbName，所以可以根据dbName创建不同的数据源”。

1、    首先在配置文件中配置一个默认数据源。
在ApplicationContext.xml中配置数据源。
(1)    配置sqlSessionFactory，指定数据源为dynamicDataSource：

<bean id="sqlSessionFactory" class="org.mybatis.spring.SqlSessionFactoryBean"><property name="dataSource" ref="dynamicDataSource" /><property name="configLocation" value="classpath:mybatis-config.xml"></property></bean>

(2)    配置dynamicDataSource，为自定义的DynamicDataSource类，负责切换数据源和动态增加数据源，指定了默认数据源为jdbcDataSource_nbr_bx：

<!--动态数据源的配置 --><bean id="dynamicDataSource" class="com.bx.erp.action.interceptor.DynamicDataSource" primary="true"><property name="targetDataSources"><map key-type="java.lang.String"><entry value-ref="jdbcDataSource_nbr_bx" key="jdbcDataSource_nbr_bx" /></map></property><property name="defaultTargetDataSource" ref="jdbcDataSource_nbr_bx" /></bean>

(3)    配置默认数据源jdbcDataSource_nbr_bx：

<bean id="jdbcDataSource_nbr_bx" class="org.springframework.jdbc.datasource.DriverManagerDataSource"><property name="driverClassName"><value>${driverClassName}</value></property><property name="url"><value>jdbc:mysql://localhost:3306/nbr_bx?useUnicode=true&amp;serverTimezone=Asia/Shanghai&amp;characterEncoding=utf8&amp;zeroDateTimeBehavior=CONVERT_TO_NULL</value></property><property name="username"><value>${db.nbx.mysql.username}</value></property><property name="password"><value>${db.nbx.mysql.password}</value></property></bean>

2、    新建DynamicDataSource类来实现切换不同的数据源。
Spring框架有一个AbstractRoutingDataSource类，它可以在程序运行的时候，把某个数据源动态织入到程序中，以便访问不同的数据库。它有一个determineCurrentLookupKey方法，该方法返回一个键key，通过这个key，在数据源集合中获取其中一个数据源。最后根据获取到的数据源去访问数据库。所以DynamicDataSource类可以继承AbstractRoutingDataSource，实现determineCurrentLookupKey方法，通过返回不同的key实现切换不同的数据源，默认使用配置文件中的数据源。

public class DynamicDataSource extends AbstractRoutingDataSource {@Overrideprotected Object determineCurrentLookupKey() {String dbName = DataSourceContextHolder.getDbName();if (dbName == null) {dbName = DataSourceContextHolder.DATASOURCE_jdbcDataSource_nbr_bx; // 默认使用公共DB：nbr_bx} else {this.selectDataSource(dbName);if (dbName.equals(DataSourceContextHolder.DATASOURCE_jdbcDataSource_nbr_bx)) {dbName = DataSourceContextHolder.DATASOURCE_jdbcDataSource_nbr_bx;}}logger.debug("--------> using datasource " + dbName);return dbName;}
……

3、    DynamicDataSource实现动态增加数据源。
我们在配置文件中只配置了一个默认数据源，那么需要增加数据源，才能够切换不同的数据源访问不同的数据库。
增加一个数据源的步骤如下：
例如我们要访问一个名称为nbr_test的数据库，这个访问操作对应着某一个线程，线程要访问哪一个数据库，我们可以使用ThreadLocal类来设置和获取。ThreadLocal类可以防止一个线程的变量被其它线程修改。
我们用DataSourceContextHolder封装ThreadLocal：

public class DataSourceContextHolder{public static final String DATASOURCE_jdbcDataSource_nbr_bx = "jdbcDataSource_nbr_bx";private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();public static void setDbName(String dbName) {contextHolder.set(dbName);}public static String getDbName() {return ((String) contextHolder.get());}public static void clearDbName() {contextHolder.remove();}}

在访问数据库前，我们设置访问nbr_test这个数据库：

DataSourceContextHolder.setDbName(dbName);

DynamicDataSource的determineCurrentLookupKey方法获取到当前线程的数据库名称dbName：

String dbName = DataSourceContextHolder.getDbName();

DynamicDataSource的selectDataSource方法从 targetDataSources这个hashmap中查找是否存在这个数据源，因为现在还没有这个数据源，所以需要去创建，后面创建好了之后，也会把创建好的数据源存到这个targetDataSources中：

/** @describe 数据源存在时不做处理，不存在时创建新的数据源链接，并将新数据链接添加至缓存 */public void selectDataSource(String dbName) {String sid = DataSourceContextHolder.getDbName();if (DataSourceContextHolder.DATASOURCE_jdbcDataSource_nbr_bx.equals(dbName)) {DataSourceContextHolder.setDbName(DataSourceContextHolder.DATASOURCE_jdbcDataSource_nbr_bx);return;}Object obj = this._targetDataSources.get(dbName);if (obj != null && sid.equals(dbName)) {return;} else {BasicDataSource dataSource = this.getDataSource(dbName);if (null != dataSource)this.setDataSource(dbName, dataSource);}}

     创建一个数据源：

public BasicDataSource getDataSource(String serverId) {return (BasicDataSource) createDataSource(serverId);}/** 该方法为同步方法，防止并发创建两个相同的数据库 使用双检锁的方式，防止并发 */private synchronized DataSource createDataSource(String dbName) {String url = String.format(DB_MYSQL_URL, dbName);//BasicDataSource dataSource = new BasicDataSource();dataSource.setDriverClassName(DRIVER_CLASS_NAME);dataSource.setUrl(url);dataSource.setUsername(DB_MYSQL_USERNAME);dataSource.setPassword(DB_MYSQL_PASSWORD);dataSource.setTestWhileIdle(true);dataSource.setValidationQuery("select 1");dataSource.setTestOnBorrow(true);return dataSource;}

       数据库连接信息放在了一个配置文件中，通过注解@Value来获取，访问不同的的数据库只是dbName这个字段不同，所以可以使用String.format(DB_MYSQL_URL, dbName)来修改连接url,访问不同的数据库。

       注解：

@Value("${driverClassName}")private String DRIVER_CLASS_NAME;@Value("${db.mysql.url}")private String DB_MYSQL_URL;@Value("${db.mysql.username}")private String DB_MYSQL_USERNAME;@Value("${db.mysql.password}")private String DB_MYSQL_PASSWORD;

配置文件：

# 数据库驱动
driverClassName=com.mysql.cj.jdbc.Driver
# 数据库URL
db.mysql.url=jdbc:mysql://localhost:3306/%s?useUnicode=true&serverTimezone=Asia/Shanghai&characterEncoding=utf8&zeroDateTimeBehavior=CONVERT_TO_NULL
# 数据库用户名
db.mysql.username.encryption=81A3FD464E18C4497A79CE7CC9D5B660
db.nbx.mysql.username.encryption=81A3FD464E18C4497A79CE7CC9D5B660
# 密码
db.mysql.password.encryption=16CCEF25E22FA42E89821B7B27858DE26DD8BFF1139FF2C70BECCA88E373F809
db.nbx.mysql.password.encryption=16CCEF25E22FA42E89821B7B27858DE26DD8BFF1139FF2C70BECCA88E373F809

     创建好一个数据源后，把它放到集合缓存中，下次就不用重新创建了，同时告诉父类AbstractRoutingDataSource，要使用当前这个数据源访问数据库：

BasicDataSource dataSource = this.getDataSource(dbName);if (null != dataSource)this.setDataSource(dbName, dataSource);public void setDataSource(String serverId, BasicDataSource dataSource) {this.addTargetDataSource(serverId, dataSource);DataSourceContextHolder.setDbName(serverId);}public void addTargetDataSource(String key, BasicDataSource dataSource) {this._targetDataSources.put(key, dataSource);this.setTargetDataSources(this._targetDataSources);}public void setTargetDataSources(Map<Object, Object> targetDataSources) {this._targetDataSources = targetDataSources;super.setTargetDataSources(this._targetDataSources);afterPropertiesSet();}

4、    使用Spring的AOP，将线程ThreadLocal的dbName变量清空，防止影响下一次访问数据库。

@Aspect // for aop
@Component // for auto scan
@Order(0) // 在事务前执行
public class DataSourceInterceptor {@Pointcut("execution(public * com.bx.erp.action.bo.*.*(..))")public void dataSourceSlave() {};@Before("dataSourceSlave()")public void before(JoinPoint jp) {
//		System.out.println("进入切面");}@After("dataSourceSlave()")public void removeDataSoruce(JoinPoint joinPoint) throws Throwable {DataSourceContextHolder.clearDbName();}

5、    在我们的产品中，不同的公司对应着不同的数据库dbName，所以可以根据dbName创建不同的数据源。

Company company = getCompanyFromSession(session);String dbName = company.getDbName();DataSourceContextHolder.setDbName(dbName);