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

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

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

Guava是什麼?

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

1、引入Guava包依賴

<dependency>
        <groupId>com.google.guava</groupId>
        <artifactId>guava</artifactId>
        <version>21.0</version>
</dependency>

2、自定義LocalLock註解

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

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

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

註解定義好之後就須要作AOP攔截器切面的具體實現,在 interceptor() 方法上採用的是 Around(環繞加強) ,全部帶 LocalLock 註解的都將被切面處理;spring

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

經過在內存中查詢key是否存在來判斷是否讓再次提交,和RedissetNX方法是一個原理。服務器

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

@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.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;
    }
}

控制層的實現

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

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

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

測試

接下來咱們就測試一下,我用的是postman分佈式

第一請求正常響應 在這裏插入圖片描述

緊接着請求第二次,返回結果「重複提交」,顯然咱們實現成功了 在這裏插入圖片描述

整理了幾百本各種技術電子書,送給小夥伴們。關注公號回覆【666】自行領取。和一些小夥伴們建了一個技術交流羣,一塊兒探討技術、分享技術資料,旨在共同窗習進步,若是感興趣就加入咱們吧!

在這裏插入圖片描述

不管你是剛入行、仍是已經有幾年經驗的程序員,相信這份面試提綱都會給你很多助力,長按二維碼關注 『 程序員內點事 』 ,回覆 『 offer 』 自行領取,祝你們 offer 拿到手軟

在這裏插入圖片描述

相關文章
相關標籤/搜索