package com.common.aop;
 
import java.lang.annotation.*;
 
/**
 * @description :
 * @auther Tyler
 * @date 2021/8/30
 */
 
@Target({ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface IIdempotent {
}

package com.common.aop;
 
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
 
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;
 
/**
 * @description :
 * @auther Tyler
 * @date 2021/8/30
 */
 
@Component
@Aspect
public class IdempotentAction {
 
    @Autowired
    HttpServletRequest request;
    public final static String ERROR_REPEATSUBMIT = "Repeated submission";
 
    //redis
    @Autowired
    protected StringRedisTemplate idempotentTemplate;
 
    @Pointcut("@annotation(com.tenyears.aop.IIdempotent)")
    private void controllerAspect() {
    }
 
    @Before("controllerAspect()")
    public void BeforeReturning(JoinPoint jp) throws Exception {
        IdempotentCheck(jp);
    }
 
    private void IdempotentCheck(JoinPoint jp) throws Exception {
//controller 名
        String controllerName = jp.getTarget().getClass().getName();
        Signature sig = jp.getSignature();
//方法名
        MethodSignature msig = (MethodSignature) sig;
        String method = msig.getMethod().getName();
//参数
        String body = NetUtil.getBodyString(request);
        //md5(Controller + method + body);
//md5 验证
        String signMD5= MD5.md5(controllerName+method+body);
        System.out.println("signMD5 :"+signMD5);
//存入redis 10秒过期
        if (Boolean.parseBoolean(idempotentTemplate.opsForValue().get(signMD5))) {
            throw new RuntimeException(ERROR_REPEATSUBMIT);
        } else {
            idempotentTemplate.opsForValue().set(signMD5, "true", 10, TimeUnit.SECONDS);
        }
 
    }
}
 
 
// netUtil.getBodyString方法
public static String getBodyString(HttpServletRequest request) throws Exception {
        request.setCharacterEncoding("UTF-8");
        BufferedReader br = request.getReader();
        String str = "";
        StringBuilder sBuilder = new StringBuilder("");
        while ((str = br.readLine()) != null) {
            sBuilder.append(str);
        }
        return sBuilder.toString();
    }

@SpringBootApplication
//必须要扫描到aop
@ComponentScan(basePackages={"com.common.aop"},lazyInit=true)
public class WebADApp {
    public static void main( String[] args )
    {
        SpringApplication.run(WebADApp.class, args);
 
    }
}

@IIdempotent
    @RequestMapping(value = "test7", method = RequestMethod.POST)
    public void test7(ApiResult<String> req) {
 
        System.out.println(req.getData());
    }