Spring Boot使用注解实现接口幂等

在Spring Boot中，我们可以使用自定义注解和AOP（面向切面编程）来实现接口的幂等性。下面是一个示例代码，演示了如何创建一个@Idempotent注解，并使用AOP在方法执行前进行幂等性校验。

1. 创建自定义注解

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Idempotent {
}

2. 创建切面类

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;

@Aspect
@Component
public class IdempotentAspect {
    private Map cache = new HashMap<>();

    @Around("@annotation(idempotent)")
    public Object checkIdempotent(ProceedingJoinPoint joinPoint, Idempotent idempotent) throws Throwable {
        // 获取方法参数，生成幂等性的唯一标识
        String key = generateKey(joinPoint.getArgs());

        // 检查缓存中是否存在该标识，如果存在，则表示已经执行过该方法，直接返回结果
        if (cache.containsKey(key)) {
            return cache.get(key);
        }

        // 执行方法
        Object result = joinPoint.proceed();

        // 将方法执行结果放入缓存
        cache.put(key, result);

        return result;
    }

    private String generateKey(Object[] args) {
        // 根据方法参数生成唯一标识，可以根据具体业务需求定制
        // 这里简单地将参数数组转换为字符串作为标识
        return String.join("_", args);
    }
}

3. 使用自定义注解

@RestController
public class MyController {
    @Autowired
    private MyService myService;

    @PostMapping("/my-api")
    @Idempotent
    public String myApi(@RequestBody MyRequest request) {
        // 执行业务逻辑
        return myService.processRequest(request);
    }
}

在上面的示例中，我们创建了一个@Idempotent注解，并在IdempotentAspect切面类中使用@Around注解来拦截带有@Idempotent注解的方法。在拦截的方法中，我们首先根据方法参数生成幂等性的唯一标识，然后检查缓存中是否存在该标识。如果存在，则直接返回缓存中的结果；如果不存在，则执行方法，并将方法执行结果存入缓存。

这样，我们就可以通过在需要实现幂等性的接口方法上添加@Idempotent注解来实现接口的幂等性。

请注意，上述示例只是一个简单的演示，实际应用中需要根据具体业务需求进行适当的修改和扩展。

 公众号请关注 "果酱桑", 一起学习,一起进步!