单机版java接口限流示例

场景：java某接口进行限流操作，模拟示例

引入guava的包

        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>25.1-jre</version>
        </dependency>

 

import com.google.common.util.concurrent.RateLimiter;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Random;
import java.util.concurrent.TimeUnit;

/**
 * @Author: pangyq
 * @CreateTime: 2023-01-01  14:48
 * @Description: RateLimiter是guava提供的基于令牌桶算法的实现类，可以非常简单的完成限流特技，并且根据系统的实际情况来调整生成token的速率。
 * 使用RateLimiter 实现令牌通方式限流
 * @Version: 1.0
 */
@RestController
public class IndexController {
    // create 方法中传入一个参数 以每秒为单位固定的速率值 1r/s 每秒中往桶中存入一个令牌
    // 独立线程
    RateLimiter rateLimiter = RateLimiter.create(100);
    // 相当于该接口每秒钟时间 只能支持一个客户端访问
    @RequestMapping("/addOrder")
    public String addOrder() {
        // 1.限流处理 限流正常要放在网关 客户端从桶中获取对应的令牌，为什么返回double结果，这个结果表示 从桶中拿到令牌等待时间.
        // 2. 如果获取不到令牌，就会一直等待.设置服务降级处理（相当于配置在规定时间内如果没有获取到令牌的话，直接走服务降级。）
         double acquire = rateLimiter.acquire();
         System.out.println("从桶中获取令牌等待的时间:" + acquire);
        // 如果在500毫秒内如果没有获取到令牌的话，则直接走服务降级处理
        boolean tryAcquire = rateLimiter.tryAcquire(500, TimeUnit.MILLISECONDS);
        if (!tryAcquire) {
            System.out.println("别抢了， 在抢也是一直等待的， 还是放弃吧！！！");
            return "别抢了， 在抢也是一直等待的， 还是放弃吧！！！";
        }
        // 2. 业务逻辑处理  生成随机boolean值测试
        Random randNum = new Random();
        boolean addOrderResult = randNum.nextBoolean();
        if (addOrderResult) {
            System.out.println("恭喜您，抢购成功! 等待时间:" + rateLimiter.acquire());
            return "恭喜您，抢购成功!";
        }
        return "抢购失败!";
    }
}

 

 

 

另外上网搜索了令牌桶算法，仅此记录。