灵活实现自定义注解

自定义注解传参灵活实现

1.业务背景：spring boot + 实现业务级别的防重复提交

2.实现方式：提交方法上面加注解，传入业务id或业务参数作为key

3.代码实现

3.1 自定义注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface SetLock {
    String value();
}

3.2 注解处理逻辑

@Aspect
@Component
public class AnAspect {

    @Around("@annotation(SetLock)")
    public Object around(ProceedingJoinPoint joinPoint, SetLock an) throws Throwable {
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();

        String[] parameterNames = signature.getParameterNames();
        Object[] args = joinPoint.getArgs();
		
		// 使用spel实现取值，可以注册默认对象或默认方法处理
        StandardEvaluationContext context = new StandardEvaluationContext();
        ExpressionParser parser = new SpelExpressionParser();
        for (int i = 0; i < parameterNames.length; i++) {
            //注册方法参数
            context.setVariable(parameterNames[i], args[i]);
        }

        // 获取注解参数
        String value = an.value();
        // 获取注解参数中{}的参数
        Pattern pattern = Pattern.compile("\\{(.*?)\\}");
        Matcher matcher = pattern.matcher(value);        

        StringBuffer sb = new StringBuffer();
        while (matcher.find()) {
            String expression = matcher.group(1);
            Object result = parser.parseExpression(expression).getValue("#" +context);
             matcher.appendReplacement(sb, result.toString());
        }
		matcher.appendTail(sb);
        
        //这里就组装好了最后的key，如：Key:id-100002-00001
        String key = sb.toString();
        //拿到key之后可以处理自己的逻辑了，需要锁的话就写锁的逻辑，需要其他处理就写其他逻辑
        
        
        return joinPoint.proceed();
    }
}

3.3 注解使用

@Service
public class YourService {

    @SetLock("Key:id-{id}-{user.code}") // 这里支持spel表达式方式传值，如 user.name、list[0].code 等
    public void someMethod(String id, int param2, User user) {
        // 方法实现
    }
}