Tenlion/cm-cloud
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加参数自动校验位置

Browse Source
This commit is contained in:
wenc000 2020-05-13 17:06:59 +08:00
parent 16adee1c52
commit e9d8315ef0
1 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
cloud-common-plugin/src/main/java/com/cm/common/aspect/ApiParamsAspect.java
View File

@ -46,6 +46,14 @@ public class ApiParamsAspect {
public void oauthCutPoint() { public void oauthCutPoint() {
} }
@Pointcut("execution(public * com.cm.common.article.controller..*.*(..))")
public void articleCutPoint() {
}
@Pointcut("execution(public * com.cm.common.wechat.controller..*.*(..))")
public void wechatCutPoint() {
}
@Before("apiCutPoint()") @Before("apiCutPoint()")
public void beforeApiCutPoint(JoinPoint joinPoint) throws ParamsException { public void beforeApiCutPoint(JoinPoint joinPoint) throws ParamsException {
beforeCutPoint(joinPoint); beforeCutPoint(joinPoint);
@ -61,6 +69,16 @@ public class ApiParamsAspect {
beforeCutPoint(joinPoint); beforeCutPoint(joinPoint);
} }
@Before("articleCutPoint()")
public void beforeArticleCutPoint(JoinPoint joinPoint) throws ParamsException {
beforeCutPoint(joinPoint);
}
@Before("wechatCutPoint()")
public void beforeWechatCutPoint(JoinPoint joinPoint) throws ParamsException {
beforeCutPoint(joinPoint);
}
private void beforeCutPoint(JoinPoint joinPoint) throws ParamsException { private void beforeCutPoint(JoinPoint joinPoint) throws ParamsException {
Signature signature = joinPoint.getSignature(); Signature signature = joinPoint.getSignature();
Object[] args = joinPoint.getArgs(); Object[] args = joinPoint.getArgs();
@ -73,10 +91,10 @@ public class ApiParamsAspect {
if (method.isAnnotationPresent(CheckRequestBodyAnnotation.class)) { if (method.isAnnotationPresent(CheckRequestBodyAnnotation.class)) {
LOG.debug("校验参数"); LOG.debug("校验参数");
Parameter[] parameters = method.getParameters(); Parameter[] parameters = method.getParameters();
for(Parameter parameter : parameters) { for (Parameter parameter : parameters) {
if(parameter.isAnnotationPresent(RequestBody.class)) { if (parameter.isAnnotationPresent(RequestBody.class)) {
for(Object arg: args) { for (Object arg : args) {
if(parameter.getType() == arg.getClass()) { if (parameter.getType() == arg.getClass()) {
AnnotationUtil.checkField(arg); AnnotationUtil.checkField(arg);
} }
} }