唠唠叨叨
复工不到一个月疯狂的加班中，最近没什么时间写博客更文有点慢，好在作的项目中不全是增删改的业务逻辑，仍是有一些比较有意思实用的技术点，因此整理出来分享给你们。和那些搞高端技术的大佬比不了，咱就是个还在一线搬砖的码农。javascript
有没有遇到过这种状况：因为网速等缘由，网页响应很慢，提交一次表单后发现服务久久没响应，而后你就疯狂点击提交按钮（12306就常常被这样怒怼），若是作过防重复提交还好，不然那是什么级别的灾难就很差说了。。。java
今天主要是用 自定义注解、 AOP、· Guava 包中Cache来生成一种本地锁，来达到的防重复提交效果，总体的实现比较简单，没有什么太大的难度，代码也是比较少，，因为是基于内存的缓存，所以这种实现方式并不适用于分布式服务。旨在给你们介绍一种实现防重复提交的方案，要是有什么说的不对的地方你们温柔一点撕，毕竟人家仍是个20出头30郎当岁的孩子。程序员
Guava是什么？
guava包是个啥？作过Java的小伙伴应该多少都有所了解，它是google “嫌弃” JAVA自带的类库很差用，自行研发的一套工具包，对JDK工具作了很好的拓展。例如：并发[Concurrency]、缓存[Caches]、 函数式风格[Functional idioms]、 字符串处理[Strings]等。缓存
总之一句话guava包很好用多去了解下，能不造的轮子我们就尽可能不去造，由于我们本身造的轮子可能不太圆。很少说废话了，来看看具体的实现。服务器
代码撸起来
一、引入Guava包依赖
第一步引入Guava的依赖包没什么好说的并发
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>21.0</version>
</dependency>
二、自定义注解
服务端实现防止重复提交，通常都是利用AOP自定义注解的的方式，做用于controller的入口方法。自定义一个LocalLock注解用于须要防止重复提交的方法上。app
/**
* 锁的注解
*
*/
@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是否存在来判断是否让再次提交，和Redis的setNX方法是一个比较像。工具
这里咱们设置同一个方法，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;
}
}
三、控制层的实现
咱们将注解加在控制层方法上，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;
}
}
四、测试
接下来咱们就测试一下，预期结果：5秒内只有第一次的提交会正常返回，其他的显示“请勿重复提交”，看看执行结果是否是咱们预期的那样，这里用postman测试。
第一次请求正常响应：
紧接着请求第二次，返回结果“重复提交”，显然咱们实现成功了
小福利：
获取到一些极客课程 ，嘘~，免费 送给小伙伴们。公号【程序员内点事】回复【极客】自行领取