基于AbstractRoutingDataSource实现动态切换数据源
/**
* DataSource注解接口
*/
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface DataSourceAnnotation {
String value() default DataSourceSelect.MASTER;
}
/**
* 注解前后增强获取数据库清空数据库
*/
@Aspect
@Component
@Logger
public class DatasourceAspect {
@Pointcut("@annotation(com.example.demo.config.DataSourceAnnotation)")
public void pointcut() {
}
@Before("pointcut()")
public void beforeExecute(JoinPoint joinPoint) {
Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
DataSourceAnnotation annotation = method.getAnnotation(DataSourceAnnotation.class);
if (annotation == null) {
annotation = joinPoint.getTarget().getClass().getAnnotation(DataSourceAnnotation.class);
DynamicDataSource.setDataSource(DataSourceSelect.MASTER);
}else{
// 切换数据源
DynamicDataSource.setDataSource (annotation.value());
}
}
@After("pointcut()")
public void afterExecute() {
DynamicDataSource.clear();
}
}
定义数据源类来存储多个数据源选择
public interface DataSourceSelect {
/**
* 主库源
*/
String MASTER="master";
/**
* 从库源
*/
String SLAVE ="slave";
}
继承AbstractRoutingDataSource重写方法实现把默认数据源和目标数据源注入
并提供ThreadLocal线程保证数据源隔离。
public class DynamicDataSource extends AbstractRoutingDataSource {
private final Logger logger = LoggerFactory.getLogger(DynamicDataSource.class);
private static final ThreadLocal<String> DATA_SOURCE_CONTEXT_HOLDER = new ThreadLocal<>();
public DynamicDataSource(DataSource defaultTargetDataSource, Map<Object, Object> targetDataSources) {
super.setDefaultTargetDataSource(defaultTargetDataSource);
super.setTargetDataSources(targetDataSources);
super.afterPropertiesSet();
}
@Override
protected Object determineCurrentLookupKey() {
return getDataSource();
}
static void setDataSource(String dataSource) {
DATA_SOURCE_CONTEXT_HOLDER.set(dataSource);
}
public static String getDataSource() {
return DATA_SOURCE_CONTEXT_HOLDER.get();
}
static void clear() {
DATA_SOURCE_CONTEXT_HOLDER.remove();
}
}
把数据源配置路径与数据源匹配,并利用子类调用有参构造调用父类AbstractRoutingDataSource的方法把数据源信息加载进datasource
/**
* 配置多数据源
* @author xiaohe
* @version V1.0.0
*/
@Configuration
public class DynamicDataSourceConfig {
@Bean
@ConfigurationProperties("spring.datasource.master")
public DataSource masterDataSource(){
return DruidDataSourceBuilder.create().build();
}
@Bean
@ConfigurationProperties("spring.datasource.slave")
public DataSource slaveDataSource(){
return DruidDataSourceBuilder.create().build();
}
@Bean
@Primary
public DynamicDataSource dataSource(DataSource masterDataSource, DataSource slaveDataSource) {
Map<Object, Object> targetDataSources = new HashMap<>(5);
targetDataSources.put(DataSourceSelect.MASTER, masterDataSource);
targetDataSources.put(DataSourceSelect.SLAVE, slaveDataSource);
return new DynamicDataSource(masterDataSource, targetDataSources);
}
}
Application启动项,exclude = DataSourceAutoConfiguration.class排除自动注入数据源的配置,加上Import 用来加加载我们定义的datasource。
@ComponentScan(basePackages ="com.example.demo.*")
@EnableJpaRepositories("com.example.demo.repository")
@Import(DynamicDataSourceConfig.class)
@SpringBootApplication(exclude = DataSourceAutoConfiguration.class)
public class DruidApplication {
public static void main(String[] args) {
SpringApplication.run(DruidApplication.class, args);
}
}
FQA:
9. 注解没有被注入到ioc容器导致注解失效。
10. 多数据源的动态切换,在程序运行时,把数据源数据源动态织入到程序中,灵活的进行数据源切换。
- 基于多数据源的动态切换,我们可以实现读写分离,这么做缺点也很明显,无法动态的增加数据源。
只支持单库事务,也就是说切换数据源要在开启事务之前执行。
spring DataSourceTransactionManager进行事务管理,开启事务,会将数据源缓存到DataSourceTransactionObject对象中进行后续的commit rollback等事务操作。