说明
通过上篇博文《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<>();