Springboot 2.x 中怎么避免重復提交

Springboot 2.x 中怎么避免重復提交 ，很多新手對此不是很清楚，為了幫助大家解決這個難題，下面小編將為大家詳細講解，有這方面需求的人可以來學習下，希望你能有所收獲。

Guava是什么？

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

一、引入Guava包依賴

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

二、自定義LocalLock注解

自定義一個LocalLock注解用于需要防止重復提交的方法上

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

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

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

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

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

那么這個注解該怎么用呢？

@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在一定時間內不可以重復提交了

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

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

測試

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

第一請求正常響應

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

整理了幾百本各類技術電子書，送給小伙伴們。關注公號回復【666】自行領取。和一些小伙伴們建了一個技術交流群，一起探討技術、分享技術資料，旨在共同學習進步，如果感興趣就加入我們吧！

看完上述內容是否對您有幫助呢？如果還想對相關知識有進一步的了解或閱讀更多相關文章，請關注億速云行業資訊頻道，感謝您對億速云的支持。