Java 网关怎么实现基于令牌桶算法的限流拦截

风晨大大_1432

风晨大大_1432

2026-09-08

233人浏览

原创

Java网关令牌桶限流核心是在请求入口动态判断放行,需用AtomicLong+CAS保证线程安全,惰性填充令牌,并通过GlobalFilter集成Spring Cloud Gateway,拒绝时返回429响应。

java 网关怎么实现基于令牌桶算法的限流拦截

Java 网关实现基于令牌桶算法的限流拦截,核心是**在请求入口处动态判断是否允许通过**,关键在于:令牌桶的线程安全维护、实时填充逻辑、以及与网关过滤器(如 Spring Cloud Gateway 的 GlobalFilter)的集成。

令牌桶模型的 Java 实现要点

令牌桶需支持并发访问,推荐使用 AtomicLong 存储剩余令牌数,并配合时间戳记录上次填充时间。每次请求到来时:

  • 先计算从上次填充到当前时刻应新增的令牌数(rate × 经过秒数),向上取整或向下取整需统一策略(常用向下取整 + 取 min 限制最大容量)
  • 更新剩余令牌:remaining = min(capacity, current + added)
  • 尝试消耗一个令牌:if (remaining > 0) { remaining--; return true; } else return false;
  • 整个过程需用 CAS(如 compareAndSet)保证原子性,避免锁竞争影响吞吐

在 Spring Cloud Gateway 中集成限流过滤器

以 GlobalFilter 方式嵌入路由链路,示例结构如下:

  • 定义一个 RateLimiter 接口实现类(如 TokenBucketRateLimiter),封装上述令牌桶逻辑
  • 在 filter 方法中提取请求标识(如用户 ID、IP、接口路径等),构造唯一 key,获取对应桶实例(可用 ConcurrentHashMap 缓存桶对象,key → TokenBucket)
  • 调用 tryAcquire() 判断是否放行;若拒绝,直接返回 429 Too Many Requests 响应(注意设置 ServerHttpResponse 并终止 chain)
  • 建议将桶容量、填充速率等参数按 key 动态配置(如从 Nacos 或数据库加载),支持灰度调整

提升稳定性的实用技巧

生产环境需兼顾性能与一致性:

Alibabacloud Sdk Client Initialization For Java
Alibabacloud Sdk Client Initialization For Java

在 Java 中初始化和管理阿里云 SDK客户端。包括单例模式、线程安全、endpoint 与 region 配置、VPC 终端节点、同步与异步等。

下载
  • 避免为每个请求都新建桶对象,复用已存在的桶实例;对高频 key(如热门接口)可预热初始化
  • 令牌填充不依赖定时任务,而是在每次请求时惰性计算,降低系统开销
  • 考虑分布式场景:单机令牌桶只适用于单实例网关;跨节点需引入 Redis + Lua 脚本实现原子操作(如 INCR + 过期时间 + 条件判断)
  • 日志和监控不可少:记录被限流的 key、时间、频次,接入 Prometheus 暴露 rate_limit_rejected_total 等指标

简单可运行的令牌桶参考代码片段

(非完整类,仅示意核心逻辑)

public class TokenBucket {
    private final long capacity;
    private final double refillRatePerSec; // 如 100.0
    private final AtomicLong tokens = new AtomicLong();
    private final AtomicLong lastRefillTime = new AtomicLong(System.nanoTime());
<pre class="brush:php;toolbar:false;">public TokenBucket(long capacity, double refillRatePerSec) {
    this.capacity = capacity;
    this.refillRatePerSec = refillRatePerSec;
    this.tokens.set(capacity);
}

public boolean tryAcquire() {
    long now = System.nanoTime();
    long nanosSinceLastRefill = now - lastRefillTime.get();
    double seconds = nanosSinceLastRefill / 1_000_000_000.0;
    long newTokens = (long) (seconds * refillRatePerSec);
    if (newTokens > 0) {
        long updated = Math.min(capacity, tokens.get() + newTokens);
        tokens.set(updated);
        lastRefillTime.set(now);
    }
    return tokens.get() > 0 && tokens.compareAndSet(tokens.get(), tokens.get() - 1);
}

}

不复杂但容易忽略细节,关键是把“时间驱动填充”和“CAS 安全消费”两个动作揉进一次请求判断里。

Java免费学习笔记:立即使用
解锁 Java 大师之旅:从入门到精通的终极指南

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

2023.06.15

9097

6

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

2023.07.05

6282

9

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

2023.07.31

5612

8

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.01

1004

3

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.08.02

848

3

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

1196

5

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.02

2409

5

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

2023.08.03

19731

3

配置java环境变量
配置java环境变量

配置Java环境变量是为了让操作系统能够识别和使用Java的相关命令和功能。本专题为大家提供配置java环境变量相关文章,帮助大家解决问题。

2023.08.03

1095

8

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习