Springboot AOP利用反射获取请求参数
2019-06-10 本文已影响0人
追杀丘比特
思想:
我们在某些业务场景下需要对接口的入参进行校验或者权限验证,因此需要获取接口的参数列表依次来支持我们的逻辑操作,因此需要我们获取接口的参数,下面是利用自定义注解配合Aop来实现的一个思路:
- 首先定义一个切面类:
@Aspect 用于声明一个类为切面
加在类上,如下:
@Aspect
@Component
public class AuthorizationAspect {
...
}
- 自定义注解
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Authorization {
/**
* 业务代码
* @return
*/
String[] businessType();
}
- 定义获取参数的方法
/**
* 获取参数列表
*
* @param joinPoint
* @return
* @throws ClassNotFoundException
* @throws NoSuchMethodException
*/
private static Map<String, Object> getFieldsName(ProceedingJoinPoint joinPoint) {
// 参数值
Object[] args = joinPoint.getArgs();
ParameterNameDiscoverer pnd = new DefaultParameterNameDiscoverer();
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
Method method = signature.getMethod();
String[] parameterNames = pnd.getParameterNames(method);
Map<String, Object> paramMap = new HashMap<>(32);
for (int i = 0; i < parameterNames.length; i++) {
paramMap.put(parameterNames[i], args[i]);
}
return paramMap;
}
- 定义Aop,进行注解扫描
@Aspect
@Component
public class AuthorizationAspect {
/**
定义切点
*/
@Pointcut("@annotation(Authorization)")
public void executeService() {}
/**
环绕织入
*/
@Around("executeService()")
public Object proceed(ProceedingJoinPoint thisJoinPoint) throws Throwable {
MethodSignature signature = (MethodSignature) thisJoinPoint.getSignature();
Method method = signature.getMethod();
Authorization authCode = method.getAnnotation(Authorization.class);
Object[] args = thisJoinPoint.getArgs();
//获取到请求参数
Map<String, Object> fieldsName = getFieldsName(thisJoinPoint);
...
}
}
- 以上都定义好了之后,在需要验证的controller接口上加上自定义注解
@Authorization
,如下:
@Api(tags = {"接口"})
@RestController
@RequestMapping("test")
public class ResearchController {
@Authorization(businessType = "6")
@ApiOperation(value = "测试")
@PostMapping("/get")
public ResponseResult get(String id,@RequestBody List<Integer> sourceTypes){
System.out.println(id);
return new ResponseResult();
}
这样在请求接口的时候,我们就能拿到请求参数列表