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

Merge branch 'new_3A' of https://axzsource.com/infra/axzo-framework.git into new_3A

Browse Source
This commit is contained in:
zhourui 周锐 2022-08-02 13:22:01 +08:00
commit 990fd2493d
1 changed files with 50 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
axzo-auth-spring-boot-starter/src/main/java/cn/axzo/framework/auth/service/ContextInfoBuilderAspect.java
View File

@ -1,19 +1,5 @@
package cn.axzo.framework.auth.service; package cn.axzo.framework.auth.service;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Objects;
import javax.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.Order;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import cn.axzo.framework.auth.AuthException; import cn.axzo.framework.auth.AuthException;
import cn.axzo.framework.auth.annotation.PreBuildContext; import cn.axzo.framework.auth.annotation.PreBuildContext;
import cn.axzo.framework.auth.constants.AuthConstants; import cn.axzo.framework.auth.constants.AuthConstants;
@ -34,8 +20,21 @@ import cn.hutool.core.text.CharSequenceUtil;
import cn.hutool.http.HttpException; import cn.hutool.http.HttpException;
import cn.hutool.http.HttpRequest; import cn.hutool.http.HttpRequest;
import cn.hutool.json.JSONUtil; import cn.hutool.json.JSONUtil;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Objects;
import javax.servlet.http.HttpServletRequest;
import lombok.SneakyThrows; import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.core.annotation.Order;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
@Slf4j @Slf4j
@Aspect @Aspect
@ -58,12 +57,46 @@ public class ContextInfoBuilderAspect {
return (String) podNamespacePropertyDefiner.getSystemProperties().get("CUSTOM_ENV"); return (String) podNamespacePropertyDefiner.getSystemProperties().get("CUSTOM_ENV");
} }
@Around(value = "@within(preBuildContext)") @Pointcut(value = "@annotation(org.springframework.web.bind.annotation.RequestMapping)")
public Object classHandler(ProceedingJoinPoint pjp, PreBuildContext preBuildContext) { public void requestMapping() {
}
@Pointcut(value = "@annotation(org.springframework.web.bind.annotation.PostMapping)")
public void postMapping() {
}
@Pointcut(value = "@annotation(org.springframework.web.bind.annotation.GetMapping)")
public void getMapping() {
}
@Pointcut(value = "@annotation(org.springframework.web.bind.annotation.PutMapping)")
public void putMapping() {
}
@Pointcut(value = "@annotation(org.springframework.web.bind.annotation.DeleteMapping)")
public void deleteMapping() {
}
@Pointcut(value = "@annotation(org.springframework.web.bind.annotation.PatchMapping)")
public void patchMapping() {
}
@Pointcut("requestMapping() || postMapping() || getMapping() || putMapping() || deleteMapping()|| patchMapping()")
public void mappingAnnotations() {
}
/**
* 切入含有@PreBuildContext && @RestController 注解的类
*/
@Around(value = "@within(preBuildContext) && @within(restController)")
public Object classHandler(ProceedingJoinPoint pjp, PreBuildContext preBuildContext, RestController restController) {
return handle(pjp, preBuildContext); return handle(pjp, preBuildContext);
} }
@Around(value = "@annotation(preBuildContext)") /**
* 切入含有@PreBuildContext && @RequestMapping/@PostMapping/@GetMapping/@PutMapping/@DeleteMapping/@PatchMapping 之一注解的方法
*/
@Around(value = "@annotation(preBuildContext) && mappingAnnotations()")
public Object methodHandler(ProceedingJoinPoint pjp, PreBuildContext preBuildContext) { public Object methodHandler(ProceedingJoinPoint pjp, PreBuildContext preBuildContext) {
return handle(pjp, preBuildContext); return handle(pjp, preBuildContext);
} }