新疆信息港 > 財(cái)經(jīng) >Springboot 2.x 如何解決重復(fù)提交（本地鎖的實(shí)踐）!

Springboot 2.x 如何解決重復(fù)提交（本地鎖的實(shí)踐）!

2020-11-20 12:03:09

來源：互聯(lián)網(wǎng)

閱讀：-

推薦閱讀：Java程序員面完字節(jié)面小米，字節(jié)涼涼后沒想到拿到小米高薪offerBAT工作超十年，總結(jié)這份文檔讓您成為Java崗位春招offer收割機(jī)字節(jié)跳動(dòng)Java3面“涼涼”~不負(fù)韶華，努力復(fù)習(xí)備戰(zhàn)“金三銀四”有沒有遇到過這種情況：網(wǎng)頁響應(yīng)很慢，提交一次表單后發(fā)現(xiàn)沒反應(yīng)，然后你...

BAT工作超十年，總結(jié)這份文檔讓您成為Java崗位春招offer收割機(jī)

字節(jié)跳動(dòng)Java3面“涼涼”~不負(fù)韶華，努力復(fù)習(xí)備戰(zhàn)“金三銀四”

有沒有遇到過這種情況：網(wǎng)頁響應(yīng)很慢，提交一次表單后發(fā)現(xiàn)沒反應(yīng)，然后你就瘋狂點(diǎn)擊提交按鈕（12306就經(jīng)常被這樣怒懟），如果做過防重復(fù)提交還好，否則那是什么級別的災(zāi)難就不好說了。。。

本文主要是應(yīng)用自定義注解、 spring AOP、· Guava Cache 生成一種本地鎖，來達(dá)到的防重復(fù)提交效果，由于是基于內(nèi)存的緩存，所以這種實(shí)現(xiàn)方式并不適用于分布式服務(wù)

Guava是什么？

guava包是google嫌棄JAVA自帶的類庫不好用，自行研發(fā)的一套工具包，對JDK工具做了很好的拓展。例如：并發(fā)[Concurrency]、緩存[Caches]、函數(shù)式風(fēng)格[Functional idioms]、字符串處理[Strings]等等。

一、引入Guava包依賴

        com.google.guava        guava        21.0

二、自定義LocalLock注解

自定義一個(gè)LocalLock注解用于需要防止重復(fù)提交的方法上

/** * 鎖的注解 * */@Target(ElementType.METHOD)@Retention(RetentionPolicy.RUNTIME)@Documented@Inheritedpublic @interface LocalLock {    /**     * @author fly     */    String key() default "";}

注解定義好以后就需要做AOP攔截器切面的具體實(shí)現(xiàn)，在 interceptor() 方法上采用的是 Around（環(huán)繞增強(qiáng)），所有帶 LocalLock 注解的都將被切面處理；

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

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

那么這個(gè)注解該怎么用呢？

@Aspect@Configurationpublic class LockMethodInterceptor {    private static final Cache CACHES = CacheBuilder.newBuilder()            // 最大緩存 100 個(gè)            .maximumSize(1000)            // 設(shè)置寫緩存后 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("請勿重復(fù)請求");            }            // 如果是第一次請求,就將 key 當(dāng)前對象壓入緩存中            CACHES.put(key, key);        }        try {            return pjp.proceed();        } catch (Throwable throwable) {            throw new RuntimeException("服務(wù)器異常");        } finally {            // TODO 為了演示效果,這里就不調(diào)用 CACHES.invalidate(key); 代碼了        }    }    /**     * key 的生成策略,如果想靈活可以寫成接口與實(shí)現(xiàn)類的方式（TODO 后續(xù)講解）     *     * @param keyExpress 表達(dá)式     * @param args       參數(shù)     * @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;    }}

控制層的實(shí)現(xiàn)

我們將注解加在控制層方法上，key = "city:arg[0] key自己定義，arg[0]這個(gè)匹配規(guī)則表示替換成第一個(gè)參數(shù)。那么就實(shí)現(xiàn)city:token在一定時(shí)間內(nèi)不可以重復(fù)提交了

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