第一步先创建一个注解类
@Retention(RetentionPolicy.RUNTIME)
@Target(value = {ElementType.METHOD})
@Documented
public @interface CompanyConditionAnno {
}
第二步,定义切面
@Aspect
@Component
public class LogAspect {
//定义切点,指向一个我刚定义的注解
@Pointcut("@annotation(org.siteinfo.conf.CompanyConditionAnno)")
public void companyCodeCondition() {
}
//定义切面,这里使用的是最强大的@Around,从我刚指定的切点切入程序运行中
@Around(value = " companyCodeCondition() ")
public Object insertLog(ProceedingJoinPoint pc) throws Throwable {
MethodSignature signature = (MethodSignature) pc.getSignature();
//获取切入方法的对象
Method method = signature.getMethod();
//获取参数值
Object[] args = pc.getArgs();
//获取参数名
DefaultParameterNameDiscoverer discoverer = new DefaultParameterNameDiscoverer();
String[] parameterNames = discoverer.getParameterNames(method);
//对这些参数名进行遍历,找到我们要修改的参数,获取到这个参数名在数组中的下标,并将我们需要赋的值放到参数值数据的相应位置,进行参数替换。
我这边是对所有参数中 的companyCode参数进行的拦截,如果参数是空串,我这边就给方法设置一个默认值。
for (int i = 0; i < parameterNames.length; i++) {
if (StringUtils.isEmpty(args[i]) && (parameterNames[i].contains("companyCode") || parameterNames[i].contains("company_code") || parameterNames[i].contains("companycode"))) {
args[i] = "WHZSSC01001";
}
}
//执行程序,带上修改过的参数
Object proceed = pc.proceed(args);
return proceed;
}
}