springboot学习(五)：MyBatis动态数据源配置

说明

通过上篇博文《MyBatis多数据源配置》，学习了MyBatis多数据源的配置。在实际环境中，数据库一般配置为主从结构的形式，甚至是一主多从的形式，这时，我们希望读操作都在从数据库中进行，增删改操作都在主数据库中进行，希望mybatis能够动态地选择数据库。通过本篇博文，学习记录下通过注解+AOP和只使用AOP两种方式实现动态数据源配置。文中使用主机的mysql作为主数据库，虚机中的mysql作为从数据库。

正文

1.通过spirngboot构建项目

主要依赖 mybatis+mysql+aspect

<dependency>
    <groupId>org.mybatis.spring.boot</groupId>
    <artifactId>mybatis-spring-boot-starter</artifactId>
    <version>1.3.2</version>
</dependency>
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <scope>runtime</scope>
</dependency>
<dependency>
    <groupId>org.aspectj</groupId>
    <artifactId>aspectjweaver</artifactId>
    <version>1.8.7</version>
</dependency>

2.在application.properties配置数据库

#主库
datasource.master.driverClassName=com.mysql.jdbc.Driver
datasource.master.url=jdbc:mysql://ip:3306/dbname?characterEncoding=UTF-8
datasource.master.username=**
datasource.master.password=**

#从库
datasource.slave.driverClassName=com.mysql.jdbc.Driver
datasource.slave.url=jdbc:mysql://ip:3306/dbname?characterEncoding=UTF-8
datasource.slave.username=**
datasource.slave.password=**

3.在启动类关闭自动配置数据源

这个很重要，如果没有关闭，在启动程序时会发生循环依赖的错误

@EnableAutoConfiguration(exclude = {DataSourceAutoConfiguration.class})
@SpringBootApplication
public class DynamicDatasourceAnnoApplication {
   

    public static void main(String[] args) {
        SpringApplication.run(DynamicDatasourceAnnoApplication.class, args);
    }
}

4.配置数据源

要实现数据源的动态选择，AbstractRoutingDataSource这个抽象类很重要，通过阅读源码，发现数据源在该类中以key-value的形式存储在map中，其中在determineTargetDataSource()方法中，通过determineCurrentLookupKey()方法得到key，在map中寻找需要的数据源，而determineCurrentLookupKey()是抽象方法，我们可以通过继承这个抽象类实现该方法，配置我们期望的数据源。

public abstract class AbstractRoutingDataSource extends AbstractDataSource implements InitializingBean {
   

protected DataSource determineTargetDataSource() {
        Assert.notNull(this.resolvedDataSources, "DataSource router not initialized");
        Object lookupKey = this.determineCurrentLookupKey();
        DataSource dataSource = (DataSource)this.resolvedDataSources.get(lookupKey);
        if (dataSource == null && (this.lenientFallback || lookupKey == null)) {
            dataSource = this.resolvedDefaultDataSource;
        }

        if (dataSource == null) {
            throw new IllegalStateException("Cannot determine target DataSource for lookup key [" + lookupKey + "]");
        } else {
            return dataSource;
        }
}

protected abstract Object determineCurrentLookupKey();

在了解了基本原理后，开始配置数据源，继承AbstractRoutingDataSource 类

DynamicDataSource

public class DynamicDataSource extends AbstractRoutingDataSource {
   
    @Override
    protected Object determineCurrentLookupKey() {
        return DynamicDataSourceHolder.getType();
    }
}

实现了该方法后，如何返回一个key，这个key从何而来？
每个请求都是一个线程，我们可以再创建一个类，其中使用ThreadLocal线程局部变量保存每个线程所请求方法对应的key，通过该类的静态方法返回该请求的key

DynamicDataSourceHolder

public class DynamicDataSourceHolder {
    private static final ThreadLocal<DataSourceType> dataSourceHolder = new ThreadLocal<>();

    private static final Set<DataSourceType> dataSourceTypes = new HashSet<>();