唠唠叨叨
复工不到一个月疯狂的加班中，最近没什么时间写博客更文有点慢，好在做的项目中不全是增删改的业务逻辑，还是有一些比较有意思实用的技术点，所以整理出来分享给大家。和那些搞高端技术的大佬比不了，咱就是个还在一线搬砖的码农。
有没有遇到过这种情况：由于网速等原因，网页响应很慢，提交一次表单后发现服务久久没响应，然后你就疯狂点击提交按钮（12306就经常被这样怒怼），如果做过防重复提交还好，否则那是什么级别的灾难就不好说了。。。
今天主要是用 自定义注解、 AOP、· Guava 包中Cache来生成一种本地锁，来达到的防重复提交效果，整体的实现比较简单，没有什么太大的难度，代码也是比较少，，由于是基于内存的缓存，因此这种实现方式并不适用于分布式服务。旨在给大家介绍一种实现防重复提交的方案，要是有什么说的不对的地方大家温柔一点撕，毕竟人家还是个20出头30郎当岁的孩子。
在这里插入图片描述
Guava是什么？
guava包是个啥？做过Java的小伙伴应该多少都有所了解，它是google “嫌弃” JAVA自带的类库不好用，自行研发的一套工具包，对JDK工具做了很好的拓展。例如：并发[Concurrency]、缓存[Caches]、 函数式风格[Functional idioms]、 字符串处理[Strings]等。
总之一句话guava包很好用多去了解下，能不造的轮子咱们就尽量不去造，因为咱们自己造的轮子可能不太圆。不多说废话了，来看看具体的实现。
代码撸起来
1、引入Guava包依赖
第一步引入Guava的依赖包没什么好说的
1<dependency>
2 <groupId>com.google.guava</groupId>
3 <artifactId>guava</artifactId>
4 <version>21.0</version>
5</dependency>2、自定义注解
服务端实现防止重复提交，一般都是利用AOP自定义注解的的方式，作用于controller的入口方法。自定义一个LocalLock注解用于需要防止重复提交的方法上。
1/**
2 * 锁的注解
3 *
4 */
5@Target(ElementType.METHOD)
6@Retention(RetentionPolicy.RUNTIME)
7@Documented
8@Inherited
9public @interface LocalLock {
10
11 /**
12 * @author fly
13 */
14 String key() default \"\";
15}注解定义好以后就需要做AOP拦截器切面的具体实现，在 interceptor() 方法上采用的是 Around（环绕增强） ，因此所有带 LocalLock 注解的都将被切面处理；
1 @Around(\"execution(public * *(..)) && @annotation(com.chengxy.annotation.LocalLock)\")既然是缓存，那紧跟的属性一定要有过期时间，通过expireAfterWrite 设置缓存的过期时间，maximumSize设置缓存的个数。
通过在内存中查询key是否存在来判断是否让再次提交，和Redis的setNX方法是一个比较像。
这里我们设置同一个方法，5秒钟内相同参数的请求只允许执行一次。
那么这个注解该怎么用呢？
1@Aspect
2@Configuration
3public class LockMethodInterceptor {
4
5 private static final Cache<String, Object> CACHES = CacheBuilder.newBuilder()
6 // 最大缓存 100 个
7 .maximumSize(1000)
8 // 设置写缓存后 5 秒钟过期
9 .expireAfterWrite(5, TimeUnit.SECONDS)
10 .build();
11
12 @Around(\"execution(public * *(..)) && @annotation(com.chengxy.annotation.LocalLock)\")
13 public Object interceptor(ProceedingJoinPoint pjp) {
14 MethodSignature signature = (MethodSignature) pjp.getSignature();
15 Method method = signature.getMethod();
16 LocalLock localLock = method.getAnnotation(LocalLock.class);
17 String key = getKey(localLock.key(), pjp.getArgs());
18 if (!StringUtils.isEmpty(key)) {
19 if (CACHES.getIfPresent(key) != null) {
20 throw new RuntimeException(\"请勿重复请求\");
21 }
22 // 如果是第一次请求,就将 key 当前对象压入缓存中
23 CACHES.put(key, key);
24 }
25 try {
26 return pjp.proceed();
27 } catch (Throwable throwable) {
28 throw new RuntimeException(\"服务器异常\");
29 } finally {
30 // TODO 为了演示效果,这里就不调用 CACHES.invalidate(key); 代码了
31 }
32 }
33
34 /**
35 * key 的生成策略,如果想灵活可以写成接口与实现类的方式（TODO 后续讲解）
36 *
37 * @param keyExpress 表达式
38 * @param args 参数
39 * @return 生成的key
40 */
41 private String getKey(String keyExpress, Object[] args) {
42 for (int i = 0; i < args.length; i++) {
43 keyExpress = keyExpress.replace(\"arg[\" + i + \"]\", args[i].toString());
44 }
45 return keyExpress;
46 }
47}3、控制层的实现
我们将注解加在控制层方法上，key = \"city:arg[0] key自己定义，arg[0]这个匹配规则表示替换成第一个参数。那么就实现city:token在一定时间内不可以重复提交了
1@RestController
2@RequestMapping(\"/city\")
3public class BookController {
4
5 @LocalLock(key = \"city:arg[0]\")
6 @GetMapping
7 public String query(@RequestParam String token) {
8 return \"ok- \" + token;
9 }
10}4、测试
接下来我们就测试一下，预期结果：5秒内只有第一次的提交会正常返回，其余的显示“请勿重复提交”，看看执行结果是不是我们预期的那样，这里用postman测试。
第一次请求正常响应：
在这里插入图片描述
紧接着请求第二次，返回结果“重复提交”，显然我们实现成功了
在这里插入图片描述
今天就说这么多，如果本文对您有一点帮助，希望能得到您一个点赞哦
您的认可才是我写作的动力！
小福利：
评论【666】私信 ，有一份极客时间课程小礼物