讲师中心 微信公众号
AI工具推荐 视频效率加速

Linux怎么在Shell中实现简单的令牌桶算法进行平滑流量限速

夜强酱_1645

夜强酱_1645

发布时间:2026-10-18 10:17:31

|

895人浏览过

|

来源于php中文网

原创

令牌桶算法能允许突发流量且保持平均速率,而单纯sleep只能实现固定间隔限流,无法应对真实场景中的流量波动需求。

linux怎么在shell中实现简单的令牌桶算法进行平滑流量限速

为什么直接用 rate 或 sleep 不够用

单纯用 sleep 0.1 控制每秒 10 次调用,会把请求压成固定间隔(比如严格 0.1s/次),但实际中你可能希望:允许短时间突发(比如 5 个请求瞬间进来)、之后自动降速、整体仍不超平均速率。这就是令牌桶要解决的问题——它比固定 sleep 更贴近真实限速需求。

Shell 本身没内置令牌桶,但可以用 date +%s.%N + 简单状态变量模拟核心逻辑,关键在于「按时间匀速产令牌」+「消耗前检查余额」。

用 Bash 实现带桶容量的限速函数 throttle

下面这个 throttle 函数支持设置最大速率(如每秒 2 个)和桶大小(如最多攒 5 个令牌),所有状态只靠几个变量维持,无需外部依赖或临时文件:

throttle() {
    local rate=${1:-1}      # 每秒允许请求数(浮点也行,如 0.5)
    local burst=${2:-1}     # 最大令牌数(桶容量)
    local now=$(date +%s.%N)
<pre class='brush:php;toolbar:false;'># 初始化或更新令牌
if [[ -z "$_throttle_last" ]]; then
    _throttle_tokens=$burst
    _throttle_last=$now
else
    local elapsed=$(echo "$now - $_throttle_last" | bc -l)
    local new_tokens=$(echo "$elapsed * $rate" | bc -l)
    _throttle_tokens=$(echo "scale=0; $_throttle_tokens + $new_tokens" | bc -l)
    _throttle_tokens=$(echo "if ($_throttle_tokens > $burst) $burst else $_throttle_tokens" | bc -l)
    _throttle_last=$now
fi

# 消耗一个令牌
if (( $(echo "$_throttle_tokens >= 1" | bc -l) )); then
    _throttle_tokens=$(echo "$_throttle_tokens - 1" | bc -l)
    return 0
else
    # 等待直到能拿到令牌(计算还需等多久)
    local wait=$(echo "(1 - $_throttle_tokens) / $rate" | bc -l)
    sleep $(echo "if ($wait < 0) 0 else $wait" | bc -l)
    _throttle_tokens=0
    _throttle_last=$(date +%s.%N)
    return 0
fi

}

Obsidian (Linux)
Obsidian (Linux)

操作 Obsidian 库(纯 Markdown 笔记),并通过 notesmd-cli 实现自动化。

下载

使用示例:for i in {1..10}; do echo "req $i"; throttle 3 4; done 表示每秒最多 3 次、最多突发 4 次。

  • rate 为 0.2 时,平均 5 秒一次;burst 设为 3 就允许开头连发 3 次
  • 必须用 bc,因为 Shell 原生不支持浮点运算;若系统没装 bc,得先 apt install bc 或 yum install bc
  • 所有状态变量(_throttle_tokens 等)是全局的,不能并发调用;如需多进程安全,得改用文件锁或 redis

遇到 command not found: bc 或精度不准怎么办

bc 缺失是最常见报错,但还有更隐蔽的问题:某些嵌入式 BusyBox 环境里 date +%N 不可用,或 bc 版本太老不支持 scale。这时得降级处理:

  • 用 date +%s 替代 %N(秒级精度),适合对突发不敏感的场景,比如限速日志轮转
  • 避免 bc:用整数微秒计算,例如把速率换算成「每毫秒多少令牌」,再用 $(( )) 整数运算,但会损失小数精度
  • 简单替代方案:用 tc(traffic control)在网卡层限速,比如 tc qdisc add dev lo root tbf rate 1mbit burst 32kbit latency 400ms,但这作用于网络流,不是 shell 进程级限速

和 ratelimit 工具或 systemd-run --scope 的区别

有人会想到用现成工具,比如 ratelimit(C 写的小程序)或 systemd-run --scope --scope-property=CPUQuota=10%。它们确实更稳,但有硬约束:

  • ratelimit 只能包装命令(ratelimit -r 10 -w 1 curl ...),无法嵌入脚本逻辑做条件限速
  • systemd-run 依赖 systemd,且 CPUQuota 是粗粒度资源限制,不是请求计数器,对 I/O 密集型任务(如频繁 curl)几乎无效
  • 自己写的 throttle 函数可以放在任意位置:循环里、管道后、甚至 find -exec 中,控制粒度最细

真正难的是时间精度和跨进程状态同步——如果脚本被多个终端同时运行,每个实例都有自己的 _throttle_tokens,那就不是“全局限速”而是“单实例限速”。这时候就得上 Redis 或文件锁,纯 Shell 就到头了。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

更多
页面置换算法
页面置换算法

页面置换算法是操作系统中用来决定在内存中哪些页面应该被换出以便为新的页面提供空间的算法。本专题为大家提供页面置换算法的相关文章,大家可以免费体验。

5816

2023.08.14

磁盘配额是什么
磁盘配额是什么

磁盘配额是计算机中指定磁盘的储存限制,就是管理员可以为用户所能使用的磁盘空间进行配额限制,每一用户只能使用最大配额范围内的磁盘空间。php中文网为大家提供各种磁盘配额相关的内容,教程,供大家免费下载安装。

4070

2023.06.21

如何安装LINUX
如何安装LINUX

本站专题提供如何安装LINUX的相关教程文章,还有相关的下载、课程,大家可以免费体验。

3473

2023.06.29

linux find
linux find

find是linux命令,它将档案系统内符合 expression 的档案列出来。可以指要档案的名称、类别、时间、大小、权限等不同资讯的组合,只有完全相符的才会被列出来。find根据下列规则判断 path 和 expression,在命令列上第一个 - ( ) , ! 之前的部分为 path,之后的是 expression。还有指DOS 命令 find,Excel 函数 find等。本站专题提供linux find相关教程文章,还有相关

2653

2023.06.30

linux修改文件名
linux修改文件名

本专题为大家提供linux修改文件名相关的文章,这些文章可以帮助用户快速轻松地完成文件名的修改工作,大家可以免费体验。

6470

2023.07.05

linux系统安装教程
linux系统安装教程

linux系统是一种可以免费使用,自由传播,多用户、多任务、多线程、多CPU的操作系统。本专题提供linux系统安装教程相关的文章,大家可以免费体验。

6102

2023.07.06

linux查看文件夹大小
linux查看文件夹大小

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。linux怎么查看文件夹大小呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

3553

2023.07.20

linux查看ip命令
linux查看ip命令

本专题为大家提供linux查看ip命令相关文章内容,感兴趣的朋友可以免费下载体验试试。

3045

2023.07.20

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

20

2026.10.16

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Miniconda官方安装文档
Miniconda官方安装文档

共0课时 | 0人学习

phpMyAdmin 安装文档
phpMyAdmin 安装文档

共0课时 | 0人学习

安装 Xdebug 并配置 phpstorm
安装 Xdebug 并配置 phpstorm

共1课时 | 185人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn