转载

springboot + aop的最佳实践,再也不怕双击666

复工不到一个月疯狂的加班中,最近没什么时间写博客更文有点慢,好在做的项目中不全是增删改的业务逻辑,还是有一些比较有意思实用的技术点,所以整理出来分享给大家。和那些搞高端技术的大佬比不了,咱就是个还在一线搬砖的码农。

有没有遇到过这种情况:由于网速等原因,网页响应很慢,提交一次表单后发现服务久久没响应,然后你就疯狂点击提交按钮(12306就经常被这样怒怼),如果做过防重复提交还好,否则那是什么级别的灾难就不好说了。。。

今天主要是用 自定义注解、 AOP、 · Guava 包中Cache来生成一种本地锁,来达到的防重复提交效果,整体的实现比较简单,没有什么太大的难度,代码也是比较少,,由于是基于内存的缓存, 因此这种实现方式并不适用于分布式服务 。旨在给大家介绍一种实现防重复提交的方案,要是有什么说的不对的地方大家温柔一点撕,毕竟人家还是个20出头30郎当岁的孩子。

springboot + aop的最佳实践,再也不怕双击666

Guava是什么?

guava 包是个啥?做过Java的小伙伴应该多少都有所了解,它是 google “嫌弃” JAVA 自带的类库不好用,自行研发的一套工具包,对JDK工具做了很好的拓展。例如:并发[ Concurrency ]、缓存[ Caches ]、 函数式风格[ Functional idioms ]、 字符串处理[ Strings ]等。

总之一句话 guava 包很好用多去了解下,能不造的轮子咱们就尽量不去造,因为咱们自己造的轮子可能不太圆。不多说废话了,来看看具体的实现。

代码撸起来

1、引入Guava包依赖

第一步引入 Guava 的依赖包没什么好说的

<dependency>
        <groupId>com.google.guava</groupId>
        <artifactId>guava</artifactId>
        <version>21.0</version>
</dependency>
复制代码

2、自定义注解

服务端实现防止重复提交,一般都是利用 AOP 自定义注解的的方式,作用于 controller 的入口方法。自定义一个 LocalLock 注解用于需要防止重复提交的方法上。

/**
 * 锁的注解
 *
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface LocalLock {

    /**
     * @author fly
     */
    String key() default "";
}
复制代码

注解定义好以后就需要做AOP拦截器切面的具体实现,在 interceptor() 方法上采用的是 Around (环绕增强) ,因此所有带 LocalLock 注解的都将被切面处理;

@Around("execution(public * *(..)) && @annotation(com.chengxy.annotation.LocalLock)")
复制代码

既然是缓存,那紧跟的属性一定要有过期时间,通过 expireAfterWrite 设置缓存的过期时间, maximumSize 设置缓存的个数。

通过在内存中查询key是否存在来判断是否让再次提交,和 RedissetNX 方法是一个比较像。

这里我们设置同一个方法,5秒钟内相同参数的请求只允许执行一次。

那么这个注解该怎么用呢?

@Aspect
@Configuration
public class LockMethodInterceptor {

    private static final Cache<String, Object> CACHES = CacheBuilder.newBuilder()
            // 最大缓存 100 个
            .maximumSize(1000)
            // 设置写缓存后 5 秒钟过期
            .expireAfterWrite(5, TimeUnit.SECONDS)
            .build();

    @Around("execution(public * *(..)) && @annotation(com.chengxy.annotation.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 < args.length; i++) {
            keyExpress = keyExpress.replace("arg[" + i + "]", args[i].toString());
        }
        return keyExpress;
    }
}
复制代码

3、控制层的实现

我们将注解加在控制层方法上, key = "city:arg[0] key自己定义, arg[0] 这个匹配规则表示替换成第一个参数。那么就实现 city:token 在一定时间内不可以重复提交了

@RestController
@RequestMapping("/city")
public class BookController {

    @LocalLock(key = "city:arg[0]")
    @GetMapping
    public String query(@RequestParam String token) {
        return "ok- " + token;
    }
}
复制代码

4、测试

接下来我们就测试一下,预期结果:5秒内只有第一次的提交会正常返回,其余的显示 “请勿重复提交” ,看看执行结果是不是我们预期的那样,这里用 postman 测试。

第一次请求正常响应:

springboot + aop的最佳实践,再也不怕双击666

紧接着请求第二次,返回结果“重复提交”,显然我们实现成功了

springboot + aop的最佳实践,再也不怕双击666

今天就说这么多,如果本文对您有一点帮助,希望能得到您一个点赞:+1:哦

您的认可才是我写作的动力!

小福利:

通过合法手段,获取到一些极客付费课程 ,嘘~, 免费 送给小伙伴们。公众号回复【 极客 】自行领取

整理了一些Java方面的架构、面试资料,有需要的小伙伴可以关注公众号【 程序员内点事

springboot + aop的最佳实践,再也不怕双击666
原文  https://juejin.im/post/5e800631518825736512c038
正文到此结束
Loading...