Java实现令牌桶算法：详细讲解与代码示例

默语

于 2024-11-13 08:15:00 发布

阅读量1.8w

点赞数 8

分类专栏： JAVA 文章标签： java 开发语言

div class=creativecommons h1a href=https://devpress.csdn.net/beijing target=_blankstrong北京城市开发者社区主理人-默语 /strong/a/h1 /div

本文链接：https://blog.csdn.net/qq_42055933/article/details/143694116

版权

JAVA 专栏收录该内容

205 篇文章

订阅专栏

Java实现令牌桶算法：详细讲解与代码示例

Java实现令牌桶算法：详细讲解与代码示例

在这里插入图片描述

博主默语带您 Go to New World.
✍ 个人主页—— 默语的博客👦🏻 优秀内容
 《java 面试题大全》
《java 专栏》
《idea技术专区》
《spring boot 技术专区》
《MyBatis从入门到精通》
《23种设计模式》
《经典算法学习》
《spring 学习》
《MYSQL从入门到精通》数据库是开发者必会基础之一~
🍩惟余辈才疏学浅，临摹之作或有不妥之处，还请读者海涵指正。☕🍭
🪁 吾期望此文有资助于尔，即使粗浅难及深广，亦备添少许微薄之助。苟未尽善尽美，敬请批评指正，以资改进。！💻⌨

默语是谁？

大家好，我是默语，别名默语博主，擅长的技术领域包括Java、运维和人工智能。我的技术背景扎实，涵盖了从后端开发到前端框架的各个方面，特别是在Java 性能优化、多线程编程、算法优化等领域有深厚造诣。

目前，我活跃在CSDN、掘金、阿里云和 51CTO等平台，全网拥有超过10万的粉丝，总阅读量超过1400 万。统一 IP 名称为默语或者默语博主。我是 CSDN 博客专家、阿里云专家博主和掘金博客专家，曾获博客专家、优秀社区主理人等多项荣誉，并在 2023 年度博客之星评选中名列前 50。我还是 Java 高级工程师、自媒体博主，北京城市开发者社区的主理人，拥有丰富的项目开发经验和产品设计能力。希望通过我的分享，帮助大家更好地了解和使用各类技术产品，在不断的学习过程中，可以帮助到更多的人，结交更多的朋友.

我的博客内容涵盖广泛，主要分享技术教程、Bug解决方案、开发工具使用、前沿科技资讯、产品评测与使用体验。我特别关注云服务产品评测、AI 产品对比、开发板性能测试以及技术报告，同时也会提供产品优缺点分析、横向对比，并分享技术沙龙与行业大会的参会体验。我的目标是为读者提供有深度、有实用价值的技术洞察与分析。

Java实现令牌桶算法：详细讲解与代码示例

摘要
令牌桶算法是一种常见的限流算法，通过为请求分配令牌来限制请求的进入频率，以此来平滑流量。本文面向初学者，将详细介绍令牌桶算法的设计思路，并给出完整的Java代码示例，帮助大家更好地理解和实现令牌桶限流算法。

引言

在现代的分布式系统中，限流是保障系统稳定性的重要手段之一。限流算法可以有效控制突发流量对系统的冲击，使请求平稳进入系统。相较于漏桶算法，令牌桶算法能够更加灵活地应对突发流量，因此在限流场景中得到了广泛应用。本文将介绍令牌桶算法的设计原理，Java代码实现，以及如何优化限流效果。

正文

什么是令牌桶算法？

令牌桶算法（Token Bucket）通过在固定时间间隔生成令牌，并为请求分配令牌来决定是否允许请求进入。如果令牌桶满了，新增的令牌会被丢弃，而没有令牌的请求则会被限制进入系统。这样可以有效平滑请求流量，避免流量暴增导致系统过载。

令牌桶算法的设计思路

核心设计要点

令牌生成：令牌会以固定的速率生成，存放在一个容量固定的令牌桶中。
令牌桶的容量限制：如果令牌桶达到上限，新增的令牌将被丢弃。
请求处理：每个请求进来时，先从令牌桶中申请令牌；如果成功获得令牌，则允许请求继续，否则请求被丢弃或延迟。

举例说明

假设令牌桶的容量为10，每秒生成5个令牌。当一个请求到来时，如果桶内有令牌，则允许请求通过并消耗一个令牌。若桶中无令牌，且令牌生成速率较慢，则该请求被拒绝。

令牌桶算法的代码实现

以下是基于Java的令牌桶算法实现，通过ScheduledExecutorService定期生成令牌并放入令牌桶中，以控制请求的限流效果。

import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class TokenBucketRateLimiter {
    private final int bucketCapacity; // 令牌桶的容量
    private final int refillRate; // 令牌生成速率（每秒生成的令牌数）
    private AtomicInteger tokens; // 当前令牌数

    public TokenBucketRateLimiter(int bucketCapacity, int refillRate) {
        this.bucketCapacity = bucketCapacity;
        this.refillRate = refillRate;
        this.tokens = new AtomicInteger(0);
        startRefilling(); // 启动定期生成令牌的线程
    }

    // 尝试获取一个令牌
    public boolean tryAcquire() {
        int currentTokens = tokens.get();
        if (currentTokens > 0) {
            if (tokens.compareAndSet(currentTokens, currentTokens - 1)) {
                System.out.println("请求获得令牌，允许通过");
                return true;
            }
        }
        System.out.println("请求被拒绝：无可用令牌");
        return false;
    }

    // 定期为令牌桶添加令牌
    private void startRefilling() {
        ScheduledExecutorService executor = Executors.newScheduledThreadPool(1);
        executor.scheduleAtFixedRate(() -> {
            int currentTokens = tokens.get();
            if (currentTokens < bucketCapacity) {
                int newTokens = Math.min(bucketCapacity, currentTokens + refillRate);
                tokens.set(newTokens);
                System.out.println("新增令牌，当前令牌数：" + newTokens);
            }
        }, 0, 1, TimeUnit.SECONDS); // 每秒填充令牌
    }

    public static void main(String[] args) {
        TokenBucketRateLimiter rateLimiter = new TokenBucketRateLimiter(10, 2); // 容量为10，每秒生成2个令牌

        // 模拟10个请求
        for (int i = 0; i < 10; i++) {
            boolean allowed = rateLimiter.tryAcquire();
            if (!allowed) {
                System.out.println("请求 " + i + " 被拒绝");
            }
            try {
                Thread.sleep(200); // 模拟请求间隔
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }
        }
    }
}