2020-02-06 12:34:31 編程技匠

有沒有遇到過這種情況：網頁響應很慢，提交一次表單後發現沒反應，然後你就瘋狂點擊提交按鈕（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@Inheritedpublic @interface LocalLock {    /**     * @author fly     */    String key() default "";}/<code>

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

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

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

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

<code>@Aspect@Configurationpublic 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>