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安装这个库:

<code class="bash">composer require bandwidth-throttle/token-bucket</code>

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

<code class="php"><?php

require 'vendor/autoload.php'; // 引入Composer自动加载文件

use bandwidthThrottle\tokenBucket\Rate;
use bandwidthThrottle\tokenBucket\TokenBucket;
use bandwidthThrottle\tokenBucket\storage\FileStorage; // 使用文件存储,简单易用

// 1. 定义存储:这里我们使用文件存储,将令牌桶的状态保存在一个文件中。
// 注意:在生产环境中,更推荐使用RedisStorage或MemcachedStorage以获得更好的性能和并发支持。
$storage = new FileStorage(__DIR__ . "/api.bucket");

// 2. 定义速率:每秒生成10个令牌。
$rate = new Rate(10, Rate::SECOND);

// 3. 创建令牌桶实例:
// 第一个参数是桶的容量(burst capacity),这里设置为10,意味着桶里最多可以积攒10个令牌。
// 第二个参数是令牌的生成速率。
// 第三个参数是存储实例。
$bucket = new TokenBucket(10, $rate, $storage);

// 4. 初始化令牌桶:
// 这一步非常重要,它会初始化令牌桶的状态。
// 建议在应用启动时(例如框架的bootstrap文件或部署脚本中)执行一次,而不是每次请求都执行,
// 否则会增加不必要的存储通信开销。
// $bucket->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!";

?></code>

代码解析:

Stylar
Stylar

多功能一站式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来实现这一功能:

<code class="php"><?php

require 'vendor/autoload.php';

use bandwidthThrottle\tokenBucket\Rate;
use bandwidthThrottle\tokenBucket\TokenBucket;
use bandwidthThrottle\tokenBucket\BlockingConsumer;
use bandwidthThrottle\tokenBucket\storage\FileStorage;

$storage  = new FileStorage(__DIR__ . "/api.bucket");
$rate     = new Rate(10, Rate::SECOND);
$bucket   = new TokenBucket(10, $rate, $storage);

// $bucket->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)!";

?></code>

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面临高并发挑战时,不妨试试这个强大的库,它一定会让你事半功倍!

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
composer是什么插件
composer是什么插件

Composer是一个PHP的依赖管理工具,它可以帮助开发者在PHP项目中管理和安装依赖的库文件。Composer通过一个中央化的存储库来管理所有的依赖库文件,这个存储库包含了各种可用的依赖库的信息和版本信息。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

163

2023.12.25

什么是分布式
什么是分布式

分布式是一种计算和数据处理的方式,将计算任务或数据分散到多个计算机或节点中进行处理。本专题为大家提供分布式相关的文章、下载、课程内容,供大家免费下载体验。

433

2023.08.11

分布式和微服务的区别
分布式和微服务的区别

分布式和微服务的区别在定义和概念、设计思想、粒度和复杂性、服务边界和自治性、技术栈和部署方式等。本专题为大家提供分布式和微服务相关的文章、下载、课程内容,供大家免费下载体验。

252

2023.10.07

登录token无效
登录token无效

登录token无效解决方法:1、检查token的有效期限,如果token已经过期,需要重新获取一个新的token;2、检查token的签名,如果签名不正确,需要重新获取一个新的token;3、检查密钥的正确性,如果密钥不正确,需要重新获取一个新的token;4、使用HTTPS协议传输token,建议使用HTTPS协议进行传输 ;5、使用双因素认证,双因素认证可以提高账户的安全性。

6684

2023.09.14

登录token无效怎么办
登录token无效怎么办

登录token无效的解决办法有检查Token是否过期、检查Token是否正确、检查Token是否被篡改、检查Token是否与用户匹配、清除缓存或Cookie、检查网络连接和服务器状态、重新登录或请求新的Token、联系技术支持或开发人员等。本专题为大家提供token相关的文章、下载、课程内容,供大家免费下载体验。

846

2023.09.14

token怎么获取
token怎么获取

获取token值的方法:1、小程序调用“wx.login()”获取 临时登录凭证code,并回传到开发者服务器;2、开发者服务器以code换取,用户唯一标识openid和会话密钥“session_key”。想了解更详细的内容,可以阅读本专题下面的文章。

1112

2023.12.21

token什么意思
token什么意思

token是一种用于表示用户权限、记录交易信息、支付虚拟货币的数字货币。可以用来在特定的网络上进行交易,用来购买或出售特定的虚拟货币,也可以用来支付特定的服务费用。想了解更多token什么意思的相关内容可以访问本专题下面的文章。

2250

2024.03.01

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

761

2023.08.03

抖漫入口地址合集
抖漫入口地址合集

本专题整合了抖漫入口地址相关合集,阅读专题下面的文章了解更多详细地址。

1

2026.03.17

热门下载

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

精品课程

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

共86课时 | 3.5万人学习

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

共28课时 | 2.6万人学习

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

共93课时 | 7.6万人学习

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

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