package com.codecoord.semaphore.controller;

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

import java.util.concurrent.Semaphore;

/**
 * 限流側(cè)測試
 *
 * @author tianxincode@163.com
 * @since 2020-08-06
 */
@RestController
public class SemaphoreController {
    /**
     * 最大信號量，例如此處3，生成環(huán)境可以做成可配置項，通過注入方式進行注入
     */
    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 可能的參數(shù)校驗，注意如果參數(shù)校驗不通過，一定要調(diào)用release方法
        /*if (valid(xxx)) {
            SEMAPHORE.release();
        }*/

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