0

0

如何限制API请求速度,bandwidth-throttle/token-bucket助你构建健壮服务

WBOY

WBOY

发布时间:2025-07-20 12:24:14

|

672人浏览过

|

来源于php中文网

原创

最近在开发一个处理用户提交数据的程序时,遇到了一个棘手的问题:用户输入的文本中包含各种非ASCII字符,例如中文、日文、特殊符号等等。这些字符导致程序在处理字符串时效率低下,甚至出现错误。为了解决这个问题,我尝试了多种方法,最终找到了voku/portable-ascii这个库。 Composer在线学习地址:学习地址

想象一下,你的API正在“裸奔”

在一个风和日丽的日子,你的新api上线了。起初一切顺利,但随着用户量的增长,一些“不速之客”开始频繁地调用你的接口。他们可能是爬虫,可能是恶意攻击者,也可能是无意中触发了高频请求的用户。

很快,服务器CPU飙升,内存告警,数据库连接池耗尽,你的服务开始变得缓慢,甚至直接崩溃。用户怨声载道,而你却手足无措。你尝试过简单的IP限制,但很快发现这治标不治本,因为IP可以轻易更换,而且会误伤正常用户。你意识到,你需要一个更智能、更灵活的限流机制。

手动实现一个健壮的限流系统绝非易事。你可能需要考虑:

  • 并发问题: 多个请求同时到达时,如何保证限流逻辑的原子性?
  • 状态存储: 限流状态(例如某个用户已发起了多少请求)需要持久化,是存文件、Redis还是数据库?
  • 限流算法: 漏桶、令牌桶,哪种更适合我的场景?如何实现?
  • 灵活性: 不同的接口需要不同的限流策略,如何配置和管理?

这些问题让开发者头疼不已,如果自己从零开始造轮子,不仅耗时耗力,还容易引入新的bug。

引入救星:bandwidth-throttle/token-bucket

正当我为此苦恼时,我发现了bandwidth-throttle/token-bucket这个Composer库。它是一个基于令牌桶(Token Bucket)算法的PHP实现,专门用于解决资源使用率限制的问题。无论是限制API调用频率、控制文件下载带宽,还是防止表单重复提交,它都能轻松应对。

什么是令牌桶算法? 简单来说,你可以把令牌桶想象成一个固定容量的桶,里面不断地以恒定速率生成“令牌”。每次请求想要访问资源时,必须先从桶里取走一个令牌。如果桶里没有令牌了,请求就必须等待,直到有新的令牌生成,或者直接被拒绝。这样,即使短时间内有大量请求涌入,桶里令牌的数量也限制了实际处理的速率,从而保护了你的服务。

bandwidth-throttle/token-bucket库的亮点在于:

  • 线程安全: 在多进程/高并发环境下也能稳定工作,避免了常见的竞态条件。
  • 灵活的存储机制: 支持文件、会话、全局等多种存储方式,满足不同粒度的限流需求。
  • 易于集成: 通过Composer安装,API设计直观,几行代码即可实现限流。

如何使用Composer和bandwidth-throttle/token-bucket实现API限流

首先,使用Composer安装这个库:

composer require bandwidth-throttle/token-bucket

安装完成后,你就可以在你的项目中引入并使用了。下面我们以一个常见的API限流场景为例:限制所有请求每秒最多10次。

bootstrap(10); // 首次运行时取消注释,之后可以注释掉

// 在实际的API处理逻辑前进行限流检查
if (!$bucket->consume(1, $seconds)) { // 尝试消耗1个令牌
    // 如果令牌不足,则返回429 Too Many Requests状态码,并告知客户端多久后可以重试。
    http_response_code(429);
    header(sprintf("Retry-After: %d", floor($seconds)));
    echo "Too Many Requests. Please try again in " . floor($seconds) . " seconds.";
    exit();
}

// 如果成功消耗了令牌,则继续处理API请求
echo "API response: Data fetched successfully!";

?>

代码解析:

YouMind
YouMind

AI内容创作和信息整理平台

下载
  • FileStorage(__DIR__ . "/api.bucket"): 创建一个文件存储实例。api.bucket是用于保存令牌桶状态的文件名。在生产环境中,你可能更倾向于使用RedisStorageMemcachedStorage来支持更高效的并发访问分布式部署
  • Rate(10, Rate::SECOND): 定义了令牌的生成速率为每秒10个。你也可以设置为Rate::MINUTERate::HOUR等。
  • TokenBucket(10, $rate, $storage): 创建一个令牌桶实例。10是桶的容量,意味着即使在短时间内,你最多也只能处理10个突发请求。
  • $bucket->bootstrap(10): 初始化令牌桶,并填充初始令牌数量(这里是10个)。请注意,这个方法应该在应用部署或启动时执行一次,而不是在每个请求中执行,以避免不必要的I/O操作。
  • $bucket->consume(1, $seconds): 尝试从桶中消耗1个令牌。
    • 如果成功消耗,consume()返回true,你可以继续处理请求。
    • 如果令牌不足,consume()返回false,并通过引用将需要等待的秒数赋值给$seconds变量。此时,你可以向客户端返回HTTP 429状态码,并附带Retry-After头,告知客户端何时可以重试。

阻塞式消费(BlockingConsumer)

在某些场景下,你可能不希望直接拒绝请求,而是让请求等待直到有令牌可用。bandwidth-throttle/token-bucket也提供了BlockingConsumer来实现这一功能:

bootstrap(10); // 同样,在应用启动时执行一次

$consumer = new BlockingConsumer($bucket);

