wangli/axzo-framework-commons
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat：(hotfix) controller处理完后，把threadLocal给清空掉

Browse Source
This commit is contained in:
李龙 2025-02-07 17:21:23 +08:00
parent 6a8b93548a
commit d199b7cbe8
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
axzo-common-data-permission/src/main/java/cn/axzo/framework/datapermission/aop/DataPermissionInterceptor.java
View File

@ -17,7 +17,10 @@ import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.annotation.AfterThrowing;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Lazy;
import org.springframework.core.annotation.Order;
@ -31,8 +34,6 @@ import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;
import static cn.axzo.framework.datapermission.context.DataPermissionContextHolder.DATA_PERMISSION_HEADER;
/**
* 解析数据权限规则，把解析后的数据放置在上下文中DataPermissionContext
*/
@ -81,8 +82,7 @@ public class DataPermissionInterceptor {
@Around(value = "@within(dataPermission) && @within(restController)")
@SneakyThrows
public Object classHandler(ProceedingJoinPoint pjp, DataPermission dataPermission, RestController restController) {
handle(dataPermission);
return proceed(pjp);
return proceed(pjp, dataPermission);
}
/**
@ -91,16 +91,16 @@ public class DataPermissionInterceptor {
@Around(value = "@annotation(dataPermission) && mappingAnnotations()")
@SneakyThrows
public Object methodHandler(ProceedingJoinPoint pjp, DataPermission dataPermission) {
handle(dataPermission);
return proceed(pjp);
return proceed(pjp, dataPermission);
}
private static Object proceed(ProceedingJoinPoint pjp) throws Throwable {
private Object proceed(ProceedingJoinPoint pjp, DataPermission dataPermission) throws Throwable {
Object result;
try {
handle(dataPermission);
result = pjp.proceed(pjp.getArgs());
} finally {
ContextInfoHolder.clear();
DataPermissionContextHolder.clear();
}
return result;
}

3
axzo-common-data-permission/src/main/java/cn/axzo/framework/datapermission/context/DataPermissionContextHolder.java
View File

@ -24,6 +24,9 @@ public class DataPermissionContextHolder {
DATA_PERMISSION_CONTEXT.set(dataPermission);
}
public static void clear() {
DATA_PERMISSION_CONTEXT.remove();
}
@Data
@Builder