嘮嘮叨叨
復工不到一個月瘋狂的加班中，最近沒什麼時間寫博客更文有點慢，好在作的項目中不全是增刪改的業務邏輯，仍是有一些比較有意思實用的技術點，因此整理出來分享給你們。和那些搞高端技術的大佬比不了，咱就是個還在一線搬磚的碼農。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測試。
第一次請求正常響應：
緊接着請求第二次，返回結果「重複提交」，顯然咱們實現成功了
小福利：
獲取到一些極客課程 ，噓~，免費 送給小夥伴們。公號【程序員內點事】回覆【極客】自行領取