echo "Attempting to consume token...\n";
// 这行代码会阻塞,直到有1个令牌可用。
// 如果桶中没有令牌,它会暂停脚本执行,直到令牌被补充。
$consumer->consume(1); 

echo "API response: Data fetched successfully after waiting (if needed)!";

?>

BlockingConsumer在令牌不足时会暂停脚本执行,直到令牌被补充。这对于需要保证所有请求最终都能被处理的场景非常有用,例如后台任务队列处理、消息消费等,但对于Web API来说,直接返回429通常是更好的选择,因为它不会长时间占用服务器资源。

不同范围的存储(Scope of the storage)

这个库最棒的特性之一是它对限流范围的灵活支持:

  • RequestScope 仅在当前请求生命周期内限流。例如,限制一个文件下载过程中每秒的最大带宽。每个请求都有独立的限流。
  • SessionScope 基于用户会话限流。例如,限制每个登录用户每分钟的API调用次数。
  • GlobalScope 全局限流。所有进程/请求共享同一个令牌桶。这正是我们上面API限流示例所使用的场景,它通过文件锁或Redis/Memcached的原子操作保证了线程安全。

根据你的具体需求,选择合适的存储实现是构建高效限流系统的关键。

总结:bandwidth-throttle/token-bucket的强大之处

通过bandwidth-throttle/token-bucket,我们能够:

  1. 提升服务稳定性: 有效防止API被滥用或遭受DDoS攻击,确保核心服务的正常运行。
  2. 优化资源利用: 避免服务器资源耗尽,在有限的资源下提供更稳定的服务。
  3. 实现公平访问: 确保所有用户都能获得相对公平的资源访问机会。
  4. 简化开发: 无需手动实现复杂的限流逻辑和并发控制,大大降低开发和维护成本。
  5. 高度灵活: 支持多种限流速率、桶容量和存储策略,满足各种复杂场景的需求。

结合Composer的依赖管理能力,bandwidth-throttle/token-bucket成为了PHP开发者构建健壮、高效服务的利器。下次当你的API面临高并发挑战时,不妨试试这个强大的库,它一定会让你事半功倍!

相关专题

更多
php文件怎么打开
php文件怎么打开

打开php文件步骤:1、选择文本编辑器;2、在选择的文本编辑器中,创建一个新的文件,并将其保存为.php文件;3、在创建的PHP文件中,编写PHP代码;4、要在本地计算机上运行PHP文件,需要设置一个服务器环境;5、安装服务器环境后,需要将PHP文件放入服务器目录中;6、一旦将PHP文件放入服务器目录中,就可以通过浏览器来运行它。

2490

2023.09.01

php怎么取出数组的前几个元素
php怎么取出数组的前几个元素

取出php数组的前几个元素的方法有使用array_slice()函数、使用array_splice()函数、使用循环遍历、使用array_slice()函数和array_values()函数等。本专题为大家提供php数组相关的文章、下载、课程内容,供大家免费下载体验。

1593

2023.10.11

php反序列化失败怎么办
php反序列化失败怎么办

php反序列化失败的解决办法检查序列化数据。检查类定义、检查错误日志、更新PHP版本和应用安全措施等。本专题为大家提供php反序列化相关的文章、下载、课程内容,供大家免费下载体验。

1485

2023.10.11

php怎么连接mssql数据库
php怎么连接mssql数据库

连接方法:1、通过mssql_系列函数;2、通过sqlsrv_系列函数;3、通过odbc方式连接;4、通过PDO方式;5、通过COM方式连接。想了解php怎么连接mssql数据库的详细内容,可以访问下面的文章。

952

2023.10.23

php连接mssql数据库的方法
php连接mssql数据库的方法

php连接mssql数据库的方法有使用PHP的MSSQL扩展、使用PDO等。想了解更多php连接mssql数据库相关内容,可以阅读本专题下面的文章。

1414

2023.10.23

html怎么上传
html怎么上传

html通过使用HTML表单、JavaScript和PHP上传。更多关于html的问题详细请看本专题下面的文章。php中文网欢迎大家前来学习。

1234

2023.11.03

PHP出现乱码怎么解决
PHP出现乱码怎么解决

PHP出现乱码可以通过修改PHP文件头部的字符编码设置、检查PHP文件的编码格式、检查数据库连接设置和检查HTML页面的字符编码设置来解决。更多关于php乱码的问题详情请看本专题下面的文章。php中文网欢迎大家前来学习。

1445

2023.11.09

php文件怎么在手机上打开
php文件怎么在手机上打开

php文件在手机上打开需要在手机上搭建一个能够运行php的服务器环境,并将php文件上传到服务器上。再在手机上的浏览器中输入服务器的IP地址或域名,加上php文件的路径,即可打开php文件并查看其内容。更多关于php相关问题,详情请看本专题下面的文章。php中文网欢迎大家前来学习。

1305

2023.11.13

Java 桌面应用开发(JavaFX 实战)
Java 桌面应用开发(JavaFX 实战)

本专题系统讲解 Java 在桌面应用开发领域的实战应用,重点围绕 JavaFX 框架,涵盖界面布局、控件使用、事件处理、FXML、样式美化(CSS)、多线程与UI响应优化,以及桌面应用的打包与发布。通过完整示例项目,帮助学习者掌握 使用 Java 构建现代化、跨平台桌面应用程序的核心能力。

0

2026.01.14

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
第二十四期_PHP8编程
第二十四期_PHP8编程

共86课时 | 3.4万人学习

成为PHP架构师-自制PHP框架
成为PHP架构师-自制PHP框架

共28课时 | 2.4万人学习

第二十三期_PHP编程
第二十三期_PHP编程

共93课时 | 6.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号