update 优化多数据源aop顺序问题

This commit is contained in:
疯狂的狮子li 2021-06-22 13:43:05 +08:00
parent 1caf76f64b
commit 40bba563d6
2 changed files with 22 additions and 14 deletions

View File

@ -1,19 +1,25 @@
package com.ruoyi.common.enums; package com.ruoyi.common.enums;
import lombok.AllArgsConstructor;
import lombok.Getter;
/** /**
* 数据源 * 数据源
* *
* @author ruoyi * @author Lion Li
*/ */
public enum DataSourceType @AllArgsConstructor
{ public enum DataSourceType {
/** /**
* 主库 * 主库
*/ */
MASTER, MASTER("master"),
/** /**
* 从库 * 从库
*/ */
SLAVE SLAVE("slave");
@Getter
private final String source;
} }

View File

@ -20,7 +20,7 @@ import java.util.Objects;
* @author ruoyi * @author ruoyi
*/ */
@Aspect @Aspect
@Order(1) @Order(-500)
@Component @Component
public class DataSourceAspect { public class DataSourceAspect {
@ -34,7 +34,9 @@ public class DataSourceAspect {
DataSource dataSource = getDataSource(point); DataSource dataSource = getDataSource(point);
if (Validator.isNotNull(dataSource)) { if (Validator.isNotNull(dataSource)) {
DynamicDataSourceContextHolder.push(dataSource.value().name()); DynamicDataSourceContextHolder.poll();
String source = dataSource.value().getSource();
DynamicDataSourceContextHolder.push(source);
} }
try { try {