SpringBoot学习：本地锁解决重复提交问题

• 自定义注解

/**
 * @author: hs
 * @Date: 2019/4/11 18:14
 * @Description:
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface LocalLock {
    /**
     * id或者token
     */
    String key() default "";

    /**
     * 过期时间 TODO 由于用的 guava 暂时就忽略这属性吧 集成 redis 需要用到
     *
     * @author fly
     */
    int expire() default 5;
}

• 自定义拦截器

@Aspect
@Configuration
public class LocalLockInterceptor {

    private static final Cache<String, Object> CACHES = CacheBuilder.newBuilder()
            //最大缓存数
            .maximumSize(1000L)
            //缓存key过期时间
            .expireAfterWrite(5L, TimeUnit.SECONDS)
            .build();

    @Around("execution(public * *(..)) && @annotation(com.db.mysqldb.config.lock.LocalLock)")
    public Object interceptor(ProceedingJoinPoint pjp) {
        //获取方法映射
        MethodSignature signature = (MethodSignature) pjp.getSignature();
        //获取方法
        Method method = signature.getMethod();
        LocalLock localLock = method.getAnnotation(LocalLock.class);
        String key = getKey(localLock.key(), pjp.getArgs());
        if (!StringUtils.isEmpty(key)) {
            if (!Objects.equal(CACHES.getIfPresent(key), null)) {
                throw new RuntimeException("请勿重复请求");
            }
            // 如果是第一次请求,就将 key 当前对象压入缓存中
            CACHES.put(key, key);
        }
        try {
            return pjp.proceed();
        } catch (Throwable throwable) {
            throw new RuntimeException("服务器异常");
        } finally {
            // TODO 为了演示效果,这里就不调用 CACHES.invalidate(key); 代码了
            //CACHES.invalidate(key);
        }
    }

    /**
     * key 的生成策略,如果想灵活可以写成接口与实现类的方式
     *
     * @param keyExpress 表达式
     * @param args       参数
     * @return 生成的key
     */
    private String getKey(String keyExpress, Object[] args) {
        for (int i = 0; i < args.length; i++) {
            keyExpress = keyExpress.replace("arg[" + i + "]", args[i].toString());
        }
        return keyExpress;
    }

}

• controller

@RestController
@Slf4j
public class LockController {

    @LocalLock(key = "lock:arg[0]")
    @GetMapping("/lock/{id}")
    public String lock(@PathVariable("id") Integer id){

        return "success"+id;
    }
}

-测试如图：