问题:有时候想对传入的参数,进行一些校验,并且方法比较多的情况下,可以参考使用注解或者直接AOP切点进行校验或者逻辑处理。
1 定义注解
@Inherited
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD})
public @interface RuntimeStatus {
/**
* 区分 project\stage\exam流程处理
*
* @return
*/
RuntimeStatusMeta key();
/**
* 指定参数值
*
* @return
*/
String propertyName() default "";
/**
* 指定参数是否是对象
*
* @return
*/
boolean isArgObj() default false;
}
2 定义AOP
@Aspect
@Component
@Slf4j
public class RuntimeStatusAop {
// cloud.core.annotion.RuntimeStatus:RuntimeStatus包地址
@Pointcut(value = "@annotation(cloud.core.annotion.RuntimeStatus)")
private void cutPermission() {
}
// 直接使用AOP切面
// @Pointcut("execution(public * cloud.controller.ExamController.*(..))")
// private void examState() {
//
// }
@Before("cutPermission()")
public void doSession(JoinPoint point) {
// 获取注解值
MethodSignature methodSignature = (MethodSignature) point.getSignature();
Method method = methodSignature.getMethod();
RuntimeStatus runtimeStatus = method.getAnnotation(RuntimeStatus.class);
RuntimeStatusMeta meta = runtimeStatus.key();
String propertyName = runtimeStatus.propertyName();
Object value = null;
// 请求的方法参数值
Object[] args = point.getArgs();
// 获取ID
if (runtimeStatus.isArgObj()) {
value = PropertyUtil.getFieldValueByName(args[0], propertyName);
} else {
// 请求的方法参数名称
LocalVariableTableParameterNameDiscoverer parameterNameDiscoverer = new LocalVariableTableParameterNameDiscoverer();
String[] paramNames = parameterNameDiscoverer.getParameterNames(method);
if (args != null && paramNames != null) {
for (int i = 0; i < args.length; i++) {
// params += " " + paramNames[i] + ": " + args[i];
if (paramNames[i].equals(propertyName)) {
value = args[i];
break;
}
}
}
}
log.info("propertyName => {}", propertyName);
log.info("meta => {}", meta.getName() + "、" + meta.getCode());
log.info("value => {}",value);
// 获取了参数值,接下来逻辑处理
。。。
}
说明:注解和AOP各有优劣势,AOP表达式直接切点的话,可以使用在无参数传入或者参数具备同样格式的方式;注解的话适用于需要处理传入参数,并且参数名称或则对象不统一的情况;
3 使用
(1) 如果传入的是基本类型的参数,直接PropertyName 直接写参数名称
@RequestMapping(value = "/delById", method = RequestMethod.POST)
@RuntimeStatus(key = RuntimeStatusMeta.EXAM, propertyName = "id")
public ResultUtil delById(int id) {
}
(2) 如果传入的是对象类型参数,标记isArgObj为true,然后指定对象里面的属性名称,这样就可以在AOP里,单独对是对象参数获取其属性值
@RequestMapping(value = "/update", method = RequestMethod.POST)
@RuntimeStatus(key = RuntimeStatusMeta.EXAM,isArgObj = true,propertyName = "id")
public ResultUtil update(@RequestBody Exam obj) {