在项目中引入 redis
定义注解类
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface Idempotent {
String[] fields();
long timeout() default 10l;
}
定义aop 切面实现类
import cn.hutool.crypto.digest.MD5;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.zalando.problem.Problem;
import org.zalando.problem.Status;
import javax.annotation.Resource;
import java.util.LinkedHashMap;
import java.util.List;
@Aspect
@Component
@Slf4j
public class IdempotentAspect {
@Resource
private RedisCache redisCache;
@Pointcut("@annotation(com.*.*.common.annotation.Idempotent)")
public void pointcut() {
}
@Before(value = "pointcut() && @annotation(idempotent)")
public void doBefore(JoinPoint joinPoint, Idempotent idempotent) {
List<String> args = Lists.newArrayList(idempotent.fields());
MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
String[] parameterNames = methodSignature.getParameterNames();
LinkedHashMap<String, Object> params = Maps.newLinkedHashMap();
for (int i = 0; i < parameterNames.length; i++) {
String name = parameterNames[i];
if (args.contains(name)) {
params.put(name, joinPoint.getArgs()[i]);
}
}
String paramsStr = JsonUtils.objToString(params);
String md5 = MD5.create().digestHex(paramsStr.getBytes());
if(!redisCache.setIfAbsent(md5, md5, 10)){
throw Problem.valueOf(Status.BAD_REQUEST, "重复提交");
}
}
}
在controller 方法中添加@Idempotent注解
@PostMapping("/submit")
@Idempotent(fields = {"info"},timeout = 10)
public OrderInfoResDTO submit(@Valid @RequestBody OrderInfoDTO info) {
throw Problem.valueOf(Status.BAD_REQUEST, "正常结束流程");
}