springboot + aop的最佳實踐，再也不怕輕按兩下666

唠唠叨叨

複工不到一個月瘋狂的加班中，最近沒什麼時間寫部落格更文有點慢，好在做的項目中不全是增删改的業務邏輯，還是有一些比較有意思實用的技術點，是以整理出來分享給大家。和那些搞高端技術的大佬比不了，咱就是個還在一線搬磚的碼農。

有沒有遇到過這種情況：由于網速等原因，網頁響應很慢，送出一次表單後發現服務久久沒響應，然後你就瘋狂點選送出按鈕（12306就經常被這樣怒怼），如果做過防重複送出還好，否則那是什麼級别的災難就不好說了。。。

今天主要是用

自定義注解、

AOP、

·

Guava

包中Cache來生成一種本地鎖，來達到的防重複送出效果，整體的實作比較簡單，沒有什麼太大的難度，代碼也是比較少，，由于是基于記憶體的緩存，是以這種實作方式并不适用于分布式服務。旨在給大家介紹一種實作防重複送出的方案，要是有什麼說的不對的地方大家溫柔一點撕，畢竟人家還是個20出頭30郎當歲的孩子。

Guava是什麼？

guava

包是個啥？做過Java的小夥伴應該多少都有所了解，它是

google

“嫌棄”

JAVA

自帶的類庫不好用，自行研發的一套工具包，對JDK工具做了很好的拓展。例如：并發[

Concurrency

]、緩存[

Caches

]、 函數式風格[

Functional idioms

]、 字元串處理[

Strings

]等。

總之一句話

guava

包很好用多去了解下，能不造的輪子咱們就盡量不去造，因為咱們自己造的輪子可能不太圓。不多說廢話了，來看看具體的實作。

代碼撸起來

1、引入Guava包依賴

第一步引入

Guava

的依賴包沒什麼好說的

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

2、自定義注解

服務端實作防止重複送出，一般都是利用

AOP

自定義注解的的方式，作用于

controller

的入口方法。自定義一個

LocalLock

注解用于需要防止重複送出的方法上。

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

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

注解定義好以後就需要做AOP攔截器切面的具體實作，在

interceptor()

方法上采用的是

Around

（環繞增強） ，是以所有帶

LocalLock

注解的都将被切面處理；

@Around("execution(public * *(..)) && @annotation(com.chengxy.annotation.LocalLock)")           

既然是緩存，那緊跟的屬性一定要有過期時間，通過

expireAfterWrite

設定緩存的過期時間，

maximumSize

設定緩存的個數。

通過在記憶體中查詢key是否存在來判斷是否讓再次送出，和

Redis

的

setNX

方法是一個比較像。

這裡我們設定同一個方法，5秒鐘内相同參數的請求隻允許執行一次。

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

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

3、控制層的實作

我們将注解加在控制層方法上，

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

4、測試

接下來我們就測試一下，預期結果：5秒内隻有第一次的送出會正常傳回，其餘的顯示

“請勿重複送出”

，看看執行結果是不是我們預期的那樣，這裡用

postman

測試。

第一次請求正常響應：

緊接着請求第二次，傳回結果“重複送出”，顯然我們實作成功了

小福利：

擷取到一些極客課程 ，噓~，免費 送給小夥伴們。公号【程式員内點事】回複【極客】自行領取