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测试数据

  * @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();

         }

     }

}