Java使用Semaphore對單接口進行限流

package com.codecoord.semaphore.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.Semaphore;

/**
 * 限流側測試
 *
 * @author [email protected]
 * @since 2020-08-06
 */
@RestController
public class SemaphoreController {
    /**
     * 最大信號量，例如此處3，生成環境可以做成可配置項，通過註入方式進行註入
     */
    private static final int MAX_SEMAPHORE = 3;
    /**
     * Semaphore主限流，全局就行
     */
    private static final Semaphore SEMAPHORE = new Semaphore(MAX_SEMAPHORE);

    @RequestMapping("/limit")
    public String limit() {
        // 01.使用非阻塞tryAcquire，如果獲取不到就快速返回失敗
        if (!SEMAPHORE.tryAcquire()) {
            return "請求頻率超過限制：" + MAX_SEMAPHORE;
        }

        // 02. 如果能進入到這裡，說明一定獲取到瞭許可證
        /// todo 可能的參數校驗，註意如果參數校驗不通過，一定要調用release方法
        /*if (valid(xxx)) {
            SEMAPHORE.release();
        }*/

        try {
            // 03. 模擬業務處理，假如需要1s
            Thread.sleep(1000);
            return "業務處理成功";
        } catch (InterruptedException e) {
            // 錯誤處理
            return "業務處理失敗";
        } finally {
            // 04. 一定要釋放，否則導致接口假死無法處理請求
            SEMAPHORE.release();
        }
    }
}