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

夜强酱_1645

夜强酱_1645

2026-10-18

895人浏览

原创

令牌桶算法能允许突发流量且保持平均速率,而单纯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 <p>}</p><div class="aritcle_card flexRow artxards">
											<div class="artcardd flexRow">
												<a class="aritcle_card_img" rel="nofollow" href="/xiazai/skill8828" title="Obsidian (Linux)"><img
														src="https://img.php.cn/upload/skill/000/000/081/179228531040284.jpg" alt="Obsidian (Linux)" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
												<div class="aritcle_card_info flexColumn">
													<a rel="nofollow" href="/xiazai/skill8828" title="Obsidian (Linux)" class="overflowclass">Obsidian (Linux)</a>
													<p class="overflowclass">操作 Obsidian 库(纯 Markdown 笔记),并通过 notesmd-cli 实现自动化。</p>
												</div>
												<a rel="nofollow" href="/xiazai/skill8828" title="Obsidian (Linux)" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
												</a>
											</div>
										</div>

使用示例: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 就到头了。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2023.08.14

5836

4

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

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

2023.06.21

4070

4

如何安装LINUX
如何安装LINUX

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

2023.06.29

3473

10

linux find
linux find

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

2023.06.30

2653

5

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

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

2023.07.05

6470

7

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

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

2023.07.06

6122

7

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

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

2023.07.20

3553

8

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

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

2023.07.20

3045

4

linux查看cpu使用率
linux查看cpu使用率

在linux的系统维护中,可能需要经常查看cpu使用率,分析系统整体的运行情况。本专题为大家带来了linux查看cpu使用率的相关文章,感兴趣的朋友千万不要错过了。

2023.07.25

3279

6

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
尚观shell视频教程
尚观shell视频教程

共8课时 | 1.9万人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

Buffalo框架快速入门指南
Buffalo框架快速入门指南

共0课时 | 0人学习