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 tianxincode@163.com
 * @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();
        }
    }
}