实用脚本能批量限流吗？

wen 实用脚本 2026-06-10 11

本文目录导读：

实用脚本能批量限流吗？

场景一：对多个不同的“资源/接口/用户”分别进行限流（批量配置限流规则）
场景二：用“一个脚本程序”本身去限制“多个调用方”的请求（脚本作为限流器）
关键决策：你需要的是“配置脚本”还是“运行时脚本”？
你的具体需求是什么？

“实用脚本能批量限流吗？”——答案是：可以，但需要明确的场景定义。

“批量限流”这个词在运维和开发中通常有两种含义，脚本的实现方式完全不同，我们需要先区分你要限流的对象是什么：

对多个不同的“资源/接口/用户”分别进行限流（批量配置限流规则）

这是最常见的需求：比如有100个API接口，每个接口独立限制为 100次/秒，或者有1000个用户，每个用户限制 10次/分钟。

脚本方式（推荐：使用现成工具 + 配置文件驱动）：

Nginx + Lua 脚本（最主流）可以通过 Lua 脚本编写一个通用的限流函数，然后从一个配置文件（如 JSON 或 CSV）中批量读取接口或 IP 列表，动态生成限流规则。

-- 伪代码示例：从配置列表批量设置限流
local limits = {
    { path = "/api/v1/user", rate = 100, burst = 150 },
    { path = "/api/v1/order", rate = 50, burst = 70 },
    -- ...
}
for i, rule in ipairs(limits) do
    -- 通过 ngx.shared.DICT 或 redis 为每个规则创建独立的计数器
    -- 实际逻辑需接入 lua-resty-limit-traffic
end

Python 脚本 + Redis（适用于自定义网关或微服务） 脚本逻辑：读取一个 rules.csv 文件，然后向 Redis 或 etcd 写入对应的限流策略（sliding window 或 token bucket 算法）。

# 示例：批量写入 Redis Sliding Window 限流键
import csv, time, redis
r = redis.Redis()
with open('api_limits.csv') as f:
    reader = csv.DictReader(f)
    for row in reader:
        key = f"ratelimit:{row['api_name']}"
        # 设置窗口大小和阈值（具体算法略）
        r.set(f"{key}:limit", row['limit_per_second'])

Shell 脚本 + iptables (TCP层面限流) 针对 IP 或端口进行批量限流（不太精确，适合粗粒度场景）。

for ip in $(cat ip_list.txt); do
    iptables -A INPUT -s $ip -m limit --limit 10/s -j ACCEPT
    iptables -A INPUT -s $ip -j DROP
done

这种场景下，脚本非常适合，它本质上是“批量配置生成器和下发器”，难点不在脚本，而在于底层的限流算法引擎（Nginx/Redis/内核）。

用“一个脚本程序”本身去限制“多个调用方”的请求（脚本作为限流器）

这是指写一个脚本（如 Python Web 服务）来接收请求，并在脚本内部进行限流。

脚本做法（通常不推荐用于高并发）：

使用 Token Bucket 算法库 你可以用一个脚本启动多个 Worker（例如使用 Gunicorn + 多进程），然后在全局变量或共享内存中维护一个限流表。

from flask import Flask
from pyrate_limiter import BucketFullException, Duration, RequestRate
from pyrate_limiter import MemoryListBucket
app = Flask(__name__)
# 为每个 client_id 动态创建限流器（批量管理）
rate_limiters = {}
def get_rate_limiter(client_id):
    if client_id not in rate_limiters:
        # 每个客户端独立限流
        rate_limiters[client_id] = MemoryListBucket()
    return rate_limiters[client_id]

缺点： 脚本进程重启会导致内存清空，且多进程之间的状态同步（如文件锁或 Redis）会削弱脚本的简单性。对于单机百万级别并发，脚本不是好选择。

关键决策：你需要的是“配置脚本”还是“运行时脚本”？

目标	最佳脚本工具	是否实用
批量配置 100个接口的限流规则	Python 读取 CSV -> 写入 Nginx conf / Redis	✅ 非常实用
批量测试限流效果	Shell + `ab` / `wrk` 工具	✅ 非常实用
用脚本代替专用网关实时限流	Go 或 Python 异步框架	⚠️ 能写，但性能不如 nginx/caddy
脚本本身不处理请求，只做管理员操作	Ansible / Shell 远程修改防火墙规则	✅ 非常实用