Springboot 2.x 如何解決重複提交 (本地鎖的實踐)




有沒有遇到過這種情況:網頁響應很慢,提交一次表單後發現沒反應,然後你就瘋狂點擊提交按鈕(12306就經常被這樣怒懟),如果做過防重複提交還好,否則那是什麼級別的災難就不好說了。。。

本文主要是應用 自定義註解、 spring AOP、· Guava Cache 生成一種本地鎖,來達到的防重複提交效果,由於是基於內存的緩存,所以這種實現方式並不適用於分佈式服務

Guava是什麼?

guava包是google嫌棄JAVA自帶的類庫不好用,自行研發的一套工具包,對JDK工具做了很好的拓展。例如:併發[Concurrency]、緩存[Caches]、 函數式風格[Functional idioms]、 字符串處理[Strings]等等。

一、引入Guava包依賴

<code><dependency>
<groupid>com.google.guava/<groupid>
<artifactid>guava/<artifactid>
<version>21.0/<version>
/<dependency>/<code>

二、自定義LocalLock註解

自定義一個LocalLock註解用於需要防止重複提交的方法上

<code>/**
* 鎖的註解
*
*/
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface LocalLock {

/**
* @author fly
*/
String key() default "";
}/<code>

註解定義好以後就需要做AOP攔截器切面的具體實現,在 interceptor() 方法上採用的是 Around(環繞增強) ,所有帶 LocalLock 註解的都將被切面處理;

既然是緩存,那緊跟的屬性一定要有過期時間,通過expireAfterWrite 設置緩存的過期時間,maximumSize設置緩存的個數。

通過在內存中查詢key是否存在來判斷是否讓再次提交,和Redis的setNX方法是一個原理。

那麼這個註解該怎麼用呢?

<code>@Aspect
@Configuration
public class LockMethodInterceptor {

private static final Cache<string> CACHES = CacheBuilder.newBuilder()
// 最大緩存 100 個
.maximumSize(1000)
// 設置寫緩存後 5 秒鐘過期

.expireAfterWrite(5, TimeUnit.SECONDS)
.build();

@Around("execution(public * *(..)) && @annotation(com.battcn.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;
}
}/<string>/<code>

控制層的實現

我們將註解加在控制層方法上,key = "city:arg[0] key自己定義,arg[0]這個匹配規則表示替換成第一個參數。那麼就實現city:token在一定時間內不可以重複提交了

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

@LocalLock(key = "city:arg[0]")
@GetMapping
public String query(@RequestParam String token) {
return "ok- " + token;
}
}/<code>

測試

接下來我們就測試一下,我用的是postman

第一請求正常響應

Springboot 2.x 如何解決重複提交 (本地鎖的實踐)

緊接著請求第二次,返回結果“重複提交”,顯然我們實現成功了

Springboot 2.x 如何解決重複提交 (本地鎖的實踐)

很多時候我們都被一些技術高大上又抽象的專業名稱所迷惑,看似遙不可及晦澀難懂,但事實上動手實踐一下,你會發現簡單得很!

學無止境,一起加油!


作者:程序員內點事
鏈接:https://juejin.im/post/5e3ad419518825495c657c2d


分享到:


相關文章: