SpringBoot 防止表单重复提交-本地锁(11)

        org.springframework.boot
            spring-boot-starter-web
        org.springframework.boot
            spring-boot-starter-aop
        com.google.guava
            guava
            21.0

package cn.hanzhenya.anno;

import java.lang.annotation.*;

/**
 * @Classname LocalLock
 * @Description 防止重复提交的注解
 * @Date 2020/10/11 10:23
 * @Created by hzy
 * @Version 1.0
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface LocalLock {

    String key() default "";
    /**
     * TODO 由于用的 guava 暂时就忽略该属性吧 集成 redis 需要用到
     * 默认时间5秒
     */
    int expire() default 5 * 1000;
}

package cn.hanzhenya.interceptor;

import cn.gorit.anno.LocalLock;
import com.google.common.cache.Cache;
import com.google.common.cache.CacheBuilder;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Configuration;
import org.springframework.util.StringUtils;

import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;

/**
 * @Classname LockMethodInterceptor
 * @Description TODO
 * @Date 2020/10/12 19:50
 * @Created by hzy
 * @Version 1.0
 */
@Aspect
@Configuration
public class LockMethodInterceptor {
    // 构建缓存对象，设置过期时间
    private static final Cache CACHES = CacheBuilder.newBuilder()
            // 最大缓存 100 个
            .maximumSize(1000)
            // 设置写缓存后 5 秒钟过期
            .expireAfterWrite(5, TimeUnit.SECONDS)
            .build();

    // 在 interceptor() 上采用 环绕通知，以及指定 @annotation 所有带 LocalLock 注解都将被切面处理
    @Around("execution(public * *(..)) && @annotation(cn.gorit.anno.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 (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); 代码了
        }
    }

    /**
     * key 的生成策略,如果想灵活可以写成接口与实现类的方式（TODO 后续讲解）
     *
     * @param keyExpress 表达式
     * @param args       参数
     * @return 生成的key
     */
    private String getKey(String keyExpress, Object[] args) {
        for (int i = 0; i 

// 这里模拟增删改查，就不用数据库了
@RestController
public class FirstController {

    static List list = new ArrayList();
    {
        // 初始化数据，没有用到数据库，所有数据都是模拟的
        list.add(new User(1,"牛二",13));
        list.add(new User(2,"张三",14));
        list.add(new User(3,"王五",15));
    }
    /**
     * 查询所有的用户
     * @return
     */
    @RequestMapping(value = "/user",method = RequestMethod.GET)
    public Result queryAllUsers() {
        // Result 也是自定义的一个实体类
        return new Result(200,"查询成功",list);
    }

    /**
     * 根据 ID 查询用户
     */
    @RequestMapping(value = "/user/{id}",method = RequestMethod.GET)
    public Result queryUserById(@PathVariable("id") Integer id) {
        return new Result(200,"查询成功",list.get(id-1));
    }

    /**
     * 添加用户，返回添加的结果
     */
    @LocalLock(key = "user:arg[0]")
    @RequestMapping(value = "/user",method = RequestMethod.POST)
    public Result addUser(User user) {
        list.add(user);
        return new Result(200,"添加成功",list);
    }

查询所有用户
    查询所有用户
    
根据 ID 查询用户

    请输入用户ID：查询

    
添加用户