投票项目_实现一个简单的限流

柚＆.屿づ

已于 2024-02-26 20:26:14 修改

阅读量230

点赞数 3

分类专栏：投票项目文章标签： java springboot

于 2024-02-26 20:26:04 首次发布

本文链接：https://blog.csdn.net/qq_63997064/article/details/136307901

版权

投票项目专栏收录该内容

6 篇文章 0 订阅

订阅专栏

1. 导入依赖

<!--    限制请求速率   -->
        <dependency>
            <groupId>com.google.guava</groupId>
            <artifactId>guava</artifactId>
            <version>30.1-jre</version>
        </dependency>

2. 案例使用

    // 例如，创建一个每秒允许生成1个验证码的限流器
    RateLimiter rateLimiter = RateLimiter.create(1.0); // 参数表示每秒令牌生成速率
    // 生成验证码,返回的是 base64
    @GetMapping("/getCaptchaBase64")
    @ApiOperation("返回base64验证码接口")
    public Object getCaptchaBase64(HttpServletRequest request, HttpServletResponse response) {
        if (!rateLimiter.tryAcquire()) {
            // 如果获取令牌失败（即超过限流阈值）
            Map<String, String> result = new HashMap<>();
            result.put("message", "请求过于频繁，请稍后再试");
            return result;
        }
其他代码...
}